2 files changed, 10 insertions, 9 deletions

diff --git a/overlays/cake-prometheus-exporter/cake-prometheus-exporter.py b/overlays/cake-prometheus-exporter/cake-prometheus-exporter.py
index be5202d0..1be9525c 100644
--- a/overlays/cake-prometheus-exporter/cake-prometheus-exporter.py
+++ b/overlays/cake-prometheus-exporter/cake-prometheus-exporter.py
@@ -63,15 +63,15 @@ class CAKEMetrics:
             for tin_name, tin_data in zip(tin_names, qdisc['tins']):
                 tins[tin_name] = {
                     'bytes': tin_data['sent_bytes'],
-                    'packets': tin_data['sent_packets']
+                    'packets': tin_data['sent_packets'],
+                    'ecn_mark': tin_data['ecn_mark'],
+                    'drops': tin_data['drops'],
                 }
 
             attrs[qdisc['dev']] = {
-                'bytes': qdisc['bytes'],
-                'packets': qdisc['packets'],
-                'drops': qdisc['drops'],
                 'overlimits': qdisc['overlimits'],
                 'requeues': qdisc['requeues'],
+                'bandwidth': qdisc['options']['bandwidth'],
                 'tins': tins
             }
 
@@ -83,12 +83,13 @@ class CAKEMetrics:
     def prometheus(self):
         metrics = ''
 
-        metrics += _format_prom_metrics('cake_bytes', 'counter', [({'dev': dev}, self.attrs[dev]['bytes']) for dev in self.attrs])
-        for packet_counter in ['packets', 'overlimits', 'requeues', 'drops']:
+        metrics += _format_prom_metrics('cake_bandwidth', 'gauge', [({'dev': dev}, self.attrs[dev]['bandwidth']) for dev in self.attrs])
+
+        for packet_counter in ['overlimits', 'requeues']:
             metrics += _format_prom_metrics(f'cake_{packet_counter}', 'counter', [({'dev': dev}, self.attrs[dev][packet_counter]) for dev in self.attrs])
 
-        metrics += _format_prom_metrics('cake_tin_bytes', 'counter', [({'dev': dev, 'tin': tin}, self.attrs[dev]['tins'][tin]['bytes']) for dev in self.attrs for tin in self.attrs[dev]['tins']])
-        metrics += _format_prom_metrics('cake_tin_packets', 'counter', [({'dev': dev, 'tin': tin}, self.attrs[dev]['tins'][tin]['packets']) for dev in self.attrs for tin in self.attrs[dev]['tins']])
+        for packet_counter in ['bytes', 'packets', 'drops', 'ecn_mark']:
+            metrics += _format_prom_metrics(f'cake_{packet_counter}', 'counter', [({'dev': dev, 'tin': tin}, self.attrs[dev]['tins'][tin][packet_counter]) for dev in self.attrs for tin in self.attrs[dev]['tins']])
 
         return metrics.encode('utf-8')
 
diff --git a/overlays/cake-prometheus-exporter/default.nix b/overlays/cake-prometheus-exporter/default.nix
index 69a5008c..a1c62a88 100644
--- a/overlays/cake-prometheus-exporter/default.nix
+++ b/overlays/cake-prometheus-exporter/default.nix
@@ -1,6 +1,6 @@
 { final, prev, ... }:
 let
-  inpPython = final.python310.override {};
+  inpPython = final.python3.override {};
   python = inpPython.withPackages (ps: with ps; []);
 in {
   cake-prometheus-exporter = prev.stdenv.mkDerivation rec {