From 868b50e5b63e708a46cf33e3fc1626dece02ff1e Mon Sep 17 00:00:00 2001 From: Alsia Plybeah Date: Wed, 10 Jul 2024 11:11:08 -0400 Subject: [PATCH] update config call --- analytics/config.py | 5 ++++- analytics/src/analytics/cli.py | 4 ++-- analytics/tests/integrations/test_slack.py | 4 ++-- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/analytics/config.py b/analytics/config.py index a52d5d25c..7bc91adc3 100644 --- a/analytics/config.py +++ b/analytics/config.py @@ -17,4 +17,7 @@ class DBSettings(PydanticBaseEnvConfig): password: str = Field(alias="DB_PASSWORD") ssl_mode: str = Field(alias="DB_SSL_MODE") slack_bot_token: str = Field(alias="ANALYTICS_SLACK_BOT_TOKEN") - reporting_channel_id: str = Field(alias="ANALYTICS_REPORTING_CHANNEL_ID") \ No newline at end of file + reporting_channel_id: str = Field(alias="ANALYTICS_REPORTING_CHANNEL_ID") + +def get_db_settings() -> DBSettings: + return DBSettings() \ No newline at end of file diff --git a/analytics/src/analytics/cli.py b/analytics/src/analytics/cli.py index 841cdd5a6..68d49d48e 100644 --- a/analytics/src/analytics/cli.py +++ b/analytics/src/analytics/cli.py @@ -241,9 +241,9 @@ def show_and_or_post_results( """Optionally show the results of a metric and/or post them to slack.""" # defer load of settings until this command is called # this prevents an error if ANALYTICS_SLACK_BOT_TOKEN env var is unset - from config import DBSettings + from config import get_db_settings - settings = DBSettings() + settings = get_db_settings() # optionally display the burndown chart in the browser if show_results: diff --git a/analytics/tests/integrations/test_slack.py b/analytics/tests/integrations/test_slack.py index 56dbb2e99..32ec6b6bc 100644 --- a/analytics/tests/integrations/test_slack.py +++ b/analytics/tests/integrations/test_slack.py @@ -5,9 +5,9 @@ from slack_sdk import WebClient from analytics.integrations.slack import FileMapping, SlackBot -from config import DBSettings +from config import get_db_settings -settings = DBSettings() +settings = get_db_settings() client = WebClient(token=settings.slack_bot_token)