This repository has been archived on 2024-02-23. You can view files and clone it, but cannot push or open issues or pull requests.
memberPlumbing/old_xml_based.py

111 lines
3.7 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import requests
from xml.etree import ElementTree as ET
import csv
from passwords import *
#credentialRanges = {"6": {"min": "20176", "max": "20350"}}
memberLevels = ["CMS Staff",
"CMS Weekends Only",
"CMS Weekdays Only",
"CMS Unlimited",
"CMS Nights & Weekends"]
XML = ET.Element("VertXMessage")
#TODO: both those might need more stuff:
# recordOffset="0" recordCount="4" moreRecords="false"
cardholders = ET.SubElement(XML, "hid:Cardholders",
attrib={"action": "AD", "recordOffset": "0"})
credentials = ET.SubElement(XML, "hid:Credentials", attrib={"action": "AD"})
def doRequest(xml):
return requests.get(
'https://172.18.51.11/cgi-bin/vertx_xml.cgi',
params={'XML': b'<?xml version="1.0" encoding="UTF-8"?>' + ET.tostring(xml)},
auth=requests.auth.HTTPDigestAuth(DOOR_USERNAME, DOOR_PASSWORD),
verify=False)
def makeRoleSet(roleSetID, scheduleID):
roleSet = ET.SubElement(XML, "hid:RoleSet", attrib={"action": "UD",
"roleSetID": str(roleSetID)})
roles = ET.SubElement(roleSet, "hid:Roles")
ET.SubElement(roles, "hid:Role", attrib={"roleID": str(roleSetID),
"scheduleID": str(scheduleID),
"resourceID": "0"})
def makeCardHolder(id, fname, lname):
attrib={"cardhlderID": str(id),
"forename": fname,
"surname": lname,
# "email": "", #TODO
# "phone": "", #TODO
"roleSetID": str(id)}
ET.SubElement(cardholders, "hid:Cardholder", attrib=attrib)
def makeCredential(cardNum, cardHolderID):
ET.SubElement(credentials, "hid:Credential",
attrib={"isCard": "true", #TODO: needed?
"cardNumber": str(cardNum),
"cardholderID": str(cardHolderID),
"formatID": "6"})
def handleRow(index, row):
makeCardHolder(index, row["First Name"], row["Last Name"])
memberLevel = [roleSetID for roleSetID, name in enumerate(memberLevels, 1)
if row[name] != ""]
if len(memberLevel) == 1:
makeRoleSet(index, memberLevel[0])
else:
print(row["First Name"], row["Last Name"], "has no/too many member levels!")
if row["Access Card Number"] != "":
makeCredential(row["Access Card Number"], index)
def deleteStuff():
delXML = ET.Element("VertXMessage")
queryXML = ET.Element("VertXMessage")
ET.SubElement(queryXML, "hid:Credentials", attrib={"action": "LR"})
r = doRequest(queryXML)
respXML = ET.XML(r.text)
for cred in respXML[0]:
ET.SubElement(delXML, "hid:Credentials",
attrib={"action": "DD",
"isCard": "true",
"rawCardNumber": cred.attrib["rawCardNumber"]})
for ii in range(1, 300):
ET.SubElement(delXML, "hid:Cardholders",
attrib={"action": "DD", "cardholderID": str(ii)})
r = doRequest(delXML)
print(r.text)
def main():
deleteStuff()
# Include schedules.xml fragment
# with open("schedules.xml") as f:
# root = ET.XML(f.read())
# for e in root:
# XML.append(e)
with open("export-16.csv") as f:
reader = csv.DictReader(f)
for index, row in enumerate(reader, 1):
handleRow(index, row)
with open("/home/adam/scratch/test.xml", "wb") as f:
f.write(ET.tostring(XML))
r = doRequest(XML)
print(r.status_code, r.text)
# if __name__ == '__main__':
# main()
main()