memberPlumbing/doorUpdater.py

#!/usr/bin/env python3

from common import doors, membershipworks, memberLevels, doorSpecificSchedules
from lib.hid.Credential import Credential
from lib.hid.DoorController import ROOT, E

class Member():
    def __init__(self, forename="", surname="", membershipWorksID="",
                 middleName="", email="", phone="",
                 cardholderID=None, doorAccess=[], onHold=False,
                 credentials=set(), levels=[], extraLevels=[], schedules=[]):
        self.forename = forename
        self.surname = surname
        self.membershipWorksID = membershipWorksID
        self.middleName = middleName
        self.email = email
        self.phone = phone
        self.cardholderID = cardholderID
        self.doorAccess = doorAccess
        self.onHold = onHold
        self.credentials = credentials
        self.levels = levels
        self.schedules = schedules

    def __repr__(self):
        return f"""Name: {self.forename} | {self.middleName} | {self.surname}
MembershipWorks ID: {self.membershipWorksID}
Email: {self.email}
Phone: {self.phone}
Cardholder ID: {self.cardholderID}
doorAccess: {self.doorAccess}
On Hold? {self.onHold}
Credentials: {self.credentials}
Levels: {self.levels}
Schedules: {self.schedules}
"""


class MembershipworksMember(Member):
    def __init__(self, data):
        super().__init__(data["First Name"],
                         data["Last Name"],
                         membershipWorksID=data["Account ID"],
                         email=data["Email"],
                         phone=data["Phone"],
                         onHold=data["Account on Hold"] != "")

        if data["Access Card Number"] != "":
            self.credentials = set([Credential(
                code=(data["Access Card Facility Code"],
                      data["Access Card Number"]))])
        else:
            self.credentials = set()

        levels = {k: v for k, v in memberLevels.items() if data[k] == k}
        self.levels = list(levels.keys())
        self.schedules = list(levels.values())

        self.extraLevels = {
            schedule: sum((doors for prop, doors in props.items()
                           if data[prop] == "Y"),
                          [])
            for schedule, props in doorSpecificSchedules.items()
        }

        self.doorAccess = [
            door for door, doorData in doors.items()
            if data["Access " + doorData.access + "?"] == "Y"
            and data['Access Permitted During Limited Operations'] == "Y"

        ]

    def to_DoorMember(self, door):
        doorLevels = [k for k, v in self.extraLevels.items() if door.name in v]

        if door.name not in self.doorAccess or self.onHold:
            schedules = []
        else:
            schedules = self.schedules + doorLevels

        dm = DoorMember(door, **self.__dict__)  # this is probably terrible
        dm.levels = self.levels + doorLevels
        dm.schedules = schedules

        return dm


