From b1cf735bf0038ddddd92d78dd80b0f5dafb93b69 Mon Sep 17 00:00:00 2001
From: Adam Goldsmith <adam@adamgoldsmith.name>
Date: Fri, 1 Jun 2018 13:56:46 -0400
Subject: [PATCH] events: Make it work on multiple doors

---
 events.py | 28 +++++++++++++++++-----------
 1 file changed, 17 insertions(+), 11 deletions(-)

diff --git a/events.py b/events.py
index 6a56b0f..3662c1e 100755
--- a/events.py
+++ b/events.py
@@ -7,11 +7,9 @@ import requests
 
 from common import *
 
-TARGET_IP = '172.18.51.11'
-
-def getStrings():
+def getStrings(targetIP):
     """Parses out the message strings from source."""
-    r = requests.get('https://' + TARGET_IP + '/html/en_EN/en_EN.js',
+    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]+)="([^"]*)')
@@ -59,24 +57,28 @@ def formatMessage(event):
         att['newTime'],
         " IN" if att['readerAddress'] == '0' else " OUT")
 
-def main():
+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(TARGET_IP, ET.tostring(parXMLIn))
+    parXMLOut = doXMLRequest(doorIP, ET.tostring(parXMLIn))
+    ET.dump(parXMLOut)
 
-    root = None
-    if os.path.exists("log.xml"):
+    if os.path.exists("logs/" + doorName + ".xml"):
         # read last log
-        tree = ET.ElementTree(None, "log.xml")
+        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")
@@ -85,7 +87,7 @@ def main():
                           "recordCount": str(recordCount),
                           "historyRecordMarker": parXMLOut[0].attrib["historyRecordMarker"],
                           "historyTimestamp": parXMLOut[0].attrib["historyTimestamp"]})
-    eventsXMLOut = doXMLRequest(TARGET_IP, ET.tostring(eventsXMLIn))
+    eventsXMLOut = doXMLRequest(doorIP, ET.tostring(eventsXMLIn))
     #TODO: handle modeRecords=true
 
     for index, event in enumerate(eventsXMLOut[0]):
@@ -97,7 +99,11 @@ def main():
     else:
         for event in reversed(eventsXMLOut[0]):
             root[0].insert(0, event)
-    tree.write("log.xml")
+    tree.write("logs/" + doorName + ".xml")
+
+def main():
+    for doorName, doorData in config["doors"].items():
+        getMessages(doorName, doorData["ip"])
 
 if __name__ == '__main__':
     main()