Skip to content

Commit da7bc69

Browse files
committed
adding protected signal coverage for endpoints previously missed
1 parent beeb737 commit da7bc69

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/server/endpoints/covidcast.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,7 @@ def gen(rows):
235235
def handle_export():
236236
source, signal = request.values.get("signal", "jhu-csse:confirmed_incidence_num").split(":")
237237
source_signal_sets = [SourceSignalSet(source, [signal])]
238+
source_signal_sets = restrict_by_roles(source_signal_sets)
238239
daily_signals, weekly_signals = count_signal_time_types(source_signal_sets)
239240
source_signal_sets, alias_mapper = create_source_signal_alias_mapper(source_signal_sets)
240241
start_time_set = parse_day_or_week_arg("start_day", 202001 if weekly_signals > 0 else 20200401)
@@ -317,9 +318,10 @@ def handle_backfill():
317318
example query: http://localhost:5000/covidcast/backfill?signal=fb-survey:smoothed_cli&time=day:20200101-20220101&geo=state:ny&anchor_lag=60
318319
"""
319320
require_all(request, "geo", "time", "signal")
320-
source_signal_set = parse_single_source_signal_arg("signal")
321-
daily_signals, weekly_signals = count_signal_time_types([source_signal_set])
322-
source_signal_sets, _ = create_source_signal_alias_mapper([source_signal_set])
321+
source_signal_sets = [parse_single_source_signal_arg("signal")]
322+
source_signal_sets = restrict_by_roles(source_signal_sets)
323+
daily_signals, weekly_signals = count_signal_time_types(source_signal_sets)
324+
source_signal_sets, _ = create_source_signal_alias_mapper(source_signal_sets)
323325
# don't need the alias mapper since we don't return the source
324326

325327
time_set = parse_single_time_arg("time")

0 commit comments

Comments
 (0)