Skip to content

Commit

Permalink
Adjustments on parsing parameters in DWDStationRequest
Browse files Browse the repository at this point in the history
  • Loading branch information
gutzbenj committed Jun 11, 2020
1 parent f0a8b4a commit 301b825
Showing 1 changed file with 21 additions and 7 deletions.
28 changes: 21 additions & 7 deletions python_dwd/dwd_station_request.py
Expand Up @@ -38,15 +38,29 @@ def __init__(self,
except ValueError:
raise ValueError("List of station id's can not be parsed to integers.")

self.parameter = parameter if isinstance(parameter, Parameter) \
else _parse_parameter_from_value(parameter, PARAMETER_WORDLIST_MAPPING)
try:
self.parameter = Parameter(parameter)
except ValueError:
self.parameter = _parse_parameter_from_value(
parameter, PARAMETER_WORDLIST_MAPPING)

try:
self.time_resolution = TimeResolution(time_resolution)
except ValueError:
self.time_resolution = _parse_parameter_from_value(
time_resolution, TIMERESOLUTION_WORDLIST_MAPPING)

self.time_resolution = time_resolution if isinstance(time_resolution, TimeResolution) \
else _parse_parameter_from_value(time_resolution, TIMERESOLUTION_WORDLIST_MAPPING)
self.period_type = []
for pt in cast_to_list(period_type):
if pt is None:
self.period_type.append(None)
continue

self.period_type = cast_to_list(period_type) if isinstance(period_type, (PeriodType, type(None))) \
else [_parse_parameter_from_value(period_type, PERIODTYPE_WORDLIST_MAPPING)
for period_type in cast_to_list(period_type)]
try:
self.period_type.append(PeriodType(pt))
except ValueError:
self.period_type.append(
_parse_parameter_from_value(period_type, PERIODTYPE_WORDLIST_MAPPING))

# Additional sorting required for self.period_type to ensure that for multiple
# periods the data is first sourced from historical
Expand Down

0 comments on commit 301b825

Please sign in to comment.