added load email option

Author: Lisheng-Gao

src/acquisition/quidel/quidel.py

@@ -80,7 +80,7 @@ def measurement_to_ts(m,index,startweek=None,endweek=None):
     return res
 
 class QuidelData:
-    def __init__(self, raw_path):
+    def __init__(self, raw_path, load_email=True):
         self.data_path = raw_path
         self.excel_uptodate_path = join(raw_path,'excel/uptodate')
         self.excel_history_path = join(raw_path,'excel/history')
@@ -104,7 +104,8 @@ def __init__(self, raw_path):
         ]
         self.fields_to_keep = ['fac_id','fluA','fluB','fluAll']
         self.dims_to_keep = [self.fields.index(x) for x in self.fields_to_keep]
-        self.retrieve_excels()
+        if load_email:
+            self.retrieve_excels()
         self.prepare_csv()
 
     def retrieve_excels(self):

src/acquisition/quidel/quidel_update.py

@@ -50,9 +50,9 @@
 LOCATIONS = ['hhs%d'%i for i in range(1,11)]
 DATAPATH = '/home/automation/quidel_data'
 
-def update(locations, first=None, last=None, force_update=False):
+def update(locations, first=None, last=None, force_update=False, load_email=True):
   # download and prepare data first
-  qd = quidel.QuidelData(DATAPATH)
+  qd = quidel.QuidelData(DATAPATH,load_email)
   if not qd.need_update and not force_update:
     print('Data not updated, nothing needs change.')
     return
@@ -126,10 +126,12 @@ def main():
   parser.add_argument('--first', '-f', default=None, type=int, help='first epiweek override')
   parser.add_argument('--last', '-l', default=None, type=int, help='last epiweek override')
   parser.add_argument('--force_update', '-u', action='store_true', help='force update db values')
+  parser.add_argument('--skip_email', '-s', action='store_true', help='skip email downloading step')
   args = parser.parse_args()
 
   # sanity check
-  first, last, force_update = args.first, args.last, args.force_update
+  first, last, force_update, skip_email = args.first, args.last, args.force_update, args.skip_email
+  load_email = not skip_email
   if first is not None:
     flu.check_epiweek(first)
   if last is not None:
@@ -144,7 +146,7 @@ def main():
     locations = args.location.lower().split(',')
 
   # run the update
-  update(locations, first, last, force_update)
+  update(locations, first, last, force_update, load_email)
 
 
 if __name__ == '__main__':