"""
Constant defining different variation of the file names in GTFS / our model names, mapped to corresponding models.
"""

import pt_map.models
import datetime

time_delta = int(datetime.datetime(2024,1,1).timestamp())

class_names = [
{
    "Agency": pt_map.models.Agency,
    "Stop": pt_map.models.Stop,
    "Route": pt_map.models.Route,
    "Trip": pt_map.models.Trip,
    "StopTime": pt_map.models.StopTime,
    "Calendar": pt_map.models.Calendar,
    "CalendarDate": pt_map.models.CalendarDate,
    "FareAttribute": pt_map.models.FareAttribute,
    "FareRule": pt_map.models.FareRule,
    "Timeframe": pt_map.models.Timeframe,
    "FareMedium": pt_map.models.FareMedium,
    "FareProduct": pt_map.models.FareProduct,
    "FareLegRule": pt_map.models.FareLegRule,
    "FareTransferRule": pt_map.models.FareTransferRule,
    "Area": pt_map.models.Area,
    "StopArea": pt_map.models.StopArea,
    "Network": pt_map.models.Network,
    "RouteNetwork": pt_map.models.RouteNetwork,
    "Shape": pt_map.models.Shape,
    "Frequency": pt_map.models.Frequency,
    "Transfer": pt_map.models.Transfer,
    "Pathway": pt_map.models.Pathway,
    "Level": pt_map.models.Level,
    "LocationGroup": pt_map.models.LocationGroup,
    "LocationGroupStop": pt_map.models.LocationGroupStop,
    "LocationsGeojson": pt_map.models.LocationsGeojson,
    "BookingRule": pt_map.models.BookingRule,
    "Translation": pt_map.models.Translation,
    "FeedInfo": pt_map.models.FeedInfo,
    "Attribution": pt_map.models.Attribution,
}, 
{
    "agencies": pt_map.models.Agency,
    "stops": pt_map.models.Stop,
    "routes": pt_map.models.Route,
    "trips": pt_map.models.Trip,
    "stop_times": pt_map.models.StopTime,
    "calendars": pt_map.models.Calendar,
    "calendar_dates": pt_map.models.CalendarDate,
    "fare_attributes": pt_map.models.FareAttribute,
    "fare_rules": pt_map.models.FareRule,
    "timeframes": pt_map.models.Timeframe,
    "fare_mediums": pt_map.models.FareMedium,
    "fare_products": pt_map.models.FareProduct,
    "fare_leg_rules": pt_map.models.FareLegRule,
    "fare_transfer_rules": pt_map.models.FareTransferRule,
    "areas": pt_map.models.Area,
    "stop_areas": pt_map.models.StopArea,
    "networks": pt_map.models.Network,
    "route_networks": pt_map.models.RouteNetwork,
    "shapes": pt_map.models.Shape,
    "frequencies": pt_map.models.Frequency,
    "transfers": pt_map.models.Transfer,
    "pathways": pt_map.models.Pathway,
    "levels": pt_map.models.Level,
    "location_groups": pt_map.models.LocationGroup,
    "location_group_stops": pt_map.models.LocationGroupStop,
    "locations": pt_map.models.LocationsGeojson,
    "booking_rules": pt_map.models.BookingRule,
    "translations": pt_map.models.Translation,
    "feed_info": pt_map.models.FeedInfo,
    "attributions": pt_map.models.Attribution,
}, 
{
    pt_map.models.Agency: "Agency",
    pt_map.models.Stop: "Stop",
    pt_map.models.Route: "Route",
    pt_map.models.Trip: "Trip",
    pt_map.models.StopTime: "StopTime",
    pt_map.models.Calendar: "Calendar",
    pt_map.models.CalendarDate: "CalendarDate",
    pt_map.models.FareAttribute: "FareAttribute",
    pt_map.models.FareRule: "FareRule",
    pt_map.models.Timeframe: "Timeframe",
    pt_map.models.FareMedium: "FareMedium",
    pt_map.models.FareProduct: "FareProduct",
    pt_map.models.FareLegRule: "FareLegRule",
    pt_map.models.FareTransferRule: "FareTransferRule",
    pt_map.models.Area: "Area",
    pt_map.models.StopArea: "StopArea",
    pt_map.models.Network: "Network",
    pt_map.models.RouteNetwork: "RouteNetwork",
    pt_map.models.Shape: "Shape",
    pt_map.models.Frequency: "Frequency",
    pt_map.models.Transfer: "Transfer",
    pt_map.models.Pathway: "Pathway",
    pt_map.models.Level: "Level",
    pt_map.models.LocationGroup: "LocationGroup",
    pt_map.models.LocationGroupStop: "LocationGroupStop",
    pt_map.models.LocationsGeojson: "LocationsGeojson",
    pt_map.models.BookingRule: "BookingRule",
    pt_map.models.Translation: "Translation",
    pt_map.models.FeedInfo: "FeedInfo",
    pt_map.models.Attribution: "Attribution",
}, 
{
    pt_map.models.Agency: "agencies",
    pt_map.models.Stop: "stops",
    pt_map.models.Route: "routes",
    pt_map.models.Trip: "trips",
    pt_map.models.StopTime: "stop_times",
    pt_map.models.Calendar: "calendars",
    pt_map.models.CalendarDate: "calendar_dates",
    pt_map.models.FareAttribute: "fare_attributes",
    pt_map.models.FareRule: "fare_rules",
    pt_map.models.Timeframe: "timeframes",
    pt_map.models.FareMedium: "fare_mediums",
    pt_map.models.FareProduct: "fare_products",
    pt_map.models.FareLegRule: "fare_leg_rules",
    pt_map.models.FareTransferRule: "fare_transfer_rules",
    pt_map.models.Area: "areas",
    pt_map.models.StopArea: "stop_areas",
    pt_map.models.Network: "networks",
    pt_map.models.RouteNetwork: "route_networks",
pt_map.models.Shape: "shapes",
    pt_map.models.Frequency: "frequencies",
    pt_map.models.Transfer: "transfers",
    pt_map.models.Pathway: "pathways",
    pt_map.models.Level: "levels",
    pt_map.models.LocationGroup: "location_groups",
    pt_map.models.LocationGroupStop: "location_group_stops",
    pt_map.models.LocationsGeojson: "locations",
    pt_map.models.BookingRule: "booking_rules",
    pt_map.models.Translation: "translations",
    pt_map.models.FeedInfo: "feed_info",
    pt_map.models.Attribution: "attributions",
}, 
]


