Handle stray dates

transitnownash · Mar 16, 2024 · 24ede57 · 24ede57
1 parent 348bec6
commit 24ede57
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lib/tasks/import.rake b/lib/tasks/import.rake
@@ -78,10 +78,14 @@ namespace :import do
     puts 'calendar_dates.txt'
     begin
       @source = GTFS::Source.build ENV.fetch('GTFS_URL', nil)
-      CalendarDate.bulk_insert do |worker|
-        @source.each_calendar_date do |row|
-          worker.add(CalendarDate.hash_from_gtfs(row))
+      @source.each_calendar_date do |row|
+        date = CalendarDate.new(CalendarDate.hash_from_gtfs(row))
+        if date.calendar_id.nil?
+          puts "[ERROR] Calendar ID is nil for #{date.service_id} / #{date.date}"
+          next
         end
+
+        date.save
       end
     rescue Errno::ENOENT
       puts '[warning] File does not exist.'