diff --git a/streamlit/security.py b/streamlit/security.py index 0701852..1ad59fd 100644 --- a/streamlit/security.py +++ b/streamlit/security.py @@ -1,10 +1,13 @@ import streamlit as st import msal import requests +import boto3 +from botocore.exceptions import ClientError + # Replace with your own values CLIENT_ID = 'effafe90-7ed7-43a3-ab03-19a0be2f1758' -CLIENT_SECRET = 'bjQ8Q~lpR2uBcGI34VDu16t73doz8Crj0YY_~dgD' +# CLIENT_SECRET = 'bjQ8Q~lpR2uBcGI34VDu16t73doz8Crj0YY_~dgD' # TENANT_ID = '' AUTHORITY = 'https://login.microsoftonline.com/organizations/' @@ -12,6 +15,35 @@ SCOPE = ['User.Read'] REDIRECT_URI = 'https://171.29.20.126:8501' + + + +def get_secret(): + + secret_name = "doczy-sso-azure-app-key" + region_name = "us-east-2" + + # Create a Secrets Manager client + session = boto3.session.Session() + client = session.client( + service_name='secretsmanager', + region_name=region_name + ) + + try: + get_secret_value_response = client.get_secret_value( + SecretId=secret_name + ) + except ClientError as e: + raise e + + secret = get_secret_value_response['SecretString'] + + return secret + + +CLIENT_SECRET = get_secret() + app = msal.ConfidentialClientApplication(CLIENT_ID, authority=AUTHORITY, client_credential=CLIENT_SECRET)