From 00d8503847110b673e70842eaa98bf087e7eae88 Mon Sep 17 00:00:00 2001
From: Antoine Van Elstraete <antoine@van-elstraete.net>
Date: Fri, 29 Oct 2021 15:22:15 +0200
Subject: [PATCH] First version

---
 mikrotik_ | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100755 mikrotik_

diff --git a/mikrotik_ b/mikrotik_
new file mode 100755
index 0000000..3563b50
--- /dev/null
+++ b/mikrotik_
@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+
+def config(name, port_max, ports_label):
+    print("graph_category network")
+    print("graph_info Mikrotik network flow")
+    print(f"graph_title Network flow for {name}")
+    print("graph_order down up")
+    print("graph_args --base 1000")
+    print("graph_vlabel bits in (-) / out (+) per ${graph_period}")
+    for port in range(1, port_max + 1):
+        print(f"down{port}.label {ports_label[port - 1]}")
+        print(f"down{port}.type DERIVE")
+        print(f"down{port}.graph no")
+        print(f"down{port}.cdef down,8,*")
+    for port in range(1, port_max + 1):
+        print(f"up{port}.label {ports_label[port - 1]}")
+        print(f"up{port}.type DERIVE")
+        print(f"up{port}.negative down")
+        print(f"up{port}.cdef up,8,*")
+
+
+if __name__ == '__main__':
+    from sys import argv
+    import subprocess
+    ports_label = []
+    InOctets = []
+    OutOctets = []
+    port_max = 0
+    host = argv[0].split("_")[1]
+    command = f"snmpwalk -v 2c -c public {host}"
+    result = subprocess.getoutput(command)
+    for line in result.splitlines():
+        if "sysName" in line:
+            name = line.split()[-1]
+        if "ifDescr" in line:
+            ports_label.append(line.split()[-1])
+        if "ifIndex" in line:
+            port_max += 1
+        if "ifInOctets" in line:
+            InOctets.append(int(line.split()[-1]))
+        if "ifOutOctets" in line:
+            OutOctets.append(int(line.split()[-1]))
+    if 1 < len(argv) <= 2 and argv[1] == "config":
+        config(name, port_max, ports_label)
+    else:
+        for port in range(1, port_max + 1):
+            print(f"down{port}.value {OutOctets[port - 1]}")
+            print(f"up{port}.value {InOctets[port - 1]}")