Updates

ETB-API/venv/lib/python3.12/site-packages/django_prometheus/models.py

@@ -0,0 +1,54 @@
+from prometheus_client import Counter
+
+from django_prometheus.conf import NAMESPACE
+
+model_inserts = Counter(
+    "django_model_inserts_total",
+    "Number of insert operations by model.",
+    ["model"],
+    namespace=NAMESPACE,
+)
+
+model_updates = Counter(
+    "django_model_updates_total",
+    "Number of update operations by model.",
+    ["model"],
+    namespace=NAMESPACE,
+)
+
+model_deletes = Counter(
+    "django_model_deletes_total",
+    "Number of delete operations by model.",
+    ["model"],
+    namespace=NAMESPACE,
+)
+
+
+def ExportModelOperationsMixin(model_name):
+    """Returns a mixin for models to export counters for lifecycle operations.
+
+    Usage:
+      class User(ExportModelOperationsMixin('user'), Model):
+          ...
+    """
+    # Force create the labels for this model in the counters. This
+    # is not necessary but it avoids gaps in the aggregated data.
+    model_inserts.labels(model_name)
+    model_updates.labels(model_name)
+    model_deletes.labels(model_name)
+
+    class Mixin:
+        def _do_insert(self, *args, **kwargs):
+            model_inserts.labels(model_name).inc()
+            return super()._do_insert(*args, **kwargs)
+
+        def _do_update(self, *args, **kwargs):
+            model_updates.labels(model_name).inc()
+            return super()._do_update(*args, **kwargs)
+
+        def delete(self, *args, **kwargs):
+            model_deletes.labels(model_name).inc()
+            return super().delete(*args, **kwargs)
+
+    Mixin.__qualname__ = f"ExportModelOperationsMixin('{model_name}')"
+    return Mixin