twitools/makedb.py

#!/usr/bin/env python3

import tools

import sys

def makeDB(path=tools.dbpath()):
 db = tools.dbHelper(path, create = True)

 db.executeQuery("CREATE TABLE tweets(`tweet_id` INTEGER NOT NULL, `in_reply_to_status_id` TEXT, `in_reply_to_user_id` TEXT, `timestamp` TEXT, `source` TEXT, `text` TEXT, `retweeted_status_id` TEXT, `retweeted_status_user_id` TEXT, `retweeted_status_timestamp` TEXT, `expanded_urls` TEXT, PRIMARY KEY(tweet_id));")
 
 db.commit()
 db.closeConnection()
 
if __name__ == "__main__":
 if len(sys.argv) > 2:
  raise ValueError(sys.argv[0] + " only takes one argument, the path of the new database file.")
 try:
  makeDB(sys.argv[1])
 except IndexError:
  makeDB()