working daemon

aaisp-to-mqtt.py

@@ -1,33 +1,113 @@
 #!/usr/bin/env python
 
-import os
 import sys
 import logging
 import json
 import urllib
 import configparser
+import paho.mqtt.client as mqtt
+import humanfriendly
 
 LOG = logging.getLogger(__name__)
 VERSION = 0.1
 
+
 def main():
     logging.basicConfig(level=logging.INFO, format='%(levelname)8s [%(asctime)s] %(message)s')
 
     if len(sys.argv) != 2:
         LOG.fatal("Config file not supplied")
-
+        sys.exit(1)
     cfgfile = sys.argv[1]
-
+    # load the config
     config = configparser.ConfigParser()
     config.read(cfgfile)
 
+    # check it has the correct sections
     for section in ["aaisp", "mqtt"]:
         if section not in config.sections():
             LOG.fatal("%s section not found in config file %s", section, cfgfile)
 
+    aaisp_username = config.get("aaisp", "username")
+    aaisp_password = config.get("aaisp", "password")
+
+    # attempt to get details from aaisp
+    LOG.info("Connecting to AAISP CHAOSv2 endpoint")
+    post_params = "control_login=%s&control_password=%s" % (aaisp_username, aaisp_password)
+    url = "https://chaos2.aa.net.uk/broadband/info"
+    response = urllib.urlopen(url, data=post_params)
+    data = json.loads(response.read())
+    if "info" not in data:
+        LOG.fatal("info section not found in AAISP CHAOSv2 response")
+        sys.exit(1)
+    circuits = data["info"]
+    LOG.info("Got %s circuits", len(circuits))
+    if len(circuits) == 0:
+        LOG.fatal("No circuits returned from AAISP CHAOSv2")
+
+    # work out unique line IDs and logins
+    logins = []
+    lines = []
+    for circuit in circuits:
+        if circuit["login"] not in logins:
+            logins.append(circuit["login"])
+        if circuit["ID"] not in lines:
+            lines.append(circuit["ID"])
+    LOG.info("* Lines: %s", ', '.join(lines))
+    LOG.info("* Logins: %s", ', '.join(logins))
+
+
+    # get MQTT config
+    mqtt_broker = config.get("mqtt", "broker")
+    mqtt_port = int(config.get("mqtt", "port"))
+    mqtt_username = config.get("mqtt", "username")
+    mqtt_password = config.get("mqtt", "password")
+    mqtt_topic_prefix = config.get("mqtt", "topic_prefix")
+    # connect to the broker
+    LOG.info("Connecting to MQTT broker %s:%s", mqtt_broker, mqtt_port)
+    client = mqtt.Client()
+    # do auth?
+    if mqtt_username is not None and mqtt_password is not None:
+        client.username_pw_set(mqtt_username, mqtt_password)
+    client.connect(mqtt_broker, mqtt_port, 60)
+    LOG.info("Connected OK to MQTT")
+
+    # version and indexes
+    publish(client=client, topic="%s/$version" % (mqtt_topic_prefix), payload=VERSION)
+    publish(client=client, topic="%s/$lines" % (mqtt_topic_prefix), payload=','.join(lines))
+    publish(client=client, topic="%s/$logins" % (mqtt_topic_prefix), payload=','.join(logins))
+    LOG.info("Published version and index messages")
+
+    # publish per circuit
+    for circuit in circuits:
+        publish_per_circuit(client=client, circuit=circuit, mqtt_topic_prefix=mqtt_topic_prefix)
+    LOG.info("Published details for %s circuits", len(circuits))
+    # disconnect
+    LOG.info("Disconnecting from MQTT")
+    client.disconnect()
 
 
     sys.exit(0)
 
+def publish_per_circuit(client, circuit, mqtt_topic_prefix):
+    line_prefix = "%s/line/%s" % (mqtt_topic_prefix, circuit["ID"])
+    login_prefix = "%s/login/%s" % (mqtt_topic_prefix, circuit["login"])
+    for prefix in [line_prefix, login_prefix]:
+        for metric in [
+            ("quota/remaining", int(circuit["quota_remaining"])),
+            ("quota/monthly", int(circuit["quota_monthly"])),
+            ("syncrate/up", int(circuit["rx_rate"])),
+            ("syncrate/down", int(circuit["tx_rate"])),
+            ("postcode", str(circuit["postcode"].strip()))
+        ]:
+            topic = "%s/%s" % (prefix, metric[0])
+            publish(client=client, topic=topic, payload=metric[1])
+            if type(metric[1]) == int:
+                publish(client=client, topic="%s/human" % (topic), payload=humanfriendly.format_size(metric[1]))
+    return
+
+def publish(client, topic, payload):
+    client.publish(topic=topic, payload=payload)
+
 if __name__ == "__main__":
     main()