Merge pull request #2141 from alphagov/update-rake-task

Allow to run etl:rerun_main for a single date
alphagov · Oct 23, 2024 · dff14d5 · dff14d5
2 parents d3998ac + 508414b
commit dff14d5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/tasks/etl.rake b/lib/tasks/etl.rake
@@ -80,12 +80,13 @@ namespace :etl do
     end
   end
 
-  desc "Run ETL Main process across a range of dates"
+  desc "Run ETL Main process across a range of dates or a single date"
   task :rerun_main, %i[from to] => [:environment] do |_t, args|
     from = args[:from].to_date
-    to = args[:to].to_date
+    to = args[:to]&.to_date
     date_range = (from..to)
-    date_range.each do |date|
+
+    date_range.compact.each do |date|
       puts "Running Etl::Main process for #{date}"
       unless Etl::Main::MainProcessor.process(date:)
         abort("Etl::Main::MainProcessor failed")