c2cwsgiutils 6.1.8.dev106__py3-none-any.whl → 6.1.8.dev112__py3-none-any.whl

c2cwsgiutils/db.py

@@ -51,7 +51,7 @@ def setup_session(
     With an accompanying tween that switches between the master and the slave DB
     connection. Uses prefixed entries in the application's settings.
 
-    The slave DB will be used for anything that is GET and OPTIONS queries. The master DB will be used for
+    The slave DB will be used for anything that is GET, HEAD and OPTIONS queries. The master DB will be used for
     all the other queries. You can tweak this behavior with the force_master and force_slave parameters.
     Those parameters are lists of regex that are going to be matched against "{VERB} {PATH}". Warning, the
     path includes the route_prefix.
@@ -107,7 +107,7 @@ def create_session(
     With an accompanying tween that switches between the master and the slave DB
     connection.
 
-    The slave DB will be used for anything that is GET and OPTIONS queries. The master DB will be used for
+    The slave DB will be used for anything that is GET, HEAD and OPTIONS queries. The master DB will be used for
     all the other queries. You can tweak this behavior with the force_master and force_slave parameters.
     Those parameters are lists of regex that are going to be matched against "{VERB} {PATH}". Warning, the
     path includes the route_prefix.
@@ -178,7 +178,10 @@ def _add_tween(
             has_force_master = any(r.match(method_path) for r in master_paths)
             if FORCE_READONLY or (
                 not has_force_master
-                and (request.method in ("GET", "OPTIONS") or any(r.match(method_path) for r in slave_paths))
+                and (
+                    request.method in ("GET", "HEAD", "OPTIONS")
+                    or any(r.match(method_path) for r in slave_paths)
+                )
             ):
                 _LOG.debug(
                     "Using %s database for: %s",
@@ -246,7 +249,7 @@ class SessionFactory(_sessionmaker):
             if FORCE_READONLY or (
                 not has_force_master
                 and (
-                    request.method in ("GET", "OPTIONS")
+                    request.method in ("GET", "OPTIONS", "HEAD")
                     or any(r.match(method_path) for r in self.slave_paths)
                 )
             ):

c2cwsgiutils/stats_pyramid/_db_spy.py

@@ -15,7 +15,7 @@ _LOG = logging.getLogger(__name__)
 _PROMETHEUS_DB_SUMMARY = prometheus_client.Summary(
     prometheus.build_metric_name("database"),
     "Database requests",
-    ["what"],
+    ["what", "engine"],
     unit="seconds",
 )
 
@@ -66,12 +66,14 @@ def _simplify_sql(sql: str) -> str:
     return re.sub(r"%\(\w+\)\w", "?", sql)
 
 
-def _create_sqlalchemy_timer_cb(what: str) -> Callable[..., Any]:
+def _create_sqlalchemy_timer_cb(what: str, engine_name: str | None = None) -> Callable[..., Any]:
     start = time.perf_counter()
 
     def after(*_args: Any, **_kwargs: Any) -> None:
-        _PROMETHEUS_DB_SUMMARY.labels({"query": what}).observe(time.perf_counter() - start)
-        _LOG.debug("Execute statement '%s' in %d.", what, time.perf_counter() - start)
+        elapsed_time = time.perf_counter() - start
+        _PROMETHEUS_DB_SUMMARY.labels(what=what, engine=engine_name).observe(elapsed_time)
+        engine_suffix = f", on {engine_name}" if engine_name else ""
+        _LOG.debug("Execute statement '%s' in %.3f%ss.", what, elapsed_time, engine_suffix)
 
     return after
 
@@ -79,8 +81,12 @@ def _create_sqlalchemy_timer_cb(what: str) -> Callable[..., Any]:
 def _before_cursor_execute(
     conn: Connection, _cursor: Any, statement: str, _parameters: Any, _context: Any, _executemany: Any
 ) -> None:
+    engine_name = getattr(conn.engine, "c2c_name", None)
     sqlalchemy.event.listen(
-        conn, "after_cursor_execute", _create_sqlalchemy_timer_cb(_simplify_sql(statement)), once=True
+        conn,
+        "after_cursor_execute",
+        _create_sqlalchemy_timer_cb(_simplify_sql(statement), engine_name),
+        once=True,
     )
 
 

{c2cwsgiutils-6.1.8.dev106.dist-info → c2cwsgiutils-6.1.8.dev112.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: c2cwsgiutils
-Version: 6.1.8.dev106
+Version: 6.1.8.dev112
 Summary: Common utilities for Camptocamp WSGI applications
 Home-page: https://github.com/camptocamp/c2cwsgiutils
 License: BSD-2-Clause

{c2cwsgiutils-6.1.8.dev106.dist-info → c2cwsgiutils-6.1.8.dev112.dist-info}/RECORD

@@ -16,7 +16,7 @@ c2cwsgiutils/broadcast/utils.py,sha256=0fQZXPu3p_5LEJpGenJwiiMxECQjJhjZBjIkBk8h-
 c2cwsgiutils/client_info.py,sha256=rvXQLyhma0kPoWQINwUzrxfG2gGIjQDNrxPUDRGJxpA,3916
 c2cwsgiutils/config_utils.py,sha256=vkBu-3GQsE94NOBOvT5FE-Ij29EUrKnDsmdUdtu_yzo,1524
 c2cwsgiutils/coverage_setup.py,sha256=BrdjYUmqYl1C-gHuKA7EI5gbQs8Dtdeb2eZKtzr-5L0,909
-c2cwsgiutils/db.py,sha256=JT5F9Dqm2r0Jsh3w3CX79ngAUtakMLpf1secfN1nQnk,16106
+c2cwsgiutils/db.py,sha256=zdlZ81hHCuPCdO7WoX5dM46NjZnvebyl9-DHf6oV8t8,16192
 c2cwsgiutils/db_maintenance_view.py,sha256=58F-p9drkhCI99GoLRPIqT5U-Pm8ckSSUEl-tNxMmjU,3088
 c2cwsgiutils/debug/__init__.py,sha256=GkYNt2fU5PFykw9HmqPEwZrF2mTJumjSu54pp38EhOM,1325
 c2cwsgiutils/debug/_listeners.py,sha256=-fk3KFeB_E4m_YFXJ5MfxuqfA1sWCje9p3FH_93WfXM,4248
@@ -56,12 +56,12 @@ c2cwsgiutils/sqlalchemylogger/handlers.py,sha256=nr9-eQsZ5d0DHbtc4Ym0_faa7qg1dF4
 c2cwsgiutils/static/favicon-16x16.png,sha256=LKk6RFvb3NlPIZdDfAodE8H9IN8KM6CMGnMx4vOHlUQ,887
 c2cwsgiutils/static/favicon-32x32.png,sha256=i4ucx08zAZARd8e7JTMGK-gb5WcOmyuDN6IN4brsEOo,1216
 c2cwsgiutils/stats_pyramid/__init__.py,sha256=alSRhpCa5Kh9JnMnR5XqcMqr5wyL8ogROprrfsIl_qU,786
-c2cwsgiutils/stats_pyramid/_db_spy.py,sha256=A61t6VKIrRRIjbyZTldmAUl_Q3ZDVFYqyxjuntzmllc,2919
+c2cwsgiutils/stats_pyramid/_db_spy.py,sha256=T6q26_6Q-ObjXPNUip5TsX-YkBl11v8vfFL1_9m_dHw,3180
 c2cwsgiutils/stats_pyramid/_pyramid_spy.py,sha256=mRiOmQXV9x8JjkGV4MsaC7sD3qO6dWUTog0bharLLD0,3517
 c2cwsgiutils/templates/index.html.mako,sha256=Yqf6xkq-RtlUnHBAmJFMCJO_XwqGVDQfSjOTx3p70_o,1416
 c2cwsgiutils/version.py,sha256=1ghPu-aKMJdfCSUrxgBENNqNQ-7JMKJr6tS14dDmW4Q,3110
-c2cwsgiutils-6.1.8.dev106.dist-info/LICENSE,sha256=6bEOU0n7ued3SA-DQCsHQaACONMMRzGHmH5XhDVeD-U,1304
-c2cwsgiutils-6.1.8.dev106.dist-info/METADATA,sha256=nBGM4f3aQn2QkNYhg5jSDQB1gvLoWXYrYgZXPyalVG0,34426
-c2cwsgiutils-6.1.8.dev106.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-c2cwsgiutils-6.1.8.dev106.dist-info/entry_points.txt,sha256=ujgqMTL1awN9qDg8WXmrF7m0fgR-hslUM6zKH86pvy0,703
-c2cwsgiutils-6.1.8.dev106.dist-info/RECORD,,
+c2cwsgiutils-6.1.8.dev112.dist-info/LICENSE,sha256=6bEOU0n7ued3SA-DQCsHQaACONMMRzGHmH5XhDVeD-U,1304
+c2cwsgiutils-6.1.8.dev112.dist-info/METADATA,sha256=vndgH0vHo0a41g39rQUEz0GAgXJ_z3yO2XEUxuXZxUM,34426
+c2cwsgiutils-6.1.8.dev112.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+c2cwsgiutils-6.1.8.dev112.dist-info/entry_points.txt,sha256=ujgqMTL1awN9qDg8WXmrF7m0fgR-hslUM6zKH86pvy0,703
+c2cwsgiutils-6.1.8.dev112.dist-info/RECORD,,