def main(): today = date.today() holidays = get_holidays(today.year) days = get_days(today, holidays) for route_id, directions in station_ids.iteritems(): scrape_route(route_id, directions, days) time.sleep(10) save_names(station_names)
def get_weekday(day): if day.month == 12 and day.day == 25 or day.month == 1 and day.day == 1: return 'none' if day.month == 12 and day.day in (24, 31): return 'holiday' holidays = get_holidays(day.year) if day in holidays or day.weekday() == 6: return 'sunday' if day.weekday() == 5: return 'saturday' return 'weekday'