plot_mhz19.py: (optionally) pass filename and startdate

[toast/airingbutler.git] / plot_mhz19.py

diff --git a/plot_mhz19.py b/plot_mhz19.py
index 4d3b90ab0b01e77a1348160cc7e3aef96e1c6512..defb0da640ee66956c9bfb5b3b0ec764703c5c11 100755 (executable)
--- a/plot_mhz19.py
+++ b/plot_mhz19.py
@@ -5,8 +5,8 @@ import argparse
 from datetime import datetime, date, timedelta
 from matplotlib import pyplot as plt
 
 from datetime import datetime, date, timedelta
 from matplotlib import pyplot as plt
 

-filename = 'log.csv'

 '''
 '''

+filename = 'log.csv'

 Time,TS,CO2,Temp
 21:02:02,0,661,24
 21:02:08,5021,657,24
 Time,TS,CO2,Temp
 21:02:02,0,661,24
 21:02:08,5021,657,24


@@ -19,17 +19,19 @@ def parsedatetime(timevalue, currentdate):
 
 def main():
     parser = argparse.ArgumentParser('Plot log.csv.')
 
 def main():
     parser = argparse.ArgumentParser('Plot log.csv.')

-    parser.add_argument('--temperature', '-t', action='store_true', help='Include temperature plot')
+    parser.add_argument('filename', nargs='?', default='log.csv', help='Name of csv file')
+    parser.add_argument('-s', '--startdate', nargs='?', default='2020-10-22', help='First date in csv file')
+    parser.add_argument('-t', '--temperature', action='store_true', help='Include temperature plot')

     args = parser.parse_args()
     args = parser.parse_args()

+    startdate = datetime.strptime(args.startdate, '%Y-%m-%d').date()

 
 

-    with open(filename) as f:
+    with open(args.filename) as f:

         reader = csv.reader(f)
         headers = next(reader)
 
         time = []
         co2 = []
         temp = []
         reader = csv.reader(f)
         headers = next(reader)
 
         time = []
         co2 = []
         temp = []

-        startdate = date(2020, 10, 22)

         for row in reader:
             dt = parsedatetime(row[0], startdate)
             if len(time) > 0 and dt - time[-1] < timedelta(hours=-2): # DST switch!
         for row in reader:
             dt = parsedatetime(row[0], startdate)
             if len(time) > 0 and dt - time[-1] < timedelta(hours=-2): # DST switch!