/streamlit-oauth

Simple OAuth Component for Streamlit App

Primary LanguagePythonMIT LicenseMIT

🔐 Streamlit OAuth: Enhance Your Data Apps with Seamless Integration

Effortlessly integrate OAuth2 authorization into your Streamlit apps using the streamlined httpx_oauth wrapper

Why Streamlit OAuth?

In today's digital landscape, authorization is crucial for accessing and utilizing data from various applications. With Streamlit OAuth, you can easily incorporate OAuth2 authorization into your Streamlit data apps, allowing you to securely and efficiently fetch data from other applications. This powerful integration unlocks the full potential of your Streamlit data-driven applications, enabling you to create more dynamic and interactive user experiences.

Installation

pip install streamlit-oauth

Getting started

To use Streamlit OAuth, you need to create an OAuth2 component with your authentication details (set your callback url to https://<YOUR ADDRESS>/component/streamlit_oauth.authorize_button):

import streamlit as st
from streamlit_oauth import OAuth2Component
import os

# Load environment variables from .env file
from dotenv import load_dotenv
load_dotenv()

# Set environment variables
AUTHORIZE_URL = os.environ.get('AUTHORIZE_URL')
TOKEN_URL = os.environ.get('TOKEN_URL')
REFRESH_TOKEN_URL = os.environ.get('REFRESH_TOKEN_URL')
REVOKE_TOKEN_URL = os.environ.get('REVOKE_TOKEN_URL')
CLIENT_ID = os.environ.get('CLIENT_ID')
CLIENT_SECRET = os.environ.get('CLIENT_SECRET')
REDIRECT_URI = os.environ.get('REDIRECT_URI')
SCOPE = os.environ.get('SCOPE')

# Create OAuth2Component instance
oauth2 = OAuth2Component(CLIENT_ID, CLIENT_SECRET, AUTHORIZE_URL, TOKEN_URL, REFRESH_TOKEN_URL, REVOKE_TOKEN_URL)

# Check if token exists in session state
if 'token' not in st.session_state:
    # If not, show authorize button
    result = oauth2.authorize_button("Authorize", REDIRECT_URI, SCOPE)
    if result and 'token' in result:
        # If authorization successful, save token in session state
        st.session_state.token = result.get('token')
        st.rerun()
else:
    # If token exists in session state, show the token
    token = st.session_state['token']
    st.json(token)
    if st.button("Refresh Token"):
        # If refresh token button is clicked, refresh the token
        token = oauth2.refresh_token(token)
        st.session_state.token = token
        st.rerun()

more examples can be found in the examples

Parameters:

  • client_id: The OAuth2 client id provided by the authorization server.
  • client_secret: The OAuth2 client secret provided by the authorization server.
  • authroize_endpoint: The authorization endpoint URL of the OAuth2 server. (Deprecated for typo.)
  • authorize_endpoint: The authorization endpoint URL of the OAuth2 server.
  • token_endpoint: The token endpoint URL of the OAuth2 server.
  • refresh_token_endpoint: The refresh token endpoint URL of the OAuth2 server.
  • revoke_token_endpoint: The revoke token endpoint URL of the OAuth2 server.
  • client: The httpx_oauth client to be used for the requests, default is None, if specified other arguments will be ignored.

authorize_button(self, name, redirect_uri, scope, height=800, width=600, key=None, extra_params={}, pkce=None, use_container_width=False, icon=None)

Generates an HTML button that initiates the OAuth2 authorization code grant flow. The button opens a popup window that prompts the user to authorize the application.

Parameters:

  • name: The name to be displayed on the button.
  • redirect_uri: The URL where the authorization server will redirect the user after the authorization process is completed.
  • scope: The OAuth2 scopes required by the application.
  • height: The height of the popup window.
  • width: The width of the popup window.
  • key: The unique key of the button component.
  • extra_params: A dictionary containing extra parameters to be sent to the authorization server.
  • pkce: accept value S256 indicating whether to use PKCE (Proof Key for Code Exchange) for the authorization code grant flow.
  • use_container_width: If True, set the button width to the container width.
  • icon: The icon to be displayed on the button.
  • auto_click: If True, the button will be clicked automatically.

Returns:

  • A dictionary containing the authorization code or access token object, depending on the flow.

refresh_token(self, token, force=False)

Refreshes the access token using the refresh token. If the token is not expired, the function returns the same token.

Parameters:

  • token: The access token object to be refreshed.
  • force: A boolean value that forces the refresh token to be used, even if the access token is not expired.

Returns:

  • A dictionary containing the new access token or the same access token if it has not expired.

revoke_token(self, token, token_type_hint='access_token')

Parameters:

  • token: The access token object to be revoked.
  • token_type_hint: A hint about the type of the token submitted for revocation.

Returns:

  • A boolean value indicating whether the token was revoked successfully.