minor refine

dev.py

@@ -2,7 +2,7 @@
 
 import logging
 
-logging.basicConfig(level=logging.DEBUG)
+logging.root.setLevel(logging.DEBUG)
 
 def main():
   from uvicorn import run

src/depends.py

@@ -8,6 +8,7 @@
 from .model import DB_URL, User, UserToken, get_engine
 
 engine = get_engine(DB_URL, settings.echo)
+_migrated = False
 
 @event.listens_for(engine, 'connect')
 def conn_wal_mode(conn, _):
@@ -16,10 +17,17 @@ def conn_wal_mode(conn, _):
 
 @event.listens_for(engine, 'engine_connect')
 def db_migrations(conn, _):
+  global _migrated
+
+  if _migrated:
+    return
+
   from .migration import run_migrations
 
   run_migrations(conn)
 
+  _migrated = True
+
 def db_session():
   with Session(engine) as session:
     yield session

src/migration.py

@@ -43,7 +43,7 @@ def run_migrations(conn):
   while version < LATEST_VERSION:
     logger.info(f'Running migration {version}')
 
-    with context.begin_transaction(_per_migration=True):
+    with context.begin_transaction():
       action = _ACTIONS[version]
       version = action(op) or (version + 1)
 
@@ -53,3 +53,5 @@ def run_migrations(conn):
         )
       )
 
+  context.connection.rollback()
+