From 4f1cb5cad15c380b34084e7ffaadeb59064db4f0 Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Mon, 18 May 2015 13:41:30 +0200 Subject: [PATCH] Get csvdb running (inital version) --- csvdb.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/csvdb.py b/csvdb.py index 17453e3..e613942 100755 --- a/csvdb.py +++ b/csvdb.py @@ -1,17 +1,15 @@ #!/usr/bin/env python3 -print("This script is not working yet... If you are trying to test it, please delete lines 3 to 5 of this file.") -import sys -sys.exit(0) +import tools import sqlite3, csv try: infile = open('tweets.csv') except IOError: - print("Please make sure that the tweets.csv from the Twitter download is located in this directory.") + raise IOError("Please make sure that the tweets.csv from the Twitter download is located in this directory.") -input = csv.reader(infile) +input = list(csv.reader(infile)) conn = sqlite3.connect('Database.db') cur = conn.cursor() @@ -19,7 +17,9 @@ cur = conn.cursor() try: 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));") except sqlite3.OperationalError: - print("Database.db already exists. Please delete it before trying to create a new one.") + raise IOError("Database.db already exists. Please delete it before trying to create a new one.") for row in input[1:]: - cur.execute("INSERT INTO tweets VALUES(?,?,?,?,?,?,?,?,?,?);",row) + cur.execute("INSERT INTO tweets VALUES(" + row[0].replace("'","''") + ",'" + row[1].replace("'","''") + "','" + row[2].replace("'","''") + "','" + row[3].replace("'","''") + "','" + row[4].replace("'","''") + "','" + row[5].replace("'","''") + "','" + row[6].replace("'","''") + "','" + row[7].replace("'","''") + "','" + row[8].replace("'","''") + "','" + row[9].replace("'","''") + "');") + +conn.commit()