reversed_file_mapping = {
    "Agency": "agency",
    "Stop": "stops",
    "Route": "routes",
    "Trip": "trips",
    "StopTime": "stop_times",
    "Calendar": "calendar",
    "CalendarDate": "calendar_dates",
    "FareAttribute": "fare_attributes",
    "FareRule": "fare_rules",
    "Timeframe": "timeframes",
    "FareMedium": "fare_media",
    "FareProduct": "fare_products",
    "FareLegRule": "fare_leg_rules",
    "FareTransferRule": "fare_transfer_rules",
    "Area": "areas",
    "StopArea": "stop_areas",
    "Network": "networks",
    "RouteNetwork": "route_networks",
    "Shape": "shapes",
    "Frequency": "frequencies",
    "Transfer": "transfers",
    "Pathway": "pathways",
    "Level": "levels",
    "LocationGroup": "location_groups",
    "LocationGroupStop": "location_group_stops",
    "LocationsGeojson": "locations_geojson",
    "BookingRule": "booking_rules",
    "Translation": "translations",
    "FeedInfo": "feed_info",
    "Attribution": "attributions"
}


case_swap = {'Agency': 'agency', 'Stop': 'stops', 'Route': 'routes', 'Trip': 'trips', 'StopTime': 'stop_times', 'Calendar': 'calendar', 'CalendarDate': 'calendar_dates', 'FareAttribute': 'fare_attributes', 'FareRule': 'fare_rules', 'Timeframe': 'timeframes', 'FareMedium': 'fare_media', 'FareProduct': 'fare_products', 'FareLegRule': 'fare_leg_rules', 'FareTransferRule': 'fare_transfer_rules', 'Area': 'areas', 'StopArea': 'stop_areas', 'Network': 'networks', 'RouteNetwork': 'route_networks', 'Shape': 'shapes', 'Frequency': 'frequencies', 'Transfer': 'transfers', 'Pathway': 'pathways', 'Level': 'levels', 'LocationGroup': 'location_groups', 'LocationGroupStop': 'location_group_stops', 'LocationsGeojson': 'locations_geojson', 'BookingRule': 'booking_rules', 'Translation': 'translations', 'FeedInfo': 'feed_info', 'Attribution': 'attributions'}


