""" these are helper classes for object creation during the scrape """
from pupa.models.person import Person
from pupa.models.organization import Organization
from pupa.models.membership import Membership


class Legislator(Person):
    _is_legislator = True
    __slots__ = ('post_id', 'party', 'chamber', '_contact_details')

    def __init__(self, name, post_id, party=None, chamber=None, **kwargs):
        super(Legislator, self).__init__(name, **kwargs)
        self.post_id = post_id
        self.party = party
        self.chamber = chamber
        self._contact_details = []

    def add_contact(self, type, value, note):
        self._contact_details.append({'type': type, 'value': value,
                                      'note': note})

    def add_committee_membership(self, com_name, role='member'):
        org = Organization(com_name, classification='committee')
        self.add_membership(org, role=role)
        org.sources = self.sources
        self._related.append(org)


class Committee(Organization):

    def __init__(self, *args, **kwargs):
        super(Committee, self).__init__(*args, **kwargs)

    def add_member(self, name, role='member', **kwargs):
        membership = Membership(None, self._id, role=role,
                                unmatched_legislator={'name': name},
                                **kwargs)
        self._related.append(membership)

""" these are helper classes for object creation during the scrape """
from pupa.models.person import Person
from pupa.models.organization import Organization
from pupa.models.membership import Membership


class Legislator(Person):
    _is_legislator = True
    __slots__ = ('post_id', 'party', 'chamber', '_contact_details', 'role')

    def __init__(self, name, post_id, party=None, chamber=None, **kwargs):
        super(Legislator, self).__init__(name, **kwargs)
        self.post_id = post_id
        self.party = party
        self.chamber = chamber
        self._contact_details = []

    def add_contact(self, type, value, note):
        self._contact_details.append({'type': type, 'value': value,
                                      'note': note})

    def add_committee_membership(self, com_name, role='member'):
        org = Organization(com_name, classification='committee')
        self.add_membership(org, role=role)
        org.sources = self.sources
        self._related.append(org)


class Committee(Organization):

    def __init__(self, *args, **kwargs):
        super(Committee, self).__init__(*args, **kwargs)

    def add_member(self, name, role='member', **kwargs):
        membership = Membership(None, self._id, role=role,
                                unmatched_legislator={'name': name},
                                **kwargs)
        self._related.append(membership)

Add a slot for legislator role
Add a slot for legislator role
