From 0b45c5f4bfc4a6e3e11843f6040e094a11c17299 Mon Sep 17 00:00:00 2001 From: Helmut Grohne Date: Sun, 23 Jun 2013 13:30:04 +0200 Subject: dedup.utils: add enbale_sqlite_foreign_keys helper Makes usage of sqlalchemy easier, cause I can invoke it once and it works for all connections. --- update_sharing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'update_sharing.py') diff --git a/update_sharing.py b/update_sharing.py index bbd19e5..e78e13f 100755 --- a/update_sharing.py +++ b/update_sharing.py @@ -2,7 +2,7 @@ import sqlalchemy -from dedup.utils import fetchiter +from dedup.utils import fetchiter, enable_sqlite_foreign_keys def add_values(conn, insert_key, files, size): rows = conn.execute("UPDATE sharing SET files = files + ?, size = size + ? WHERE package1 = ? AND package2 = ? AND func1 = ? AND func2 = ?;", @@ -39,8 +39,8 @@ def process_pkgdict(conn, pkgdict): def main(): db = sqlalchemy.create_engine("sqlite:///test.sqlite3") + enable_sqlite_foreign_keys(db) with db.begin() as conn: - conn.execute("PRAGMA foreign_keys = ON;") conn.execute("DELETE FROM sharing;") conn.execute("DELETE FROM duplicate;") readcur = conn.execute("SELECT hash FROM hash GROUP BY hash HAVING count(*) > 1;") -- cgit v1.2.3