Compare commits

..

3 Commits

Author SHA1 Message Date
Felix Blanke
230c4298a4 Make sheet_name an arg 2023-08-27 00:59:42 +02:00
Felix Blanke
ec97edff6e Make name an arg 2023-08-27 00:58:35 +02:00
Felix Blanke
caededba88 Make folder an arg 2023-08-27 00:57:58 +02:00

View File

@ -139,6 +139,9 @@ def main(
grouped: bool = False, grouped: bool = False,
special_tag: str | None = None, special_tag: str | None = None,
no_processing: bool = False, no_processing: bool = False,
folder: str = "data",
name: str = "data",
sheet_name: str = "digital",
) -> None: ) -> None:
bez_data = get_bez_data([tag], url)[0] bez_data = get_bez_data([tag], url)[0]
df = construct_dataframe( df = construct_dataframe(
@ -151,11 +154,11 @@ def main(
if dry_run: if dry_run:
print(df) print(df)
else: else:
filename = f"data/{datetime.today().strftime('%Y-%m-%d')}_data.ods" filename = f"{folder}/{datetime.today().strftime('%Y-%m-%d')}_{name}.ods"
if Path(filename).exists(): if Path(filename).exists():
print("File already exists!") print("File already exists!")
else: else:
df.to_excel(filename, sheet_name="digital") df.to_excel(filename, sheet_name=sheet_name)
if __name__ == "__main__": if __name__ == "__main__":