Skip to content

Commit 595a33b

Browse files
authored
Merge pull request #5 from LimnoTech/time-query-fix-limno
update time querying
2 parents 33c5e25 + 9bae1f7 commit 595a33b

File tree

1 file changed

+7
-12
lines changed

1 file changed

+7
-12
lines changed

pygeoapi/provider/xarray_edr.py

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -124,23 +124,18 @@ def position(self, **kwargs):
124124
data = self._data
125125

126126
if self.time_field in query_params:
127-
remaining_query = {
128-
key: val for key, val in query_params.items()
129-
if key != self.time_field
130-
}
131127
if isinstance(query_params[self.time_field], slice):
128+
remaining_query = {
129+
key: val for key, val in query_params.items()
130+
if key != self.time_field
131+
}
132132
time_query = {
133133
self.time_field: query_params[self.time_field]
134134
}
135+
data = data.sel(
136+
time_query).sel(remaining_query, method='nearest')
135137
else:
136-
time_query = {
137-
self.time_field: (
138-
data[self.time_field].dt.date ==
139-
query_params[self.time_field]
140-
)
141-
}
142-
data = data.sel(
143-
time_query).sel(remaining_query, method='nearest')
138+
data = data.sel(query_params, method='nearest')
144139
else:
145140
data = data.sel(query_params, method='nearest')
146141
except KeyError:

0 commit comments

Comments
 (0)