steve.goldsmith/memberPlumbing
1
0
Fork 0
forked from CMS/memberPlumbing
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add some comments

Browse Source
This commit is contained in:
Adam Goldsmith 2018-05-07 18:25:14 -04:00
parent b25deb6630
commit 437503a5b1
1 changed files with 30 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
doorUpdater.py
View File

@ -5,16 +5,19 @@ import urllib3
from passwords import * from passwords import *
# it's fine, ssl certs are for losers anyway
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) 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(",") 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", memberLevels = {"CMS Staff": "7x24",
"CMS Weekends Only": "Weekends Only", "CMS Weekends Only": "Weekends Only",
"CMS Weekdays Only": "Weekdays Only", "CMS Weekdays Only": "Weekdays Only",
"CMS Unlimited": "Unlimited", "CMS Unlimited": "Unlimited",
"CMS Nights & Weekends": "Nights and Weekends"} "CMS Nights & Weekends": "Nights and Weekends"}
# door access permissions to controller IP addresses
doors = { doors = {
"Rental Studio": ["172.18.51.11"], "Rental Studio": ["172.18.51.11"],
"Storage Closet": ["172.18.51.12"], "Storage Closet": ["172.18.51.12"],
@ -22,9 +25,12 @@ doors = {
"Wood Shop": ["172.18.51.14", "172.18.51.15"] "Wood Shop": ["172.18.51.14", "172.18.51.15"]
} }
def getMembershipworksData(): def getMembershipworksData():
""" Pull the members csv from the membershipworks api """
BASE_URL = "https://api.membershipworks.com/v1/" BASE_URL = "https://api.membershipworks.com/v1/"
# login
sess = requests.Session() sess = requests.Session()
r = sess.post(BASE_URL + 'usr', r = sess.post(BASE_URL + 'usr',
data={"_st": "all", data={"_st": "all",
@ -33,23 +39,27 @@ def getMembershipworksData():
"pwd": MEMBERSHIPWORKS_PASSWORD}) "pwd": MEMBERSHIPWORKS_PASSWORD})
SFtoken = r.json()['SF'] SFtoken = r.json()['SF']
# get list of member/staff IDs
r = sess.get(BASE_URL + "ylp", r = sess.get(BASE_URL + "ylp",
params={"SF": SFtoken, params={"SF": SFtoken,
"lbl": "5ae37979f033bfe8534f8799,5771675edcdf126302a2f6b9", "lbl": "5ae37979f033bfe8534f8799,5771675edcdf126302a2f6b9", # members and staff
"org": "15475", "org": "15475", # unknown
"var": "_id,nam,ctc"}) "var": "_id,nam,ctc"})
ids = [user['_id'] for user in r.json()['usr']] 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", r = sess.post(BASE_URL + "csv",
params={"SF": SFtoken}, params={"SF": SFtoken},
data={"_rt": "946702800", data={"_rt": "946702800", # unknown
"mux": "", "mux": "", # unknown
"tid": ",".join(ids), "tid": ",".join(ids), # ids of members to get
"var": "lvl,xws,xms,xsc,xrs,xfd,xac,phn,eml,lbl,nam"}) "var": "lvl,xws,xms,xsc,xrs,xfd,xac,phn,eml,lbl,nam"}) # which columns to get
return r.text return r.text
def makeMember(member, doorAuth): def makeMember(member, doorAuth):
"""Create an output CSV row for the member"""
if member["Access Card Number"] == "": if member["Access Card Number"] == "":
#print(member["First Name"], member["Last Name"], " has no card number, ignoring") #print(member["First Name"], member["Last Name"], " has no card number, ignoring")
return return
@ -75,15 +85,18 @@ def makeMember(member, doorAuth):
return out return out
def doRequest(ip, params=None, files=None): def doRequest(ip, params=None, files=None):
"""Send a request to the door control import script"""
r = requests.post( r = requests.post(
'https://' + ip + '/cgi-bin/import.cgi', 'https://' + ip + '/cgi-bin/import.cgi',
params=params, params=params,
files=files, files=files,
auth=requests.auth.HTTPDigestAuth(DOOR_USERNAME, DOOR_PASSWORD), auth=requests.auth.HTTPDigestAuth(DOOR_USERNAME, DOOR_PASSWORD),
verify=False) verify=False) # ignore insecure SSL
print(r.text) print(r.text)
def updateDoor(doorIP, csv): def updateDoor(doorIP, csv):
"""Do the update import procedure on a door control"""
doRequest(doorIP, {"task": "importInit"}) doRequest(doorIP, {"task": "importInit"})
doRequest(doorIP, doRequest(doorIP,
{"task": "importCardsPeople", "name": "cardspeopleschedule.csv"}, {"task": "importCardsPeople", "name": "cardspeopleschedule.csv"},
@ -91,6 +104,7 @@ def updateDoor(doorIP, csv):
doRequest(doorIP, {"task": "importDone"}) doRequest(doorIP, {"task": "importDone"})
def makeDoor(doorName, doorIPs, members): def makeDoor(doorName, doorIPs, members):
"""Create a CSV for the given door"""
outString = StringIO() outString = StringIO()
writer = csv.DictWriter(outString, fieldnames) writer = csv.DictWriter(outString, fieldnames)
writer.writeheader() writer.writeheader()