def store_event(data):
event = Event(id=data['key'])
event.name = data['name']
event.short_name = data['short_name']
event.event_short = data['event_code']
event.event_type_enum = data['event_type']
event.year = data['year']
event.timezone_id = data['timezone']
event.website = data['website']
event.start_date = datetime.datetime.strptime(
data['start_date'],
EVENT_DATE_FORMAT_STR) if data['start_date'] else None
event.end_date = datetime.datetime.strptime(
data['end_date'], EVENT_DATE_FORMAT_STR) if data['end_date'] else None
event.webcast_json = json.dumps(data['webcasts'])
event.venue = data['location_name']
event.city = data['city']
event.state_prov = data['state_prov']
event.country = data['country']
event.playoff_type = data['playoff_type']
event.parent_event = ndb.Key(
Event, data['parent_event_key']) if data['parent_event_key'] else None
event.divisions = [
ndb.Key(Event, div_key) for div_key in data['division_keys']
] if data['division_keys'] else []
district = store_district(data['district']) if data['district'] else None
event.district_key = district.key if district else None
return EventManipulator.createOrUpdate(event)
def test_is_maybe_match_wrong_end(self):
event_one = Event(start_date=datetime.datetime(year=2020,
month=7,
day=14,
hour=0,
minute=0,
second=0),
end_date=datetime.datetime(year=2020,
month=7,
day=15,
hour=23,
minute=59,
second=59),
city="London",
state_prov="OH")
event_two = Event(start_date=datetime.datetime(year=2020,
month=7,
day=14,
hour=0,
minute=0,
second=0),
end_date=datetime.datetime(year=2020,
month=7,
day=16,
hour=23,
minute=59,
second=59),
city="London",
state_prov="OH")
self.assertFalse(
OffseasonEventHelper.is_maybe_match(event_one, event_two))
event_two.end_date = event_one.end_date
self.assertTrue(
OffseasonEventHelper.is_maybe_match(event_one, event_two))
def store_event(data):
event = Event(id=data['key'])
event.name = data['name']
event.short_name = data['short_name']
event.event_short = data['event_code']
event.event_type_enum = data['event_type']
event.year = data['year']
event.timezone_id = data['timezone']
event.website = data['website']
event.start_date = datetime.datetime.strptime(data['start_date'], EVENT_DATE_FORMAT_STR) if data['start_date'] else None
event.end_date = datetime.datetime.strptime(data['end_date'], EVENT_DATE_FORMAT_STR) if data['end_date'] else None
event.webcast_json = json.dumps(data['webcasts'])
event.venue = data['location_name']
event.city = data['city']
event.state_prov = data['state_prov']
event.country = data['country']
district = store_district(data['district']) if data['district'] else None
event.district_key = district.key if district else None
return EventManipulator.createOrUpdate(event)
