Update makedb.py

makedb.py

@@ -3,18 +3,20 @@
 
 import os, sqlite3, sys
 
-try:
+def makeDB(path="Database.db"):
- file = sys.argv[1]
+ if os.path.isfile(path):
-except IndexError:
+  raise IOError(path + " already exists. If you want to recreate it, please delete it first, or provide a different file name.")
- file = "Database.db"
 
-if os.path.isfile(file):
+ conn = sqlite3.connect(path)
- os.remove(file)
+ cur = conn.cursor()
-
+ 
-conn = sqlite3.connect(file)
+ cur.execute("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));")
-curs = conn.cursor()
+ 
-
+ conn.commit()
-curs.execute("CREATE TABLE tweets(tweet_id numeric, timestamp text, text text);")
+ conn.close()
-
+ 
-conn.commit()
+if __name__ == "__main__":
-conn.close()
+ try:
+  makeDB(sys.argv[1])
+ except IndexError:
+  makeDB()