restrict sqlite-specific configuration to sqlite databases

1 files changed, 4 insertions, 0 deletions

diff --git a/dedup/utils.py b/dedup/utils.py
index fd30378..e163a00 100644
--- a/dedup/utils.py
+++ b/dedup/utils.py
@@ -9,11 +9,15 @@ def fetchiter(cursor):
         rows = cursor.fetchmany(1024)
 
 def enable_sqlite_foreign_keys(engine):
+    if engine.dialect.name != "sqlite":
+        return
     @sqlalchemy.event.listens_for(engine, "connect")
     def pragma_foreign_keys(connection, _):
         connection.execute("PRAGMA foreign_keys=ON;")
 
 def sqlite_add_version_compare(engine):
+    if engine.dialect.name != "sqlite":
+        return
     @sqlalchemy.event.listens_for(engine, "connect")
     def add_version_compare(connection, _):
         connection.create_collation("debian_version", version_compare)