Fallback to server token if script token isn't provided

Author: azvoleff

gefcore/__init__.py

@@ -13,7 +13,8 @@
 handler = logging.StreamHandler(stream=sys.stdout)
 logger.addHandler(handler)
 
-rollbar_token = os.getenv("ROLLBAR_SCRIPT_TOKEN")
+# Use ROLLBAR_SCRIPT_TOKEN if set, otherwise fall back to ROLLBAR_SERVER_TOKEN
+rollbar_token = os.getenv("ROLLBAR_SCRIPT_TOKEN") or os.getenv("ROLLBAR_SERVER_TOKEN")
 env = os.getenv("ENV")
 if rollbar_token and env and env not in ("test", "testing"):
     rollbar.init(rollbar_token, env)

gefcore/runner.py

@@ -15,7 +15,9 @@
 except ImportError:
     main = None
 
-rollbar.init(os.getenv("ROLLBAR_SCRIPT_TOKEN"), os.getenv("ENV"))
+# Use ROLLBAR_SCRIPT_TOKEN if set, otherwise fall back to ROLLBAR_SERVER_TOKEN
+rollbar_token = os.getenv("ROLLBAR_SCRIPT_TOKEN") or os.getenv("ROLLBAR_SERVER_TOKEN")
+rollbar.init(rollbar_token, os.getenv("ENV"))
 
 # Silence warning about file_cache being unavailable. See more here:
 # https://github.com/googleapis/google-api-python-client/issues/299