def add_legislature(self, db, data):
date_start = timestamp_to_date(data.get('date_start'))
date_end = timestamp_to_date(data.get('date_end'))
legislature = Legislature(
institution_id=data.get('institution_id'),
date_start=date_start,
date_end=date_end,
)
db.add(legislature)
db.flush()
logging.debug('Added legislature: "%s"', str(legislature))
return legislature
def add_mandate(self, db, data):
date_start = timestamp_to_date(data.get('date_start'))
date_end = timestamp_to_date(data.get('date_end'))
mandate = Mandate(
legislator_id=data.get('legislator_id'),
political_office_id=data.get('political_office_id'),
date_start=date_start,
date_end=date_end,
)
db.add(mandate)
db.flush()
logging.debug(u'Added mandate: "%s"', str(mandate))
return mandate
def add_legislator_events(self, db, data):
date = timestamp_to_date(data.get('date'))
legislator_events = LegislatorEvents(
legislator_events_type_id=data.get('legislator_events_type_id'),
legislator_id=data.get('legislator_id'),
date=date,
)
db.add(legislator_events)
db.flush()
logging.debug(u'Added legislator events: "%s"', str(legislator_events))
return legislator_events
def add_mandate_events(self, db, data):
date = timestamp_to_date(data.get('date'))
mandate_events = MandateEvents(
mandate_events_type_id=data.get('mandate_events_type_id'),
mandate_id=data.get('mandate_id'),
date=date,
)
db.add(mandate_events)
db.flush()
logging.debug('Added mandate events: "%s"', str(mandate_events))
return mandate_events
def add_legislator(self, db, data):
date_of_birth = None
if data.get('date_of_birth'):
date_of_birth = timestamp_to_date(data.get('date_of_birth'))
legislator = Legislator(
name=data.get('name'),
picture=data.get('picture'),
website=data.get('website'),
email=data.get('email'),
gender=data.get('gender'),
date_of_birth=date_of_birth,
about=data.get('about'),
)
db.add(legislator)
db.flush()
logging.debug(u'Added legislator: "%s"', str(legislator))
return legislator
def add_legislator(self, db, data):
date_of_birth = None
if data.get("date_of_birth"):
date_of_birth = timestamp_to_date(data.get("date_of_birth"))
legislator = Legislator(
name=data.get("name"),
picture=data.get("picture"),
website=data.get("website"),
email=data.get("email"),
gender=data.get("gender"),
date_of_birth=date_of_birth,
about=data.get("about"),
)
db.add(legislator)
db.flush()
logging.debug('Added legislator: "%s"', str(legislator))
return legislator
def test_can_convert_timestamp_to_date(self):
dt1 = timestamp_to_date(1435363200)
dt2 = datetime.strptime('Jun 27 2015', '%b %d %Y').date()
expect(dt1).to_equal(dt2)
