Skip to content

Commit d11e4ce

Browse files
committed
Address review comments
1 parent 03acc2f commit d11e4ce

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
@@ -248,11 +248,11 @@ def count_signal_time_types(source_signals: List[SourceSignalSet]) -> Tuple[int,
248248
"""
249249
weekly = 0
250250
daily = 0
251-
for set in source_signals:
252-
if set.signal == True:
251+
for ssset in source_signals:
252+
if ssset.signal == True:
253253
continue
254-
for s in set.signal:
255-
signal = data_signals_by_key.get((set.source, s))
254+
for s in ssset.signal:
255+
signal = data_signals_by_key.get((ssset.source, s))
256256
if not signal:
257257
continue
258258
if signal.time_type == TimeType.week:
@@ -265,18 +265,18 @@ def count_signal_time_types(source_signals: List[SourceSignalSet]) -> Tuple[int,
265265
def create_source_signal_alias_mapper(source_signals: List[SourceSignalSet]) -> Tuple[List[SourceSignalSet], Optional[Callable[[str, str], str]]]:
266266
alias_to_data_sources: Dict[str, List[DataSource]] = {}
267267
transformed_sets: List[SourceSignalSet] = []
268-
for set in source_signals:
269-
source = data_source_by_id.get(set.source)
268+
for ssset in source_signals:
269+
source = data_source_by_id.get(ssset.source)
270270
if not source or not source.uses_db_alias:
271-
transformed_sets.append(set)
271+
transformed_sets.append(ssset)
272272
continue
273273
# uses an alias
274274
alias_to_data_sources.setdefault(source.db_source, []).append(source)
275-
if set.signal is True:
275+
if ssset.signal is True:
276276
# list all signals of this source (*) so resolve to a plain list of all in this alias
277277
transformed_sets.append(SourceSignalSet(source.db_source, [s.signal for s in source.signals]))
278278
else:
279-
transformed_sets.append(SourceSignalSet(source.db_source, set.signal))
279+
transformed_sets.append(SourceSignalSet(source.db_source, ssset.signal))
280280

281281
if not alias_to_data_sources:
282282
# no alias needed

0 commit comments

Comments
 (0)