Skip to content
Snippets Groups Projects
Commit dfcac50b authored by Cain, Payton David's avatar Cain, Payton David Committed by Jeremy M Fee
Browse files

Refine defaults/return values

parent 24057929
No related branches found
No related tags found
No related merge requests found
......@@ -65,7 +65,7 @@ class DataApiQuery(BaseModel):
elements: List[str] = DEFAULT_ELEMENTS
sampling_period: SamplingPeriod = SamplingPeriod.MINUTE
data_type: Union[DataType, str] = DataType.VARIATION
dbdt: list = []
dbdt: List[str] = []
format: OutputFormat = OutputFormat.IAGA2002
@validator("data_type")
......@@ -125,11 +125,11 @@ class DataApiQuery(BaseModel):
return endtime
@validator("dbdt", always=True)
def validate_dbdt(cls, dbdt: list,) -> list:
def validate_dbdt(cls, dbdt: List[str],) -> List[str]:
"""Default dbdt based on valid elements.
"""
for channel in dbdt:
if channel not in DEFAULT_ELEMENTS:
if channel not in ELEMENTS:
raise ValueError("Specified channel not found in valid elements.")
return dbdt
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment