123456789101112131415161718 |
- from argparse import ArgumentParser
- from util import gettemp, datevalid
- import pandas as p
-
-
- def main():
- parser = ArgumentParser(description='Download kwh data from dataframe')
- parser.add_argument("-o", "--output", dest="output", help = "output pickle path", metavar="PATH", required = True)
- parser.add_argument("-s", "--start-date", dest = "startdate", help = "start date for download; format: YYYY-MM-DD; default: 2016-04-01", metavar="DATE", default = "2016-04-01", type = datevalid)
- parser.add_argument("-e", "--end-date", dest = "enddate", help = "end date for download; format: YYYY-MM-DD; default: 2019-01-01", metavar="DATE", default = "2019-01-01", type = datevalid)
- parser.add_argument("--station", dest = "station", help = "weather station to get data from; default: 2006", metavar="STATION", default = "2006")
- parser.add_argument("-v", "--verbose", dest = "verbose", action ="store_true")
- args = parser.parse_args()
- tempdata = gettemp(args.startdate.strftime("%Y-%m-%d"), args.enddate.strftime("%Y-%m-%d"), args.station, args.verbose)
- tempdata.to_pickle(args.output)
-
- if __name__ == "__main__":
- main()
|