Skip to content

Commit d46f5ce

Browse files
committed
🐛 fix: logger echo
1 parent 5907585 commit d46f5ce

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

nonebot_plugin_orm/__init__.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -186,12 +186,19 @@ def _init_table():
186186

187187

188188
def _init_logger():
189-
handler = LoguruHandler()
189+
log_level = _driver.config.log_level
190+
if isinstance(log_level, str):
191+
log_level = logging.getLevelName(log_level)
192+
193+
levels = {"alembic": log_level, "sqlalchemy": log_level}
194+
if not plugin_config.sqlalchemy_echo:
195+
levels["sqlalchemy.engine"] = levels["sqlalchemy.pool"] = logging.WARNING
190196

191-
for name in ("sqlalchemy", "alembic"):
197+
handler = LoguruHandler()
198+
for name, level in levels.items():
192199
logger = logging.getLogger(name)
193200
logger.addHandler(handler)
194-
logger.setLevel(logging.DEBUG) # NOTE: loguru will filter by level
201+
logger.setLevel(level)
195202

196203

197204
_init_logger()

0 commit comments

Comments
 (0)