import csv

# Variables pour chaque entité
agencies_columns = ['agency_id', 'agency_name', 'agency_url', 'agency_timezone', 'agency_lang', 'agency_phone', 'agency_email']
services_columns = ['service_id', 'date', 'exception_type']
weekly_services_columns = ['service_id', 'monday', 'tuesday', 'wednesday', 'thursday', 'friday', 'saturday', 'sunday','start_date','end_date']
line_columns = ['pathway_id' ,'from_stop_id' ,'to_stop_id' ,'pathway_mode' ,'is_bidirectional' ,'length', 'traversal_time','stair_count','max_slope','min_width','signposted_as','reversed_signposted_as']
route_columns = ['route_id','agency_id' ,'route_short_name' ,'route_long_name' ,'route_desc' ,'route_type', 'route_url' ,'route_color' ,'route_text_color','route_sort_order']
stop_object_columns = ['object_id','object_system','object_code']
trip_line_columns = ['trip_id','arrival_time','departure_time','stop_id','stop_sequence', 'pickup_type','drop_off_type','local_zone_id','stop_headsign','timepoint']
trip_route_columns = ['route_id','service_id  ','trip_id  ','trip_headsign ','trip_short_name', 'direction_id  ','block_id ','shape_id ','wheelchair_accessible ','bikes_allowed']
transfer_columns = ['from_stop_id' ,'to_stop_id' ,'transfer_type' ,  'min_transfer_time']

stop_columns = ['stop_id ','stop_code ','stop_name ', 'stop_desc', 'stop_lon', 'stop_lat', 'zone id', 'stop_url' ,'location_type','parent_station','stop_timezone', 'level id','wheelchair_boarding','platform_code']

# Données pour chaque entité
agencies_data = [
    ['1', 'Agency 1', 'http://www.agency1.com', 'America/Los_Angeles', 'en', '+1234567890', '[email protected]'],
    ['2', 'Agency 2', 'http://www.agency2.com', 'Europe/Paris', 'fr', '+9876543210', '[email protected]']
]

services_data = [
    ['S1', '2022-01-01', 1],
    ['S2', '2022-01-02', 0]
]

weekly_services_data = [
    ['WS1', 1, 0, 0, 0, 0, 0, 1, '2022-01-01','2022-12-31'],
    ['WS2', 0, 1, 1, 1, 1, 0, 0, '2022-01-01','2023-12-31']
]

line_data = [
    ['L1','StopA','StopB','Bus','True','10.5','20','3','10%','3m','','']
]

route_data = [
    ['R1','A1','Route A Short Name','Route A Long Name','','Bus','','','#000000','#FFFFFF'],
    ['R2','A2','Route B Short Name','Route B Long Name','','Tram','','','#FF0000','#FFFFFF']
]

stop_object_data = [
    ['O1','System A','Code A'],
    ['O2','System B','Code B']
]

trip_line_data = [
    ['T1','08:00:00','08:05:00','StopA' ,1 ,'' ,'' ,'ZoneA' ,'DestinationA' ,'true'],
    ['T2','09:00:00','09:05:00','StopB' ,2 ,'' ,'' ,'ZoneB' ,'DestinationB' ,'false']
]

trip_route_data = [
    ['R1', 'S1', 'T1', 'DestinationA', '', '0', '', '', 'true', 'true'],
    ['R2', 'S2', 'T2', 'DestinationB', '', '1', '', '', 'false', 'false']
]

transfer_data = [
    ['StopA','StopB','0','300'],
    ['StopC','StopD','1','600']
]

stop_data = [
    ['StopA','','Station A','','0.000000','0.000000','','','','','',''],
    ['StopB','','Station B','','10.000000','-10.000000','','','','Station A','','']
]

# Créer le fichier CSV
with open('gtfs_data.csv', mode='w', newline='') as csv_file: 
	writer = csv.writer(csv_file) # Write the headers and data for each entity 
	writer.writerow(agencies_columns) 
	writer.writerows(agencies_data) 
	writer.writerow(services_columns) 
	writer.writerows(services_data) 
	writer.writerow(weekly_services_columns) 
	writer.writerows(weekly_services_data) 
	writer.writerow(line_columns) 
	writer.writerows(line_data) 
	writer.writerow(route_columns) 
	writer.writerows(route_data) 
	writer.writerow(stop_object_columns) 
	writer.writerows(stop_object_data) 
	writer.writerow(trip_line_columns) 
	writer.writerows(trip_line_data) 
	writer.writerow(trip_route_columns) 
	writer.writerows(trip_route_data) 
	writer.writerow(transfer_columns) 
	writer.writerows(transfer_data) 
	writer.writerow(stop_columns) 
	writer.writerows(stop_data) 
	print("CSV file 'gtfs_data.csv' has been created and populated with data.")

CopyRun