forked from CMS/memberPlumbing
132 lines
4.5 KiB
Python
132 lines
4.5 KiB
Python
import requests
|
|
import csv
|
|
from io import StringIO
|
|
import urllib3
|
|
|
|
from passwords import *
|
|
|
|
# it's fine, ssl certs are for losers anyway
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
|
|
fieldnames = "CardNumber,CardFormat,PinRequired,PinCode,ExtendedAccess,ExpiryDate,Forename,Initial,Surname,Email,Phone,Custom1,Custom2,Schedule1,Schedule2,Schedule3,Schedule4,Schedule5,Schedule6,Schedule7,Schedule8".split(",")
|
|
|
|
# mapping of member levels to schedules
|
|
memberLevels = {"CMS Staff": "7x24",
|
|
"CMS Weekends Only": "Weekends Only",
|
|
"CMS Weekdays Only": "Weekdays Only",
|
|
"CMS Unlimited": "Unlimited",
|
|
"CMS Nights & Weekends": "Nights and Weekends"}
|
|
|
|
# door access permissions to controller IP addresses
|
|
doors = {
|
|
"Rental Studio": ["172.18.51.11"],
|
|
"Storage Closet": ["172.18.51.12"],
|
|
"Metal Shop": ["172.18.51.13"],
|
|
"Wood Shop": ["172.18.51.14", "172.18.51.15"]
|
|
}
|
|
|
|
|
|
def getMembershipworksData():
|
|
""" Pull the members csv from the membershipworks api """
|
|
BASE_URL = "https://api.membershipworks.com/v1/"
|
|
|
|
# login
|
|
sess = requests.Session()
|
|
r = sess.post(BASE_URL + 'usr',
|
|
data={"_st": "all",
|
|
"eml": MEMBERSHIPWORKS_USERNAME,
|
|
"org": "10000",
|
|
"pwd": MEMBERSHIPWORKS_PASSWORD})
|
|
SFtoken = r.json()['SF']
|
|
|
|
# get list of member/staff IDs
|
|
r = sess.get(BASE_URL + "ylp",
|
|
params={"SF": SFtoken,
|
|
"lbl": "5ae37979f033bfe8534f8799,5771675edcdf126302a2f6b9", # members and staff
|
|
"org": "15475", # unknown
|
|
"var": "_id,nam,ctc"})
|
|
ids = [user['_id'] for user in r.json()['usr']]
|
|
|
|
# get members CSV
|
|
# TODO: maybe can just use previous get instead? would return JSON
|
|
r = sess.post(BASE_URL + "csv",
|
|
params={"SF": SFtoken},
|
|
data={"_rt": "946702800", # unknown
|
|
"mux": "", # unknown
|
|
"tid": ",".join(ids), # ids of members to get
|
|
"var": "lvl,xws,xms,xsc,xrs,xfd,xac,phn,eml,lbl,nam"}) # which columns to get
|
|
return r.text
|
|
|
|
|
|
def makeMember(member, doorAuth):
|
|
"""Create an output CSV row for the member"""
|
|
if member["Access Card Number"] == "":
|
|
#print(member["First Name"], member["Last Name"], " has no card number, ignoring")
|
|
return
|
|
|
|
out = {"Forename": member["First Name"],
|
|
"Surname": member["Last Name"],
|
|
"Initial": "",
|
|
"CardNumber": member["Access Card Number"],
|
|
"CardFormat": "A901146A-241",
|
|
"PinRequired": "0",
|
|
"ExtendedAccess": "0",
|
|
"ExpiryDate": "",
|
|
"Email": member["Email"],
|
|
"Phone": member["Phone"]}
|
|
|
|
if doorAuth:
|
|
memberLevel = [v for k, v in memberLevels.items() if member[k] == k]
|
|
if len(memberLevel) == 1:
|
|
out["Schedule1"] = memberLevel[0]
|
|
else:
|
|
print(member["First Name"], member["Last Name"], "has no/too many member levels!")
|
|
|
|
return out
|
|
|
|
def doRequest(ip, params=None, files=None):
|
|
"""Send a request to the door control import script"""
|
|
r = requests.post(
|
|
'https://' + ip + '/cgi-bin/import.cgi',
|
|
params=params,
|
|
files=files,
|
|
auth=requests.auth.HTTPDigestAuth(DOOR_USERNAME, DOOR_PASSWORD),
|
|
verify=False) # ignore insecure SSL
|
|
print(r.text)
|
|
|
|
def updateDoor(doorIP, csv):
|
|
"""Do the update import procedure on a door control"""
|
|
|
|
doRequest(doorIP, {"task": "importInit"})
|
|
doRequest(doorIP,
|
|
{"task": "importCardsPeople", "name": "cardspeopleschedule.csv"},
|
|
{"importCardsPeopleButton": ("cardspeopleschedule.csv", csv, 'text/csv')})
|
|
doRequest(doorIP, {"task": "importDone"})
|
|
|
|
def makeDoor(doorName, doorIPs, members):
|
|
"""Create a CSV for the given door"""
|
|
outString = StringIO()
|
|
writer = csv.DictWriter(outString, fieldnames)
|
|
writer.writeheader()
|
|
for member in members:
|
|
member = makeMember(member, member["Access " + doorName + "?"] != "")
|
|
if member is not None:
|
|
writer.writerow(member)
|
|
|
|
print(outString.getvalue())
|
|
for doorIP in doorIPs:
|
|
outString.seek(0)
|
|
updateDoor(doorIP, outString)
|
|
pass
|
|
|
|
def main():
|
|
data = getMembershipworksData()
|
|
reader = csv.DictReader(StringIO(data))
|
|
members = list(reader)
|
|
|
|
for doorName, doorIP in doors.items():
|
|
makeDoor(doorName, doorIP, members)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|