commit 8334ca580b7b68d8411f8cd92d9a2d0c24241763
author Chris Danis <cdanis@wikimedia.org> Mon Jan 11 13:16:36 2021 -0500
committer Chris Danis <cdanis@wikimedia.org> Mon Jan 11 13:54:58 2021 -0500
tree 5f83c3e390b3739b916f3247206536ed26b5fa39
parent 32783f01cd2b41d55c856e8043a3d15c7948b601

Parse the team_ids filter from an environment variable.

Change-Id: Ibd67bb4525db5e20a3d729c43e6718a3561758cc

klaxon/__init__.py

diff --git a/klaxon/__init__.py b/klaxon/__init__.py
index 8344e94..1bd5730 100644
--- a/klaxon/__init__.py
+++ b/klaxon/__init__.py
@@ -40,6 +40,7 @@
     'KLAXON_VO_CREATE_INCIDENT_URL': None,
     'KLAXON_SECRET_KEY': None,
     'KLAXON_ADMIN_CONTACT_EMAIL': None,
+    'KLAXON_TEAM_IDS_FILTER': None,     # A comma-separated list of team IDs, or unset.
 }
 
 
@@ -67,10 +68,17 @@
     api_cache = cachetools.TTLCache(1, float(app.config['KLAXON_INCIDENT_LIST_CACHE_TTL_SECONDS']))
     api_lock = threading.RLock()
 
+    team_ids = app.config['KLAXON_TEAM_IDS_FILTER']
+    if team_ids:
+        team_ids = set(team_ids.split(','))
+    else:
+        team_ids = set()
+
     vo = VictorOps(api_id=app.config['KLAXON_VO_API_ID'], api_key=app.config['KLAXON_VO_API_KEY'],
                    create_incident_url=app.config['KLAXON_VO_CREATE_INCIDENT_URL'],
                    repository=app.config['KLAXON_REPOSITORY'],
-                   admin_email=app.config['KLAXON_ADMIN_CONTACT_EMAIL'])
+                   admin_email=app.config['KLAXON_ADMIN_CONTACT_EMAIL'],
+                   team_ids=team_ids)
 
     @cachetools.cached(api_cache, lock=api_lock)
     def fetch_victorops():