From 0b45c5f4bfc4a6e3e11843f6040e094a11c17299 Mon Sep 17 00:00:00 2001
From: Helmut Grohne <helmut@subdivi.de>
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.
---
 dedup/utils.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'dedup')

diff --git a/dedup/utils.py b/dedup/utils.py
index 2fae9fd..e4d1c10 100644
--- a/dedup/utils.py
+++ b/dedup/utils.py
@@ -1,3 +1,5 @@
+import sqlalchemy.event
+
 def fetchiter(cursor):
     rows = cursor.fetchmany()
     while rows:
@@ -5,3 +7,8 @@ def fetchiter(cursor):
             yield row
         rows = cursor.fetchmany()
 
+def enable_sqlite_foreign_keys(engine):
+    @sqlalchemy.event.listens_for(engine, "connect")
+    def pragma_foreign_keys(connection, _):
+        connection.execute("PRAGMA foreign_keys=ON;")
+
-- 
cgit v1.2.3