Spaces:
Sleeping
Sleeping
File size: 1,329 Bytes
178b25e af73c80 178b25e af73c80 178b25e af73c80 178b25e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
# common.py
import extra_streamlit_components as stx
import streamlit as st
import logging
import os
from time import time
from requests_oauthlib import OAuth2Session
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger("__name__")
logger.debug("調査用ログ")
# 接続元制御
ALLOW_IP_ADDRESS = os.environ["ALLOW_IP_ADDRESS"]
# Azure AD app registration details
CLIENT_ID = os.environ["CLIENT_ID"]
TENANT_ID = os.environ["TENANT_ID"]
# Azure API
AUTHORITY = f"https://login.microsoftonline.com/{TENANT_ID}"
REDIRECT_PATH = os.environ["REDIRECT_PATH"]
AUTHORIZATION_URL = f"{AUTHORITY}/oauth2/v2.0/authorize"
SCOPES = ["openid", "profile", "User.Read"]
# 認証用URL取得
def authorization_request():
oauth = OAuth2Session(CLIENT_ID, redirect_uri=REDIRECT_PATH, scope=SCOPES)
authorization_url, state = oauth.authorization_url(AUTHORIZATION_URL)
return authorization_url, state
#ログインの確認
def check_login():
if "token" not in st.session_state or st.session_state["token"] is None or float(st.session_state["token_expires"]) <= time():
# 認証用リンク表示
authorization_url, st.session_state["authorization_state"] = authorization_request()
st.markdown(f'[Click here to log in]({authorization_url})', unsafe_allow_html=True)
st.stop()
|