class DoorMember(Member):
    def __init__(self, door, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.door = door

    @classmethod
    def from_cardholder(cls, data, door):
        ch = cls(door=door,
                 forename=data.get('forename', ""),
                 surname=data.get('surname', ""),
                 membershipWorksID=data.attrib.get('custom2', ""),
                 middleName=data.attrib.get('middleName', ""),
                 email=data.attrib.get('email', ""),
                 phone=data.attrib.get('phone', ""),
                 cardholderID=data.attrib['cardholderID'])

        ch.credentials = set(
            Credential(hex=(c.attrib['rawCardNumber']))
            for c in data.findall('{*}Credential'))

        ch.levels = data.attrib.get('custom1', "").split('|')
        ch.schedules = [r.attrib['scheduleName']
                        for r in data.findall('{*}Role')]

        return ch

    def attribs(self):
        return {"forename": self.forename,
                "surname": self.surname,
                "middleName": self.middleName,
                "email": self.email,
                "phone": self.phone,
                "custom1":  "|".join(self.levels).replace("&", "and"),
                "custom2": self.membershipWorksID}

    def make_schedules(self, schedulesMap):
        roles = [
            E.Role({"roleID": self.cardholderID,
                    "scheduleID": schedulesMap[schedule],
                    "resourceID": "0"})
            for schedule in self.schedules]

        return E.RoleSet({"action": "UD", "roleSetID": self.cardholderID},
                         E.Roles(*roles))

    def make_credentials(self, newCredentials, cardFormats):
        out = [
            E.Credential(
                {"formatName": str(credential.code[0]),
                 "cardNumber": str(credential.code[1]),
                 "formatID": cardFormats[str(credential.code[0])],
                 "isCard": "true",
                 "cardholderID": self.cardholderID})
            for credential in newCredentials]

        return E.Credentials({"action": "AD"}, *out)


def update_door(door, members):
    cardFormats = door.get_cardFormats()
    cardholders = {member.membershipWorksID: member
                   for member in [DoorMember.from_cardholder(ch, door)
                                  for ch in door.get_cardholders()]}
    schedulesMap = door.get_scheduleMap()
    allCredentials = set(Credential(hex=c.attrib['rawCardNumber'])
                         for c in door.get_credentials())

    # TODO: can I combine requests?
    for membershipworksMember in members:
        member = membershipworksMember.to_DoorMember(door)
        # cardholder exists, compare contents
        if member.membershipWorksID in cardholders:
            ch = cardholders.pop(member.membershipWorksID)
            member.cardholderID = ch.cardholderID
            if member.attribs() != ch.attribs():  # update cardholder attributes
                print(f"- Updating profile for {member.forename} {member.surname}")
                print(f"  - Old: {ch.attribs()}")
                print(f"  - New: {member.attribs()}")
                door.doXMLRequest(ROOT(
                    E.Cardholders(
                        {"action": "UD", "cardholderID": member.cardholderID},
                        E.CardHolder(member.attribs()))))

            if member.credentials != ch.credentials:
                print(f"- Updating card for {member.forename} {member.surname}")
                print(f"  - {ch.credentials} -> {member.credentials}")

                oldCards = ch.credentials
                newCards = member.credentials

                allNewCards = set(
                    card for m in members if m != membershipworksMember
                    for card in m.credentials)

                # cards removed, and won't be reassigned to someone else
                for card in (oldCards - newCards) - allNewCards:
                    door.doXMLRequest(ROOT(E.Credentials(
                        {"action": "UD",
                         "rawCardNumber": card.hex,
                         "isCard": "true"},
                        E.Credential({"cardholderID": ""}))))

                if newCards - oldCards:  # cards added
                    for card in newCards & allNewCards:  # new card exists in another member
                        print([m for m in members
                              for card in m.credentials
                               if card in newCards])
                        raise Exception(f"Duplicate Card in input data! {card}")

                    # card existed in door, and needs to be reassigned
                    for card in newCards & allCredentials:
                        door.doXMLRequest(ROOT(E.Credentials(
                            {"action": "UD",
                             "rawCardNumber": card.hex,
                             "isCard": "true"},
                            E.Credential({"cardholderID": member.cardholderID}))))

                    # cards that never existed, and need to be created
                    if newCards - allCredentials:
                        door.doXMLRequest(ROOT(member.make_credentials(
                            newCards - allCredentials, cardFormats)))

            if member.schedules != ch.schedules:
                print("- Updating schedule for" +
                      f" {member.forename} {member.surname}:" +
                      f" {ch.schedules} -> {member.schedules}")
                door.doXMLRequest(ROOT(
                    member.make_schedules(schedulesMap)))
        else:  # cardholder did not exist, so add them
            print(f"- Adding Member:")
            print(member)
            resp = door.doXMLRequest(ROOT(
                E.Cardholders(
                    {"action": "AD"},
                    E.Cardholder(member.attribs()))))
            member.cardholderID = resp.find('{*}Cardholders/{*}Cardholder') \
                                      .attrib["cardholderID"]
            door.doXMLRequest(ROOT(
                member.make_credentials(member.credentials, cardFormats),
                member.make_schedules(schedulesMap)))

    # TODO: delete cardholders that are no longer members?


def main():
    memberData = membershipworks.get_members(
        ['Members', 'CMS Staff', 'Misc. Access'],
        "_id,nam,phn,eml,lvl,lbl,xws,xms,xsc,xas,xfd,xac,xcf,xeh,xse,xlo")

    members = [MembershipworksMember(m) for m in memberData]

    for door in doors.values():
        print(door.name, door.ip)
        update_door(door, members)


if __name__ == '__main__':
    main()