counterweight/data-xexe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

properly check that output is dwh if not file

Browse source
This commit is contained in:
Pablo Martin 2024-06-13 15:59:13 +02:00
parent aa0919c790
commit bcd9993723
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
xexe/inputs_handling.py
View file

@ -40,11 +40,12 @@ def handle_get_rates_inputs(
if rates_source not in RATES_SOURCES:
raise ValueError(f"--rates-source must be one of {RATES_SOURCES.keys()}.")
# The Path constructor is idempotent, so this works equally fine if output
# is a string or an actual Path object.
output = pathlib.Path(output)
if output.suffix != ".csv":
raise ValueError("Output must be a .csv file.")
if not output == "dwh":
# The Path constructor is idempotent, so this works equally fine if output
# is a string or an actual Path object.
output = pathlib.Path(output)
if output.suffix != ".csv":
raise ValueError("File output must be a .csv file.")
prepared_inputs = {
"date_range": date_range,