File tree Expand file tree Collapse file tree 2 files changed +12
-5
lines changed
treeherder/perf/auto_perf_sheriffing Expand file tree Collapse file tree 2 files changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,7 @@ services:
32
32
- BUG_FILER_API_KEY=${BUG_FILER_API_KEY:-}
33
33
- TLS_CERT_PATH=${TLS_CERT_PATH:-}
34
34
- TELEMETRY_ENABLE_ALERTS=${TELEMETRY_ENABLE_ALERTS:-}
35
+ - GCLOUD_PROJECT=${GCLOUD_PROJECT:-}
35
36
entrypoint : ' ./docker/entrypoint.sh'
36
37
# We *ONLY* initialize the data when we're running the backend
37
38
command : ' ./initialize_data.sh ./manage.py runserver 0.0.0.0:8000'
Original file line number Diff line number Diff line change @@ -250,15 +250,17 @@ def telemetry_alert(self):
250
250
import mozdetect
251
251
from mozdetect .telemetry_query import get_metric_table
252
252
253
- if (
254
- not os .environ .get ("GOOGLE_APPLICATION_CREDENTIALS" )
255
- and settings .SITE_HOSTNAME == "treeherder.mozilla.org"
256
- ):
253
+ if not os .environ .get ("GOOGLE_APPLICATION_CREDENTIALS" ) and self ._is_prod ():
257
254
raise Exception (
258
255
"GOOGLE_APPLICATION_CREDENTIALS must be defined in production. "
259
256
"Use GCLOUD_DIR for local testing."
260
257
)
261
258
259
+ project = "mozdata"
260
+ if self ._is_prod ():
261
+ # Defined from the GCLOUD_PROJECT env variable
262
+ project = None
263
+
262
264
ts_detectors = mozdetect .get_timeseries_detectors ()
263
265
264
266
metric_definitions = self ._get_metric_definitions ()
@@ -287,6 +289,7 @@ def telemetry_alert(self):
287
289
platform ,
288
290
android = (platform == "Mobile" ),
289
291
use_fog = True ,
292
+ project = project ,
290
293
)
291
294
if data .empty :
292
295
logger .info ("No data found" )
@@ -307,8 +310,11 @@ def telemetry_alert(self):
307
310
except Exception :
308
311
logger .info (f"Failed: { traceback .format_exc ()} " )
309
312
313
+ def _is_prod (self ):
314
+ return settings .SITE_HOSTNAME == "treeherder.mozilla.org"
315
+
310
316
def _can_run_telemetry (self ):
311
- return time (22 , 0 ) <= datetime .utcnow ().time () < time (23 , 0 )
317
+ return not self . _is_prod () or ( time (22 , 0 ) <= datetime .utcnow ().time () < time (23 , 0 ) )
312
318
313
319
def _create_detection_alert (
314
320
self ,
You can’t perform that action at this time.
0 commit comments