#primary_keys = 	{ pt_map.models.FeedInfo: None,
#    pt_map.models.Agency: "agency_id",
#	pt_map.models.Level: "level_id",
#	pt_map.models.Stop: "stop_id",
#	pt_map.models.Route: "route_id", 
#	pt_map.models.Shape: "shape_id",
#	pt_map.models.Calendar: "service_id",
#	pt_map.models.CalendarDate: None,
#	pt_map.models.Trip: "trip_id",
#	pt_map.models.LocationGroup: "location_group_id",
#	pt_map.models.LocationsGeojson: None,
#	pt_map.models.StopTime: None,
#	pt_map.models.FareAttribute: "fare_id",
#	pt_map.models.FareRule: None,
#	pt_map.models.Frequency: None,
#	pt_map.models.Transfer: None,
#	pt_map.models.Pathway: "pathway_id",
#	pt_map.models.BookingRule: "booking_rule_id",
#	pt_map.models.Translation: None,
#	pt_map.models.Attribution: "attribution_id",
#	pt_map.models.LocationGroupStop: None,
#	pt_map.models.Network: "network_id",
#	pt_map.models.RouteNetwork: None,
#	pt_map.models.Area: None,
#	pt_map.models.StopArea: None,
#	pt_map.models.FareMedium: "fare_media_id",
#	pt_map.models.FareProduct: None,
#	pt_map.models.Timeframe: None,
#	pt_map.models.FareLegRule: None,
#	pt_map.models.FareTransferRule: None,
 #}

primary_keys = {
    pt_map.models.FeedInfo: "feed_id",
    pt_map.models.Agency: "agency_id",
    pt_map.models.Level: "level_id",
    pt_map.models.Stop: "stop_id",
    pt_map.models.Route: "route_id",
    pt_map.models.Shape: "shape_pk",
    pt_map.models.Calendar: "service_id",
    pt_map.models.CalendarDate: "calendar_date_id",
    pt_map.models.Trip: "trip_id",
    pt_map.models.LocationGroup: "location_group_id",
    pt_map.models.LocationsGeojson: "location_id",
    pt_map.models.StopTime: "stop_time_id",
    pt_map.models.FareAttribute: "fare_id",
    pt_map.models.FareRule: "fare_rule_id",
    pt_map.models.Frequency: "frequency_id",
    pt_map.models.Transfer: "transfer_id",
    pt_map.models.Pathway: "pathway_id",
    pt_map.models.BookingRule: "booking_rule_id",
    pt_map.models.Translation: "translation_id",
    pt_map.models.Attribution: "attribution_id",
    pt_map.models.LocationGroupStop: "location_group_stop_id",
    pt_map.models.Network: "network_id",
    pt_map.models.RouteNetwork: "route_network_id",
    pt_map.models.Area: "area_id",
    pt_map.models.StopArea: "stop_area_id",
    pt_map.models.FareMedium: "fare_media_id",
    pt_map.models.FareProduct: "fare_product_id",
    pt_map.models.Timeframe: "timeframe_group_id",
    pt_map.models.FareLegRule: "fare_leg_rule_id",
    pt_map.models.FareTransferRule: "fare_transfer_rule_id",
}


