forgot to use the passed-in dburl

archlinuxcn · Sep 9, 2024 · 19534ad · 19534ad
1 parent 53ed81a
commit 19534ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lilac2/db.py b/lilac2/db.py
@@ -23,10 +23,10 @@ def connect_with_schema(schema, dsn):
     cur.execute(f"set search_path to '{schema}'")
   return conn
 
-def setup(engine, schema):
+def setup(dsn, schema):
   global USE, Pool, SCHEMA
   Pool = psycopg2.pool.ThreadedConnectionPool(
-    1, 10, '', partial(connect_with_schema, schema))
+    1, 10, dsn, partial(connect_with_schema, schema))
   USE = True
 
 @contextmanager