diff options
Diffstat (limited to 'dedup/utils.py')
| -rw-r--r-- | dedup/utils.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/dedup/utils.py b/dedup/utils.py index e163a00..1452edc 100644 --- a/dedup/utils.py +++ b/dedup/utils.py @@ -22,3 +22,13 @@ def sqlite_add_version_compare(engine): def add_version_compare(connection, _): connection.create_collation("debian_version", version_compare) connection.create_function("debian_version_compare", 2, version_compare) + +def postgres_enable_result_streaming(engine): + if engine.dialect.driver != "psycopg2": + return + engine.server_side_cursors = True + +def configure_database_engine(engine): + enable_sqlite_foreign_keys(engine) + sqlite_add_version_compare(engine) + postgres_enable_result_streaming(engine) |