Added Secret key from Secrets manager for SSO

This commit is contained in:
Umang Mistry
2024-04-15 14:24:59 -05:00
parent 33f86ae595
commit b2ba664150
+33 -1
View File
@@ -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)