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: if rates_source not in RATES_SOURCES:
raise ValueError(f"--rates-source must be one of {RATES_SOURCES.keys()}.") raise ValueError(f"--rates-source must be one of {RATES_SOURCES.keys()}.")
# The Path constructor is idempotent, so this works equally fine if output if not output == "dwh":
# is a string or an actual Path object. # The Path constructor is idempotent, so this works equally fine if output
output = pathlib.Path(output) # is a string or an actual Path object.
if output.suffix != ".csv": output = pathlib.Path(output)
raise ValueError("Output must be a .csv file.") if output.suffix != ".csv":
raise ValueError("File output must be a .csv file.")
prepared_inputs = { prepared_inputs = {
"date_range": date_range, "date_range": date_range,