AARETE\agupta
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
|
import streamlit as st
|
|
import security
|
|
import os
|
|
import constants
|
|
|
|
def setup_page(REDIRECT_URI):
|
|
# st.set_page_config(
|
|
# page_title=page_title,
|
|
# page_icon="👋",
|
|
# )
|
|
|
|
if st.query_params.get('code'):
|
|
security.handle_redirect(REDIRECT_URI)
|
|
|
|
access_token = st.session_state.get('access_token')
|
|
|
|
if access_token:
|
|
user_info = security.get_user_info(access_token)
|
|
st.session_state['user_info'] = user_info
|
|
return True
|
|
else:
|
|
st.write("Please sign-in to use this app.")
|
|
auth_url = security.get_auth_url(REDIRECT_URI)
|
|
st.markdown(f"<a href='{auth_url}' target='_self'>Sign In</a>", unsafe_allow_html=True)
|
|
st.stop()
|
|
|
|
def load_page_details(interface):
|
|
env_var = os.environ.get('ENVIRONMENT', 'DEV')
|
|
if env_var == 'UAT':
|
|
return (constants.DOCZY_REDIRECT_URL_UAT + str(interface), constants.DOCZY_CREATE_BATCH_URL_UAT, constants.DOCZY_PIPELINE_URL_UAT)
|
|
elif env_var == 'DEV':
|
|
return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, constants.DOCZY_PIPELINE_URL_DEV)
|
|
else:
|
|
return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, constants.DOCZY_PIPELINE_URL_DEV)
|