tracker.jordan.im: usage graphs

1 files changed, 18 insertions, 2 deletions

diff --git a/tracker.jordan.im/update-stats.py b/tracker.jordan.im/update-stats.py
index bc370b2..f0a1ec7 100755
--- a/tracker.jordan.im/update-stats.py
+++ b/tracker.jordan.im/update-stats.py
@@ -1,10 +1,12 @@
 #!/usr/bin/env python3
 
 import fileinput
+import os
+import shutil
 import sys
 import urllib.request
 
-INDEX_PATH = '/var/www/html/tracker.jordan.im/index.html'
+WWW = '/var/www/html/tracker.jordan.im'
 
 stats = dict.fromkeys(['torrents', 'connections', 'uptime'])
 
@@ -30,10 +32,24 @@ s = (
     f'with {stats["connections"]} connections per second.\n'
 )
 
-for line in fileinput.input([INDEX_PATH], inplace=True):
+for line in fileinput.input([os.path.join(WWW, 'index.html')], inplace=True):
     if line.strip().startswith('The tracker has been'):
         line = s
     sys.stdout.write(line)
 
+cpu = 'https://service.greenhost.net/cloud/Instance/graph?id=7615&template=cpu&timeframe=24'
+conn = urllib.request.Request(cpu)
+conn.add_header('Cookie', 'PHPSESSID=XXXXXXXXXXXXXXXXXXXXXXXXXX')
+
+with urllib.request.urlopen(conn) as r, open(os.path.join(WWW, 'cpu.png'), 'wb') as out:
+        out.write(r.read())
+
+net = 'https://service.greenhost.net/cloud/Instance/graph?id=7615&template=net1&timeframe=24'
+conn = urllib.request.Request(net)
+conn.add_header('Cookie', 'PHPSESSID=XXXXXXXXXXXXXXXXXXXXXXXXXX')
+
+with urllib.request.urlopen(conn) as r, open(os.path.join(WWW, 'net.png'), 'wb') as out:
+        out.write(r.read())
+
 #import code
 #code.interact(local=locals())