stats: add grafana stat key names for behavior stats (#740)

Rhizome-Conifer · Aug 20, 2019 · b9d8fe7 · b9d8fe7
1 parent 24e0a3c
commit b9d8fe7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/webrecorder/webrecorder/admincontroller.py b/webrecorder/webrecorder/admincontroller.py
@@ -119,6 +119,15 @@ def init_all_stats(self):
             name = 'Sources ' + key[len(Stats.SOURCES_KEY.format('')):]
             self.all_stats[name] = key
 
+        for key in self.redis.scan_iter(Stats.BEHAVIOR_KEY.format(stat='*', name='*')):
+            parts = key.split(':')
+            if len(parts) < 4:
+                continue
+            name = parts[3].replace('Behavior', '')
+            name = sub("([a-z])([A-Z])","\g<1>-\g<2>", name)
+            full_name = 'Behavior ' + parts[2] + ' ' + name
+            self.all_stats[full_name.title()] = key
+
         for key in self.CUSTOM_STATS:
             self.all_stats[key] = key