aswi2021jmsd-gitlab4/src/model/subject.py @ e96150c4

class Subject:
    ATTR_MAP = {"C": "country", "ST": "state", "L": "locality", "CN": "common_name", "O": "organization",
                "OU": "organization_unit", "emailAddress": "email_address"}
    INVERSE_ATTR_MAP = {k: v for v, k in ATTR_MAP.items()}

    def __init__(self, common_name=None, country=None, locality=None, state=None, organization=None,
                 organization_unit=None, email_address=None):
        self.common_name = common_name
        self.country = country
        self.locality = locality
        self.state = state
        self.organization = organization
        self.organization_unit = organization_unit
        self.email_address = email_address

    @staticmethod
    def from_dict(d):
        if not isinstance(d, dict):
            return None

        s = Subject()
        for k, v in Subject.ATTR_MAP.items():
            if k in d:
                if not isinstance(d[k], str):
                    return None
                s.__setattr__(v, d[k])

        return s

    def to_dict(self):
        return {Subject.INVERSE_ATTR_MAP[k]: v for k, v in vars(self).items()}