Init new env variable in __init__

3rd-party-integrations · Apr 12, 2021 · ad941d2 · ad941d2
1 parent ee370b8
commit ad941d2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -9,12 +9,11 @@
 from apscheduler.triggers.cron import CronTrigger
 from flask import Flask
 
-from githubapp import GitHubApp, DirectoryClient, CRON_INTERVAL, TEST_MODE
+from githubapp import GitHubApp, DirectoryClient, CRON_INTERVAL, TEST_MODE, ADD_MEMBER
 
 app = Flask(__name__)
 github_app = GitHubApp(app)
 directory = DirectoryClient()
-addUserAsMember = os.getenv('ADD_MEMBER', 'False') == 'True'
 
 # Schedule a full sync
 scheduler = BackgroundScheduler(daemon=True)
@@ -182,7 +181,7 @@ def execute_sync(org, team, slug, state):
     else:
         for user in state["action"]["add"]:
             # Validate that user is in org
-            if org.is_member(user) or addUserAsMember:
+            if org.is_member(user) or ADD_MEMBER:
                 try:
                     print(f"Adding {user} to {slug}")
                     team.add_or_update_membership(user)

diff --git a/githubapp/__init__.py b/githubapp/__init__.py
@@ -37,3 +37,5 @@
     rootlogger.warn('TEST_MODE should be set to "true" or "false"')
     rootlogger.warn(e)
     TEST_MODE = False
+# Check if should add member to organization
+ADD_MEMBER = os.getenv("ADD_MEMBER", "False") == "True"