fixing scheduler for precription summary refresh cron (#5777)

**Story card:**
[SIMPLEBACK-77](SIMPLEBACK_77_fix_prescription_summary_refresh_job)

## Because

Scheduler for refreshing reporting_patient_prescriptions is expected to
do a refresh of last 3 months dynamically. But it was scheduled to run
for fixed dates

## This addresses

From the scheduler we are passing how many months in the past we want a
refresh for, which will be evaluated in the rake task to perform a
refresh

## Test instructions

rake reporting:refresh_partitioned_table[#{reporting_table_name},
#{months_back}]
This should perform a refresh for the table passed as arg[1], for those
many months in the past as passed under arg[2]

---------

Co-authored-by: Ayushi Agrawal <ayushiagrawal@RTSL-P172G93770.local>

Author: aagrawalrtsl

config/schedule.rb

@@ -140,8 +140,7 @@ def local(time)
 end
 
 every 1.month, at: local("06:00 am"), roles: [:cron] do
-  refresh_months = (0..3).map { |i| Date.current.beginning_of_month - i.months }.reverse
-  rake "reporting:refresh_partitioned_table[reporting_patient_prescriptions, #{refresh_months.first}, #{refresh_months.last}]"
+  rake "reporting:refresh_partitioned_table[reporting_patient_prescriptions, 3]"
 end
 
 every :day, at: local("05:00 am"), roles: [:cron] do

lib/tasks/reporting.rake

@@ -47,26 +47,33 @@ namespace :reporting do
     end
   end
 
-  desc "Refresh a partitoned table for given months"
-  task :refresh_partitioned_table, [:table_name, :start_date, :end_date] => :environment do |_, args|
-    unless args[:table_name].present? && args[:start_date].present? && args[:end_date].present?
-      puts "ERROR: table_name, start_date and end_date are required."
-      puts "Usage: rake reporting:refresh_partitioned_table['reporting_patient_states', '2024-01-01' ,'2024-02-01']"
+  desc "Refresh a partitioned table for N months in the past"
+  task :refresh_partitioned_table, [:table_name, :months_back] => :environment do |_, args|
+    unless args[:table_name].present?
+      puts "ERROR: table_name is required."
+      puts "Usage: rake reporting:refresh_partitioned_table[reporting_patient_states, 6]"
+      puts "       rake reporting:refresh_partitioned_table[reporting_patient_states]"
       exit 1
     end
-
     table_name = args[:table_name]
-    start_date = Date.parse(args[:start_date]).beginning_of_month
-    end_date = Date.parse(args[:end_date]).beginning_of_month
-    months = []
-    current = start_date
-    while current <= end_date
-      months << current.strftime("%Y-%m-%d")
-      current = current.next_month
+    months_back = args[:months_back].present? ? args[:months_back].to_i : 12
+    if months_back < 0
+      puts "ERROR: months_back must be >= 0"
+      exit 1
+    end
+    current_month = Date.today.beginning_of_month
+    start_month = current_month - months_back.months
+    refresh_months = []
+    while start_month <= current_month
+      refresh_months << start_month
+      start_month = start_month.next_month
     end
-    months.each do |month|
-      puts "Refreshing partitioned table: #{table_name} for month: #{month}"
-      ActiveRecord::Base.connection.exec_query("CALL simple_reporting.add_shard_to_table('#{month}', '#{table_name}')")
+    refresh_months.each do |month|
+      formatted_month = month.strftime("%Y-%m-%d")
+      puts "Refreshing partitioned table: #{table_name} for month: #{formatted_month}"
+      ActiveRecord::Base.connection.exec_query(
+        "CALL simple_reporting.add_shard_to_table('#{formatted_month}', '#{table_name}')"
+      )
     end
   end
 end