classes_by_primary_keys = 	{ 
	'feed_id': pt_map.models.FeedInfo, 
	'agency_id': pt_map.models.Agency, 
	'level_id': pt_map.models.Level, 
	'stop_id': pt_map.models.Stop, 
	'route_id': pt_map.models.Route, 
    'shape_pk': pt_map.models.Shape, 
	'service_id': pt_map.models.Calendar, 
	'calendar_date_id': pt_map.models.CalendarDate, 
	'trip_id': pt_map.models.Trip, 
	'location_group_id': pt_map.models.LocationGroup, 
	'location_id': pt_map.models.LocationsGeojson, 
	'stop_time_id': pt_map.models.StopTime, 
	'fare_id': pt_map.models.FareAttribute, 
	'fare_rule_id': pt_map.models.FareRule, 
	'frequency_id': pt_map.models.Frequency, 
	'transfer_id': pt_map.models.Transfer, 
	'pathway_id': pt_map.models.Pathway, 
	'booking_rule_id': pt_map.models.BookingRule, 
	'translation_id': pt_map.models.Translation, 
	'attribution_id': pt_map.models.Attribution, 
	'location_group_stop_id': pt_map.models.LocationGroupStop, 
	'network_id': pt_map.models.Network, 
	'route_network_id': pt_map.models.RouteNetwork, 
	'area_id': pt_map.models.Area, 
	'stop_area_id': pt_map.models.StopArea, 
	'fare_media_id': pt_map.models.FareMedium, 
	'fare_product_id': pt_map.models.FareProduct, 
	'timeframe_group_id': pt_map.models.Timeframe, 
	'fare_leg_rule_id': pt_map.models.FareLegRule, 
    'fare_transfer_rule_id': pt_map.models.FareTransferRule,
}

foreign_keys = [
	(pt_map.models.FeedInfo, []),
	(pt_map.models.Agency, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Level, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Stop, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Stop, 'parent_station'), (pt_map.models.Level, 'level_id'), ]),
	(pt_map.models.Route, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Agency, 'agency_id'), ]),
	(pt_map.models.Shape, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Calendar, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.CalendarDate, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Trip, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Route, 'route_id'), (pt_map.models.Shape, 'shape_id'), ]),
	(pt_map.models.LocationGroup, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.LocationsGeojson, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.StopTime, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Trip, 'trip_id'), (pt_map.models.Stop, 'stop_id'), (pt_map.models.LocationGroup, 'location_group_id'), (pt_map.models.LocationsGeojson, 'location_id'), ]),
	(pt_map.models.FareAttribute, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Agency, 'agency_id'), ]),
	(pt_map.models.FareRule, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.FareAttribute, 'fare_id'), (pt_map.models.Route, 'route_id'), ]),
	(pt_map.models.Frequency, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Trip, 'trip_id'), ]),
	(pt_map.models.Transfer, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Stop, 'from_stop_id'), (pt_map.models.Stop, 'to_stop_id'), (pt_map.models.Route, 'from_route_id'), (pt_map.models.Route, 'to_route_id'), (pt_map.models.Trip, 'from_trip_id'), (pt_map.models.Trip, 'to_trip_id'), ]),
	(pt_map.models.Pathway, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Stop, 'from_stop_id'), (pt_map.models.Stop, 'to_stop_id'), ]),
	(pt_map.models.BookingRule, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Trip, 'trip_id'), ]),
	(pt_map.models.Translation, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Attribution, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Agency, 'agency_id'), (pt_map.models.Route, 'route_id'), (pt_map.models.Trip, 'trip_id'), ]),
	(pt_map.models.LocationGroupStop, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.LocationGroup, 'location_group_id'), (pt_map.models.Stop, 'stop_id'), ]),
	(pt_map.models.Network, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.RouteNetwork, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Network, 'network_id'), (pt_map.models.Route, 'route_id'), ]),
	(pt_map.models.Area, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.StopArea, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Area, 'area_id'), (pt_map.models.Stop, 'stop_id'), ]),
	(pt_map.models.FareMedium, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.FareProduct, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.Timeframe, [(pt_map.models.FeedInfo, 'feed_info_id'),]),
	(pt_map.models.FareLegRule, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.Network, 'network_id'), (pt_map.models.Area, 'from_area_id'), (pt_map.models.Area, 'to_area_id'), (pt_map.models.Timeframe, 'from_timeframe_group_id'), (pt_map.models.Timeframe, 'to_timeframe_group_id'), (pt_map.models.FareProduct, 'fare_product_id'), ]),
	(pt_map.models.FareTransferRule, [(pt_map.models.FeedInfo, 'feed_info_id'),(pt_map.models.FareProduct, 'fare_product_id'), ]),
]

fk_dict = {fk[0]: fk[1] for fk in foreign_keys}