From b22ec6f75ae00593ea064a6925b9a7476325e152 Mon Sep 17 00:00:00 2001
From: Mirko Milovanovic <mir_ko@me.com>
Date: Wed, 29 Jan 2025 22:51:19 +0100
Subject: [PATCH] Update main.py to include session ID in stats and modify
 player stats key; update telegraf.conf to add session input

---
 peertube/statnerd/main.py       | 5 +++--
 peertube/statnerd/telegraf.conf | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/peertube/statnerd/main.py b/peertube/statnerd/main.py
index 4182d5a..9b8829d 100644
--- a/peertube/statnerd/main.py
+++ b/peertube/statnerd/main.py
@@ -135,7 +135,7 @@ def downloadStats(driver: webdriver.Chrome, peersDict: dict):
             serverBytes = int(server) * (1024 ** {'B': 0, 'KB': 1, 'MB': 2, 'GB': 3}[serverUnit])
             peerBytes = int(peer) * (1024 ** {'B': 0, 'KB': 1, 'MB': 2, 'GB': 3}[peerUnit])
             
-            playerStats[stat] = {'Server': serverBytes, 'Peer': peerBytes}
+            playerStats[stat] = {'Server': serverBytes, 'Peers': peerBytes}
             
         if 'Buffer State' == stat:
             del(playerStats[stat])
@@ -150,7 +150,8 @@ def downloadStats(driver: webdriver.Chrome, peersDict: dict):
         'player': playerStats,
         'peers': peersDict,
         'url': driver.current_url,
-        'timestamp': int(time.time())
+        'timestamp': int(time.time()),
+        'session': driver.session_id
     }
 
     saveStats([stats])
diff --git a/peertube/statnerd/telegraf.conf b/peertube/statnerd/telegraf.conf
index be6f613..c798052 100644
--- a/peertube/statnerd/telegraf.conf
+++ b/peertube/statnerd/telegraf.conf
@@ -13,6 +13,7 @@
 
         [inputs.socket_listener.xpath.tags]
             url = "url"
+            session = "session"
             #id = ??
             #state = ??