allow passing parameters to create engine

graphsense · Dec 17, 2024 · 352b870 · 352b870
1 parent 8c0fc09
commit 352b870
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/tagstore/db/database.py b/src/tagstore/db/database.py
@@ -54,12 +54,12 @@
 ]
 
 
-def get_db_engine(db_url):
-    return create_engine(db_url, echo=False)
+def get_db_engine(db_url, **kwargs):
+    return create_engine(db_url, **kwargs)
 
 
-def get_db_engine_async(db_url):
-    return create_async_engine(db_url, echo=False)
+def get_db_engine_async(db_url, **kwargs):
+    return create_async_engine(db_url, **kwargs)
 
 
 def get_table_ddl_sql():

diff --git a/src/tagstore/db/queries.py b/src/tagstore/db/queries.py
@@ -484,7 +484,7 @@ def __init__(self, engine):
         self.engine = engine
 
     @staticmethod
-    def from_engine(db_url):
+    def from_url(db_url):
         return TagstoreDbAsync(get_db_engine_async(db_url))
 
     # get Tag by id