File size: 3,718 Bytes
92b3037 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
import streamlit as st
import asyncio
from httpx_oauth.clients.google import GoogleOAuth2
__version__ = "0.1"
async def write_authorization_url(client, redirect_uri):
authorization_url = await client.get_authorization_url(
redirect_uri,
scope=["profile", "email"],
extras_params={"access_type": "offline"},
)
return authorization_url
async def write_access_token(client, redirect_uri, code):
token = await client.get_access_token(code, redirect_uri)
return token
async def get_user_info(client, token):
user_id, user_email = await client.get_id_email(token)
return user_id, user_email
async def revoke_token(client, token):
return await client.revoke_token(token)
def login_button(authorization_url, app_name, app_desc):
st.markdown('''<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/css/bootstrap.min.css" rel="stylesheet"
integrity="sha384-EVSTQN3/azprG1Anm3QDgpJLIm9Nao0Yz1ztcQTwFspd3yD65VohhpuuCOmLASjC" crossorigin="anonymous">''',
unsafe_allow_html=True)
container = f'''
<div class="container-fluid border py-4 px-4 border-primary">
<h5><strong>{app_name}</strong></h5>
<p>{app_desc}</p>
<a target="_self" href="{authorization_url}">
<img class="img-fluid" src="https://i.imgur.com/YTxsnUl.png" alt="streamlit">
</a>
</div>
'''
st.markdown(container, unsafe_allow_html=True)
def logout_button(button_text):
if st.button(button_text):
asyncio.run(
revoke_token(
client=st.session_state.client,
token=st.session_state.token["access_token"],
)
)
st.session_state.user_email = None
st.session_state.user_id = None
st.session_state.token = None
st.experimental_rerun()
def login(
client_id,
client_secret,
redirect_uri,
app_name="Continue with Google",
app_desc="",
logout_button_text="Logout",
):
st.session_state.client = GoogleOAuth2(client_id, client_secret)
authorization_url = asyncio.run(
write_authorization_url(
client=st.session_state.client, redirect_uri=redirect_uri
)
)
app_desc
if "token" not in st.session_state:
st.session_state.token = None
if st.session_state.token is None:
try:
code = st.experimental_get_query_params()["code"]
except:
login_button(authorization_url, app_name, app_desc)
else:
# Verify token is correct:
try:
token = asyncio.run(
write_access_token(
client=st.session_state.client,
redirect_uri=redirect_uri,
code=code,
)
)
except:
login_button(authorization_url, app_name, app_desc)
else:
# Check if token has expired:
if token.is_expired():
login_button(authorization_url, app_name, app_desc)
else:
st.session_state.token = token
st.session_state.user_id, st.session_state.user_email = asyncio.run(
get_user_info(
client=st.session_state.client, token=token["access_token"]
)
)
logout_button(button_text=logout_button_text)
return (st.session_state.user_id, st.session_state.user_email)
else:
logout_button(button_text=logout_button_text)
return (st.session_state.user_id, st.session_state.user_email)
|