add container to testdata.py

core/core/metrics.py

@@ -2,6 +2,10 @@ from django.apps import apps
 from prometheus_client.core import CounterMetricFamily, REGISTRY
 from django.db.models import Case, Value, When, BooleanField, Count
 from inventory.models import Item
+from tickets.models import IssueThread
+
+from itertools import groupby
+
 
 class ItemCountCollector(object):
 
@@ -15,16 +19,16 @@ class ItemCountCollector(object):
 
         queryset = (
             Item.all_objects
-                .annotate(
-                    returned=Case(
-                        When(returned_at__isnull=True, then=Value(False)),
-                        default=Value(True),
-                        output_field=BooleanField()
-                    )
+            .annotate(
+                returned=Case(
+                    When(returned_at__isnull=True, then=Value(False)),
+                    default=Value(True),
+                    output_field=BooleanField()
                 )
-                .values('event__slug', 'returned', 'event_id')
-                .annotate(amount=Count('id'))
-                .order_by('event__slug', 'returned')  # Optional: order by slug and returned
+            )
+            .values('event__slug', 'returned', 'event_id')
+            .annotate(amount=Count('id'))
+            .order_by('event__slug', 'returned')  # Optional: order by slug and returned
         )
 
         for e in queryset:
@@ -32,4 +36,25 @@ class ItemCountCollector(object):
 
         yield counter
 
+
+class TicketCountCollector(object):
+
+    def collect(self):
+        counter = CounterMetricFamily("c3lf_ticket_count", "Current number of tickets", labels=['event', 'event_id'])
+
+        yield counter
+
+        if not apps.models_ready or not apps.apps_ready:
+            return
+
+        g = groupby(IssueThread.objects.all().prefetch_related('event', 'state_changes'),
+                    lambda x: (x.event.slug, x.state))
+        for (event, state), v in g:
+            counter.add_metric([event.lower(), state.lower()], len(list(v)))
+
+        yield counter
+
+
+
 REGISTRY.register(ItemCountCollector())
+REGISTRY.register(TicketCountCollector())

deploy/dev/docker-compose.yml

@@ -7,6 +7,7 @@ services:
     environment:
       - HTTP_HOST=core
       - DB_FILE=dev.db
+      - DEBUG_MODE_ACTIVE=yup
     volumes:
       - ../../core:/code
       - ../testdata.py:/code/testdata.py

deploy/testdata.py

@@ -70,6 +70,15 @@ def setup():
         issue_thread=issue_thread,
     )[0]
 
+    from inventory.models import Container
+
+    Container.objects.get_or_create(
+        id=1,
+        name='testcontainer'
+    )[0]
+
+
+
 
 def main():
     os.environ.setdefault("DJANGO_SETTINGS_MODULE", "core.settings")

deploy/testing/docker-compose.yml

@@ -30,6 +30,7 @@ services:
       - DB_USER=system3
       - DB_PASSWORD=system3
       - MAIL_DOMAIN=mail:1025
+      - DEBUG_MODE_ACTIVE=certainly
     volumes:
       - ../../core:/code
       - ../testdata.py:/code/testdata.py