clean up scripts

.config/polybar/scripts/network-openwrt-status.py

@@ -0,0 +1,112 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*- # PEP 263
+
+import requests
+from binascii import unhexlify
+
+# Configuration
+host = '192.168.8.1'
+username = 'root'
+password = ''
+
+LOGIN_URL = f"http://{host}/cgi-bin/luci/"
+MODEM_URL = f"http://{host}/cgi-bin/luci/?status=1"
+SMS_URL = f"http://{host}/cgi-bin/luci/admin/services/modem_read_sms/all?block=0"
+BW_URL = f"http://{host}/cgi-bin/luci/admin/status/realtime/bandwidth_status/wwan0"
+
+
+def scale_units(rate):
+    # input: integer
+    decimal_places = 2
+    for unit in ['bps', 'Kbps', 'Mbps', 'Gbps', 'Tbps']:
+        if rate < 1000.0:
+            if unit == 'bps':
+                decimal_places = 0
+            break
+        rate /= 1000.0
+    return f"{rate:.{decimal_places}f}{unit}"
+
+
+def sms_concatenate(data):
+    # input: list of dicts
+    sms = list()
+    concat = dict()
+    for item in data:
+        if not item['fail']:
+            if 'c_ref' in item:
+                if item['c_ref'] not in concat:
+                    concat[item['c_ref']] = list()
+                concat[item['c_ref']].append(item)
+            else:
+                sms.append(item)
+    if len(concat):
+        for key, parts in concat.items():
+            parts.sort(key=lambda x: x['c_cur'])
+            for part in parts:
+                if part['c_cur'] == 1:
+                    sms.append(part)
+                else:
+                    sms[-1]['data'] += part['data']
+    return sms
+
+
+def sms_decode(sms):
+    # input: dict
+    if sms['dcs'] == 2:
+        sms['data'] = unhexlify(sms['data']).decode('utf-16be')
+    return sms
+
+
+if __name__ == '__main__':
+    # setup session and authenticate
+    s = requests.Session()
+    s.post(LOGIN_URL, data={'luci_username': username, 'luci_password': password})
+    s.headers.update({'Content-Type': 'application/json'})
+
+    # get lte modem stats (luci-mod-microdrive)
+    r = s.get(MODEM_URL)
+    r = r.json()
+    name = r['ifg']['spn']
+    lvl = r['ifg']['rssi_lv']
+    rssi = r['ifg']['cell_lte'][0]['rssi']
+    rsrp = r['ifg']['cell_lte'][0]['rsrp']
+    sinr = r['ifg']['cell_lte'][0]['sinr']
+    result = f"{name}  {lvl}% RSSI {rssi}dBm RSRP {rsrp}dBm SINR {sinr}dB"
+
+    # get sms count (luci-mod-microdrive)
+    r = s.get(SMS_URL)
+    r = r.json()
+    mem_total = 255
+    mem_used = 0
+    sms_count = 0
+    if r['code'] == 0 and 'out' in r and 'sms' in r['out']:
+        mem_total = r['out']['mem']['total']
+        mem_used = r['out']['mem']['used']
+        data = r['out']['sms']
+        if len(data):
+            sms = sms_concatenate(data)
+            sms_count = len(sms)
+    icon = '' if sms_count else ''
+    # if mem_used > mem_total * 0.90:
+    #    sms_count = 'MANY'
+    result = result + f" {icon} {sms_count}"
+
+    # get traffic stats
+    # the response is array of array [[TIME, RXB, RXP, TXB, TXP]]
+    # we only need two last items with TIME, RXB and TXB fields
+    r = s.get(BW_URL)
+    r = r.json()
+    rxb = 0
+    txb = 0
+    if len(r) >= 2:
+        # get last two items
+        a = r.pop()
+        b = r.pop()
+        # normalize difference against time interval
+        # this code from luci XHR.poll function
+        time_delta = a[0] - b[0]
+        rxb = scale_units((a[1] - b[1]) / time_delta * 8)
+        txb = scale_units((a[3] - b[3]) / time_delta * 8)
+    result = result + f" {rxb} {txb}"
+
+    print(result)