From 1883fa85673c84bad91ccc720cf8618ad26cd652 Mon Sep 17 00:00:00 2001 From: Brendan Reilly Date: Mon, 9 Aug 2021 14:49:34 -0400 Subject: [PATCH] Update handler name used by SQLAlchemy --- module_build_service/common/monitor.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/module_build_service/common/monitor.py b/module_build_service/common/monitor.py index 77350489..add69b9c 100644 --- a/module_build_service/common/monitor.py +++ b/module_build_service/common/monitor.py @@ -15,6 +15,7 @@ from prometheus_client import ( # noqa: F401 Histogram, start_http_server, ) +import sqlalchemy from sqlalchemy import event @@ -85,9 +86,11 @@ def db_hook_event_listeners(target=None): if not target: target = db.engine - @event.listens_for(target, "dbapi_error", named=True) - def receive_dbapi_error(**kw): - db_dbapi_error_counter.inc() + if sqlalchemy.__version__[0] == "0": + # Deprecated since 0.9 + @event.listens_for(target, "dbapi_error", named=True) + def receive_dbapi_error(**kw): + db_dbapi_error_counter.inc() @event.listens_for(target, "engine_connect") def receive_engine_connect(conn, branch):