From de68e2ba2b8ca62f751696a1cdb1a002ca7b0049 Mon Sep 17 00:00:00 2001 From: Friedrich Lindenberg Date: Sat, 14 Nov 2020 22:48:56 +0100 Subject: [PATCH] Re-org a little bit :) --- dataset/database.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/dataset/database.py b/dataset/database.py index 305c89c..17de461 100644 --- a/dataset/database.py +++ b/dataset/database.py @@ -56,6 +56,8 @@ class Database(object): self.schema = schema self.engine = create_engine(url, **engine_kwargs) + self.is_postgres = self.engine.dialect.name == "postgresql" + self.is_sqlite = self.engine.dialect.name == "sqlite" def _enable_sqlite_wal_mode(dbapi_con, con_record): # reference: @@ -63,11 +65,10 @@ class Database(object): # https://stackoverflow.com/a/7831210/1890086 dbapi_con.execute("PRAGMA journal_mode=WAL") - if parsed_url.scheme.lower() == 'sqlite' and parsed_url.path != '' and sqlite_wal_mode: + if self.is_sqlite and parsed_url.path != "" and sqlite_wal_mode: # we only enable WAL mode for sqlite databases that are not in-memory - event.listen(self.engine, 'connect', _enable_sqlite_wal_mode) + event.listen(self.engine, "connect", _enable_sqlite_wal_mode) - self.is_postgres = self.engine.dialect.name == "postgresql" self.types = Types(is_postgres=self.is_postgres) self.url = url self.row_type = row_type