brian/bandwidth_probing
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated to check for env var USER_ID

Browse Source
This commit is contained in:
brb
2023-07-12 13:46:01 +02:00
parent a456ccc536
commit caeace7907
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
code/initialise_app.py
+2 -4
View File
@@ -20,15 +20,14 @@ def initialise_app(
discord_service_name: str = DISCORD_SERVICE_NAME, discord_service_name: str = DISCORD_SERVICE_NAME,
discord_webhook_url: str = DISCORD_WEBHOOK_URL, discord_webhook_url: str = DISCORD_WEBHOOK_URL,
discord_logger_level: str = DISCORD_LOGGER_LEVEL discord_logger_level: str = DISCORD_LOGGER_LEVEL
) -> int: ) -> None:
""" """
Convienience function that ensures eveything is ready Convienience function that ensures eveything is ready
before running the main loop. before running the main loop.
""" """
# load environment variables # load environment variables
load_dotenv() load_dotenv()
user_id = int(os.getenv('USER_ID', default=None)) assert 'USER_ID' in os.environ, 'environment variable USER_ID must be specified'
assert user_id is not None, 'environment variable USER_ID must be specified'
logger_level = os.getenv( logger_level = os.getenv(
'LOGGER_LEVEL', 'LOGGER_LEVEL',
default=LOGGER_LEVEL default=LOGGER_LEVEL
@@ -61,7 +60,6 @@ def initialise_app(
discord_handler.setLevel(level=level) discord_handler.setLevel(level=level)
logger.addHandler(discord_handler) logger.addHandler(discord_handler)
logging.debug('finished') logging.debug('finished')
return user_id
if __name__ == '__main__': if __name__ == '__main__':