Skip to content

Commit 888897d

Browse files
committed
Address review comments
1 parent 78be18e commit 888897d

File tree

2 files changed

+22
-22
lines changed

2 files changed

+22
-22
lines changed

src/server/_query.py

+13-13
Original file line numberDiff line numberDiff line change
@@ -126,12 +126,12 @@ def filter_geo_sets(
126126
returns the SQL sub query to filter by the given geo sets
127127
"""
128128

129-
def filter_set(set: GeoSet, i) -> str:
129+
def filter_set(gset: GeoSet, i) -> str:
130130
type_param = f"{param_key}_{i}t"
131-
params[type_param] = set.geo_type
132-
if isinstance(set.geo_values, bool) and set.geo_values:
131+
params[type_param] = gset.geo_type
132+
if isinstance(gset.geo_values, bool) and gset.geo_values:
133133
return f"{type_field} = :{type_param}"
134-
return f"({type_field} = :{type_param} AND {filter_strings(value_field, cast(Sequence[str], set.geo_values), type_param, params)})"
134+
return f"({type_field} = :{type_param} AND {filter_strings(value_field, cast(Sequence[str], gset.geo_values), type_param, params)})"
135135

136136
parts = [filter_set(p, i) for i, p in enumerate(values)]
137137

@@ -153,12 +153,12 @@ def filter_source_signal_sets(
153153
returns the SQL sub query to filter by the given source signal sets
154154
"""
155155

156-
def filter_set(set: SourceSignalSet, i) -> str:
156+
def filter_set(ssset: SourceSignalSet, i) -> str:
157157
source_param = f"{param_key}_{i}t"
158-
params[source_param] = set.source
159-
if isinstance(set.signal, bool) and set.signal:
158+
params[source_param] = ssset.source
159+
if isinstance(ssset.signal, bool) and ssset.signal:
160160
return f"{source_field} = :{source_param}"
161-
return f"({source_field} = :{source_param} AND {filter_strings(signal_field, cast(Sequence[str], set.signal), source_param, params)})"
161+
return f"({source_field} = :{source_param} AND {filter_strings(signal_field, cast(Sequence[str], ssset.signal), source_param, params)})"
162162

163163
parts = [filter_set(p, i) for i, p in enumerate(values)]
164164

@@ -172,23 +172,23 @@ def filter_set(set: SourceSignalSet, i) -> str:
172172
def filter_time_set(
173173
type_field: str,
174174
time_field: str,
175-
set: Optional[TimeSet],
175+
tset: Optional[TimeSet],
176176
param_key: str,
177177
params: Dict[str, Any],
178178
) -> str:
179179
"""
180180
returns the SQL sub query to filter by the given time set
181181
"""
182182
# safety path; should normally not be reached as time sets are enforced by the API
183-
if not set:
183+
if not tset:
184184
return "FALSE"
185185

186186
type_param = f"{param_key}_0t"
187-
params[type_param] = set.time_type
188-
if isinstance(set.time_values, bool) and set.time_values:
187+
params[type_param] = tset.time_type
188+
if isinstance(tset.time_values, bool) and tset.time_values:
189189
parts = f"{type_field} = :{type_param}"
190190
else:
191-
ranges = set.to_ranges().time_values
191+
ranges = tset.to_ranges().time_values
192192
parts = f"({type_field} = :{type_param} AND {filter_integers(time_field, ranges, type_param, params)})"
193193

194194
return f"({parts})"

src/server/endpoints/covidcast_utils/model.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -243,11 +243,11 @@ def count_signal_time_types(source_signals: List[SourceSignalSet]) -> Tuple[int,
243243
"""
244244
weekly = 0
245245
daily = 0
246-
for set in source_signals:
247-
if set.signal == True:
246+
for ssset in source_signals:
247+
if ssset.signal == True:
248248
continue
249-
for s in set.signal:
250-
signal = data_signals_by_key.get((set.source, s))
249+
for s in ssset.signal:
250+
signal = data_signals_by_key.get((ssset.source, s))
251251
if not signal:
252252
continue
253253
if signal.time_type == TimeType.week:
@@ -260,18 +260,18 @@ def count_signal_time_types(source_signals: List[SourceSignalSet]) -> Tuple[int,
260260
def create_source_signal_alias_mapper(source_signals: List[SourceSignalSet]) -> Tuple[List[SourceSignalSet], Optional[Callable[[str, str], str]]]:
261261
alias_to_data_sources: Dict[str, List[DataSource]] = {}
262262
transformed_sets: List[SourceSignalSet] = []
263-
for set in source_signals:
264-
source = data_source_by_id.get(set.source)
263+
for ssset in source_signals:
264+
source = data_source_by_id.get(ssset.source)
265265
if not source or not source.uses_db_alias:
266-
transformed_sets.append(set)
266+
transformed_sets.append(ssset)
267267
continue
268268
# uses an alias
269269
alias_to_data_sources.setdefault(source.db_source, []).append(source)
270-
if set.signal is True:
270+
if ssset.signal is True:
271271
# list all signals of this source (*) so resolve to a plain list of all in this alias
272272
transformed_sets.append(SourceSignalSet(source.db_source, [s.signal for s in source.signals]))
273273
else:
274-
transformed_sets.append(SourceSignalSet(source.db_source, set.signal))
274+
transformed_sets.append(SourceSignalSet(source.db_source, ssset.signal))
275275

276276
if not alias_to_data_sources:
277277
# no alias needed

0 commit comments

Comments
 (0)