max_liu 1 день назад
Родитель
Сommit
a87ca59781
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      src/models/database.py

+ 2 - 1
src/models/database.py

@@ -68,7 +68,8 @@ def get_engine():
             pool_pre_ping=True,  # 连接前检查连接是否有效
             echo=False,  # 设置为 True 可以打印 SQL 语句,用于调试
         )
-        logger.info(f"Database engine created for database: {os.getenv('DB_NAME', database)}")
+        db_name = database_url.rsplit('/', 1)[-1].split('?')[0]
+        logger.info(f"Database engine created for database: {db_name}")
     return _engine