From 558ffdc03334e43262ce9aae5078a8268869f9f2 Mon Sep 17 00:00:00 2001 From: Ben Kurtovic Date: Thu, 4 Sep 2014 23:14:22 -0500 Subject: [PATCH] Cleaner solution. --- tasks/afc_copyvios.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tasks/afc_copyvios.py b/tasks/afc_copyvios.py index 4e6880a..b5ba5bb 100644 --- a/tasks/afc_copyvios.py +++ b/tasks/afc_copyvios.py @@ -206,13 +206,12 @@ class AFCCopyvios(Task): query1 = "DELETE FROM cache WHERE cache_id = ?" query2 = "INSERT INTO cache VALUES (?, DEFAULT, ?, ?)" query3 = "INSERT INTO cache_data VALUES (DEFAULT, ?, ?, ?, ?)" - cache_id = sha256("1:1:" + page.get().encode("utf8")).digest() - buff = buffer(cache_id) - data = [(buff, source.url, source.confidence, source.skipped) + cache_id = buffer(sha256("1:1:" + page.get().encode("utf8")).digest()) + data = [(cache_id, source.url, source.confidence, source.skipped) for source in result.sources] with self.conn.cursor() as cursor: cursor.execute("START TRANSACTION") - cursor.execute(query1, (buff,)) - cursor.execute(query2, (buff, result.queries, result.time)) + cursor.execute(query1, (cache_id,)) + cursor.execute(query2, (cache_id, result.queries, result.time)) cursor.executemany(query3, data) cursor.execute("COMMIT")