memberPlumbing/events.py

#!/usr/bin/env python3
from collections import defaultdict
from xml.etree import ElementTree as ET
import os
import re
import requests

from common import *

def getStrings(targetIP):
    """Parses out the message strings from source."""
    r = requests.get('https://' + targetIP + '/html/en_EN/en_EN.js',
                     auth=requests.auth.HTTPDigestAuth(DOOR_USERNAME, DOOR_PASSWORD),
                     verify=False)
    regex = re.compile(r'([0-9]+)="([^"]*)')
    strings = [regex.search(s) for s in r.text.split(';')
               if s.startswith('localeStrings.eventDetails')]
    print({int(g.group(1)): g.group(2) for g in strings})

def getMessages(doorName, doorIP):
    # get parameters for messages to get?
    # honestly not really sure why this is required, their API is confusing
    parXMLIn = ET.Element("VertXMessage")
    ET.SubElement(parXMLIn, "hid:EventMessages", attrib={"action": "LR"})
    parXMLOut = doXMLRequest(doorIP, ET.tostring(parXMLIn))
    ET.dump(parXMLOut)

    if os.path.exists("logs/" + doorName + ".xml"):
        # read last log
        tree = ET.ElementTree(None, "logs/" + doorName + ".xml")
        root = tree.getroot()
        recordCount = int(parXMLOut[0].attrib["historyRecordMarker"]) - \
                      int(root[0][0].attrib["recordMarker"])
    else:
        # first run for this door
        root = None
        recordCount = 1000

    if recordCount == 0:
        print("No records to get!")
        return
    print("Getting", recordCount, "records")
    # get the actual messages
    eventsXMLIn = ET.Element("VertXMessage")
    ET.SubElement(eventsXMLIn, "hid:EventMessages",
                  attrib={"action": "LR",
                          "recordCount": str(recordCount),
                          "historyRecordMarker": parXMLOut[0].attrib["historyRecordMarker"],
                          "historyTimestamp": parXMLOut[0].attrib["historyTimestamp"]})
    eventsXMLOut = doXMLRequest(doorIP, ET.tostring(eventsXMLIn))
    #TODO: handle modeRecords=true

    for index, event in enumerate(eventsXMLOut[0]):
        event.attrib["recordMarker"] = str(int(parXMLOut[0].attrib["historyRecordMarker"]) - index)

    if root is None:
        tree = ET.ElementTree(eventsXMLOut)
    else:
        for event in reversed(eventsXMLOut[0]):
            root[0].insert(0, event)
    tree.write("logs/" + doorName + ".xml")

def main():
    for doorName, doorData in config["doors"].items():
        getMessages(doorName, doorData["ip"])

if __name__ == '__main__':
    main()