Generalize clear_date.py to handle a range of dates

Author: drw

clear_date.py

@@ -2,6 +2,7 @@
 table and SQLite files) for a single given date."""
 import sys
 from util.sqlite_util import clear_cache_for_date
+from datetime import timedelta
 from dateutil import parser
 
 from parameters.local_parameters import path
@@ -15,5 +16,17 @@
     print("Attempting to clear the SQLite cache for date = {}".format(date_i))
     clear_cache_for_date(path,reference_time,date_i)
     print("SQLite cache cleared for {}".format(date_i))
+elif len(sys.argv) == 3:
+    first_dt = parser.parse(sys.argv[1])
+    first_date_i = first_dt.date()
+    last_dt = parser.parse(sys.argv[2])
+    last_date_i = last_dt.date()
+    print("Attempting to clear the SQLite cache for dates {} through {}".format(first_date_i,last_date_i))
+    date_i = first_date_i
+    while date_i <= last_date_i:
+        clear_cache_for_date(path,reference_time,date_i)
+        print("     SQLite cache cleared for {}".format(date_i))
+        date_i += timedelta(days=1)
+    print("SQLite cache cleared for {} through {}".format(first_date_i,last_date_i))
 else:
     raise ValueError("Incorrect number of command-line parameters found.")