diff options
Diffstat (limited to 'app/models/import/gtfs.rb')
| -rw-r--r-- | app/models/import/gtfs.rb | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/app/models/import/gtfs.rb b/app/models/import/gtfs.rb index 5b332315d..75ff45f8b 100644 --- a/app/models/import/gtfs.rb +++ b/app/models/import/gtfs.rb @@ -111,6 +111,8 @@ class Import::Gtfs < Import::Base vehicle_journey.published_journey_name = trip.headsign.presence || trip.id save vehicle_journey + vehicle_journey.time_tables << referential.time_tables.find(time_tables_by_service_id[trip.service_id]) + vehicle_journey_by_trip_id[trip.id] = vehicle_journey.id end end @@ -143,6 +145,24 @@ class Import::Gtfs < Import::Base end end + def time_tables_by_service_id + @time_tables_by_service_id ||= {} + end + + def import_calendars + source.calendars.each do |calendar| + time_table = referential.time_tables.build comment: "Calendar #{calendar.service_id}" + Chouette::TimeTable.all_days.each do |day| + time_table.send("#{day}=", calendar.send(day)) + end + time_table.periods.build period_start: calendar.start_date, period_end: calendar.end_date + + save time_table + + time_tables_by_service_id[calendar.service_id] = time_table.id + end + end + def save(model) unless model.save Rails.logger.info "Can't save #{model.class.name} : #{model.errors.inspect}" |