From 05d2c2eb3d8ce1eaf041586702795944086a043b Mon Sep 17 00:00:00 2001 From: Bradley Reynolds Date: Thu, 26 Oct 2023 01:49:01 +0000 Subject: [PATCH] Test using .get_secret_value() Signed-off-by: GitHub --- src/loader/constants.py | 10 +++++----- src/loader/loader.py | 8 ++++---- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/src/loader/constants.py b/src/loader/constants.py index a61a444..bccdd19 100644 --- a/src/loader/constants.py +++ b/src/loader/constants.py @@ -7,7 +7,7 @@ """ -from pydantic_settings import BaseSettings, SettingsConfigDict +from pydantic_settings import BaseSettings, SecretStr, SettingsConfigDict class EnvConfig(BaseSettings): @@ -26,10 +26,10 @@ class _Dragonfly(EnvConfig, env_prefix="dragonfly"): base_url: str = "https://dragonfly.vipyrsec.com" auth0_domain: str = "vipyrsec.us.auth0.com" - client_id: str - client_secret: str - username: str - password: str + client_id: SecretStr + client_secret: SecretStr + username: SecretStr + password: SecretStr audience: str = "https://dragonfly.vipyrsec.com" diff --git a/src/loader/loader.py b/src/loader/loader.py index 6dc52e9..e2cf89e 100644 --- a/src/loader/loader.py +++ b/src/loader/loader.py @@ -9,10 +9,10 @@ def get_access_token(*, http_client: Client) -> str: """Get an access token from Auth0.""" payload = { - "client_id": Dragonfly.client_id, - "client_secret": Dragonfly.client_secret, - "username": Dragonfly.username, - "password": Dragonfly.password, + "client_id": Dragonfly.client_id.get_secret_value(), + "client_secret": Dragonfly.client_secret.get_secret_value(), + "username": Dragonfly.username.get_secret_value(), + "password": Dragonfly.password.get_secret_value(), "audience": Dragonfly.audience, "grant_type": "password", }