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/doorUpdater.py

128 lines
4.5 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import sys
2018-05-07 18:02:04 -04:00
import requests
import csv
from io import StringIO
from hashlib import md5
import os
2018-05-07 18:02:04 -04:00
from common import *
2018-05-07 18:02:04 -04:00
2018-05-07 18:25:14 -04:00
# mapping of member levels to schedules
2018-05-07 18:02:04 -04:00
memberLevels = {"CMS Staff": "7x24",
"CMS Weekends Only": "Weekends Only",
"CMS Weekdays Only": "Weekdays Only",
"CMS Unlimited": "Unlimited",
"CMS Nights & Weekends": "Nights and Weekends"}
def getMembershipworksData():
2018-05-07 18:25:14 -04:00
""" Pull the members csv from the membershipworks api """
2018-05-07 18:02:04 -04:00
BASE_URL = "https://api.membershipworks.com/v1/"
2018-05-07 18:25:14 -04:00
# login
r = requests.post(BASE_URL + 'usr',
data={"_st": "all",
"eml": MEMBERSHIPWORKS_USERNAME,
"org": "10000",
"pwd": MEMBERSHIPWORKS_PASSWORD})
if r.status_code != 200 or 'SF' not in r.json():
print("MembershipWorks Login Error: ", r.status_code, r.reason)
print(r.text)
sys.exit(1)
2018-05-07 18:02:04 -04:00
SFtoken = r.json()['SF']
2018-05-07 18:25:14 -04:00
# get list of member/staff IDs
r = requests.get(BASE_URL + "ylp",
params={"SF": SFtoken,
"lbl": "5ae37979f033bfe8534f8799,5771675edcdf126302a2f6b9", # members and staff
"org": "15475", # unknown
"var": "_id,nam,ctc"})
if r.status_code != 200 or 'usr' not in r.json():
print("MembershipWorks User Listing Error: ", r.status_code, r.reason)
print(r.text)
sys.exit(1)
2018-05-07 18:02:04 -04:00
ids = [user['_id'] for user in r.json()['usr']]
2018-05-07 18:25:14 -04:00
# get members CSV
# TODO: maybe can just use previous get instead? would return JSON
r = requests.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,xas,xfd,xac,phn,eml,lbl,xcf,nam"}) # which columns to get
if r.status_code != 200:
print("MembershipWorks CSV Generation Error: ", r.status_code, r.reason)
print(r.text)
sys.exit(1)
2018-05-07 18:02:04 -04:00
return r.text
2018-05-07 18:02:04 -04:00
def makeMember(member, doorAuth):
2018-05-07 18:25:14 -04:00
"""Create an output CSV row for the member"""
2018-05-07 18:02:04 -04:00
if member["Access Card Number"] == "":
#print(member["First Name"], member["Last Name"], " has no card number, ignoring")
return
out = {"Forename": member["First Name"],
2018-05-07 18:25:14 -04:00
"Surname": member["Last Name"],
"Initial": "",
"CardNumber": member["Access Card Number"],
2018-05-08 20:35:33 -04:00
"CardFormat": "A901146A-" + member["Access Card Facility Code"],
2018-05-07 18:25:14 -04:00
"PinRequired": "0",
"ExtendedAccess": "0",
"ExpiryDate": "",
"Email": member["Email"],
"Phone": member["Phone"]}
2018-05-07 18:02:04 -04:00
2018-05-08 20:27:32 -04:00
if member[doorAuth] == "Y" \
and not member["Account on Hold"] == "Account on Hold":
levels = {k: v for k, v in memberLevels.items() if member[k] == k}
out["Custom1"] = "|".join(levels.keys()).replace("&", "and")
for index, schedule in enumerate(levels.values(), 1):
#TODO: error if people have more than 8?
out["Schedule" + str(index)] = schedule
2018-05-07 18:02:04 -04:00
return out
def makeDoor(doorName, doorData, members, hashes):
2018-05-07 18:25:14 -04:00
"""Create a CSV for the given door"""
2018-05-07 18:02:04 -04:00
outString = StringIO()
writer = csv.DictWriter(outString, fieldnames)
writer.writeheader()
for member in members:
2018-05-27 17:17:36 -04:00
member = makeMember(member, "Access " + doorData["access"] + "?")
2018-05-07 18:02:04 -04:00
if member is not None:
writer.writerow(member)
print(outString.getvalue())
2018-05-27 17:17:36 -04:00
outString.seek(0)
doorHash = md5(bytes(outString.getvalue(), 'utf8')).hexdigest()
if doorHash == hashes.get(doorName):
print("Door", doorName, "not changed, not updating")
else:
hashes[doorName] = doorHash
doCSVImport(doorData["ip"], outString)
# write out hash if we sucessfully updated this door
with open('/tmp/doorUpdaterLastHash', 'w') as f:
json.dump(hashes, f)
2018-05-07 18:02:04 -04:00
def main():
data = getMembershipworksData()
2018-05-07 18:02:04 -04:00
reader = csv.DictReader(StringIO(data))
members = list(reader)
if os.path.exists('/tmp/doorUpdaterLastHash'):
with open('/tmp/doorUpdaterLastHash', 'r') as f:
hashes = json.load(f)
else:
hashes = {}
2018-05-27 17:17:36 -04:00
for doorName, doorData in config["doors"].items():
print(doorName, doorData)
makeDoor(doorName, doorData, members, hashes)
2018-05-07 18:02:04 -04:00
if __name__ == '__main__':
main()