18F/openFEC

from webservices import docs, utils

from .base import db, BaseModel


class BaseAggregate(BaseModel):
    __abstract__ = True

    committee = utils.related_committee_history('committee_id', cycle_label='cycle')
    committee_id = db.Column('cmte_id', db.String, primary_key=True, doc=docs.COMMITTEE_ID)
    cycle = db.Column(db.Integer, primary_key=True, doc=docs.RECORD_CYCLE)
    #? not sure how to document this
    total = db.Column(db.Numeric(30, 2), index=True,)
    count = db.Column(db.Integer, index=True, doc='Number of records making up the total')


class ScheduleABySize(BaseAggregate):
    __tablename__ = 'ofec_sched_a_aggregate_size_merged_mv'
    size = db.Column(db.Integer, primary_key=True)


class ScheduleAByState(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_a_aggregate_state'
    state = db.Column(db.String, primary_key=True, doc=docs.STATE_GENERIC)
    state_full = db.Column(db.String, primary_key=True, doc=docs.STATE_GENERIC)


class ScheduleAByZip(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_a_aggregate_zip'
    zip = db.Column(db.String, primary_key=True)
    state = db.Column(db.String, doc=docs.STATE_GENERIC)
    state_full = db.Column(db.String, doc=docs.STATE_GENERIC)


class ScheduleAByEmployer(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_a_aggregate_employer'
    employer = db.Column(db.String, primary_key=True, doc=docs.EMPLOYER)


class ScheduleAByOccupation(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_a_aggregate_occupation'
    occupation = db.Column(db.String, primary_key=True, doc=docs.OCCUPATION)


class ScheduleBByRecipient(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_b_aggregate_recipient'
    recipient_name = db.Column('recipient_nm', db.String, primary_key=True, doc=docs.RECIPIENT_NAME)


class ScheduleBByRecipientID(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_b_aggregate_recipient_id'
    recipient_id = db.Column('recipient_cmte_id', db.String, primary_key=True, doc=docs.RECIPIENT_ID)
    committee = utils.related_committee('committee_id')
    recipient = utils.related('CommitteeHistory', 'recipient_id', 'committee_id', cycle_label='cycle')

    @property
    def committee_name(self):
        return self.committee.name

    @property
    def recipient_name(self):
        return self.recipient.name


class ScheduleBByPurpose(BaseAggregate):
    __table_args__ = {'schema' : 'disclosure'}
    __tablename__ = 'dsc_sched_b_aggregate_purpose'
    purpose = db.Column(db.String, primary_key=True, doc=docs.PURPOSE)


class BaseSpendingAggregate(BaseAggregate):
    __abstract__ = True
    committee_id = db.Column('cmte_id', db.String, primary_key=True, doc=docs.COMMITTEE_ID)
    committee = utils.related_committee('committee_id')
    candidate_id = db.Column('cand_id', db.String, primary_key=True, doc=docs.CANDIDATE_ID)
    candidate = utils.related_candidate('candidate_id')


class ScheduleEByCandidate(BaseSpendingAggregate):
    __tablename__ = 'ofec_sched_e_aggregate_candidate_mv'

    support_oppose_indicator = db.Column(db.String, primary_key=True, doc=docs.SUPPORT_OPPOSE_INDICATOR)


class CommunicationCostByCandidate(BaseSpendingAggregate):
    __tablename__ = 'ofec_communication_cost_aggregate_candidate_mv'
    support_oppose_indicator = db.Column(db.String, primary_key=True, doc=docs.SUPPORT_OPPOSE_INDICATOR)

class ElectioneeringByCandidate(BaseSpendingAggregate):
    __tablename__ = 'ofec_electioneering_aggregate_candidate_mv'