Add old, non functional, xml based thing for reference
This commit is contained in:
parent
73600db7f6
commit
bd8376e0c3
111
old_xml_based.py
Normal file
111
old_xml_based.py
Normal file
@ -0,0 +1,111 @@
|
||||
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"]
|
||||
|
||||
ET.register_namespace("hid", "http://www.hidglobal.com/VertX")
|
||||
ET.register_namespace("hid", "http://www.hidcorp.com/VertX")
|
||||
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()
|
Reference in New Issue
Block a user