add parser & fix some whitespaces

This commit is contained in:
Yannik Schmidt
2022-01-29 16:04:45 +01:00
parent d2485c5d2b
commit d200e1660b

12
main.py
View File

@@ -1,5 +1,6 @@
#!/usr/bin/python3 #!/usr/bin/python3
import argparse
import datetime as dt import datetime as dt
import dateutil.relativedelta import dateutil.relativedelta
import os import os
@@ -150,16 +151,25 @@ class Data:
return "Data: {} {} {}".format(self.dtype, self.time, self.value) return "Data: {} {} {}".format(self.dtype, self.time, self.value)
if __name__ == "__main__": if __name__ == "__main__":
# parse arguments #
parser = argparse.ArgumentParser(description='Reference Data Collector DWD')
parser.add_argument('--target-file', default="Weatherdata.xlsx", help='File(-path) to save to')
args = parser.parse_args()
# check laste months #
checkLastMonths() checkLastMonths()
# read in csv's #
globPattern = "{}/*.csv".format(CSV_DIR) globPattern = "{}/*.csv".format(CSV_DIR)
sheets = {} sheets = {}
for f in glob.glob(globPattern): for f in glob.glob(globPattern):
sheet = pyexcel.get_sheet(file_name=f, delimiter=";") sheet = pyexcel.get_sheet(file_name=f, delimiter=";")
sheets.update({ os.path.basename(f) : sheet }) sheets.update({ os.path.basename(f) : sheet })
# open & save spreadsheet book #
book = pyexcel.get_book(bookdict=sheets) book = pyexcel.get_book(bookdict=sheets)
outfileRaw = "Wetterdaten.xlsx" outfileRaw = args.target_file
book.save_as(outfileRaw) book.save_as(outfileRaw)
# formating and style # # formating and style #