Add login with streamlit_oauth (#1)

* Add fork of yahoo_fantasy_api to reqs

* Add secrets.toml to gitignore

* Add login using streamlit_oauth

.gitignore

@@ -158,3 +158,6 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+# Streamlit
+secrets.toml

dev-requirements.txt

@@ -6,6 +6,8 @@
 #
 altair==5.0.1
     # via streamlit
+anyio==3.7.1
+    # via httpcore
 attrs==23.1.0
     # via
     #   jsonschema
@@ -17,7 +19,10 @@ blinker==1.6.2
 cachetools==5.3.1
     # via streamlit
 certifi==2023.7.22
-    # via requests
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
 charset-normalizer==3.2.0
     # via requests
 click==8.1.6
@@ -28,8 +33,19 @@ gitdb==4.0.10
     # via gitpython
 gitpython==3.1.32
     # via streamlit
+h11==0.14.0
+    # via httpcore
+httpcore==0.17.3
+    # via httpx
+httpx==0.24.1
+    # via httpx-oauth
+httpx-oauth==0.13.0
+    # via streamlit-oauth
 idna==3.4
-    # via requests
+    # via
+    #   anyio
+    #   httpx
+    #   requests
 importlib-metadata==6.8.0
     # via streamlit
 iniconfig==2.0.0
@@ -121,7 +137,16 @@ six==1.16.0
     # via python-dateutil
 smmap==5.0.0
     # via gitdb
+sniffio==1.3.0
+    # via
+    #   anyio
+    #   httpcore
+    #   httpx
 streamlit==1.25.0
+    # via
+    #   YFDashboard (pyproject.toml)
+    #   streamlit-oauth
+streamlit-oauth==0.0.1
     # via YFDashboard (pyproject.toml)
 tenacity==8.2.3
     # via streamlit

pyproject.toml

@@ -9,6 +9,7 @@ dependencies = [
     "streamlit",
     "numpy",
     "pandas",
+    "streamlit-oauth",
     ]
 
 [project.optional-dependencies]
@@ -30,3 +31,9 @@ src = ["src"]
 
 [tool.mypy]
 python_version = "3.11"
+
+[[tool.mypy.overrides]]
+module = [
+    'streamlit_oauth'
+]
+ignore_missing_imports = true

requirements.txt

@@ -6,6 +6,8 @@
 #
 altair==5.0.1
     # via streamlit
+anyio==3.7.1
+    # via httpcore
 attrs==23.1.0
     # via
     #   jsonschema
@@ -15,7 +17,10 @@ blinker==1.6.2
 cachetools==5.3.1
     # via streamlit
 certifi==2023.7.22
-    # via requests
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
 charset-normalizer==3.2.0
     # via requests
 click==8.1.6
@@ -24,8 +29,19 @@ gitdb==4.0.10
     # via gitpython
 gitpython==3.1.32
     # via streamlit
+h11==0.14.0
+    # via httpcore
+httpcore==0.17.3
+    # via httpx
+httpx==0.24.1
+    # via httpx-oauth
+httpx-oauth==0.13.0
+    # via streamlit-oauth
 idna==3.4
-    # via requests
+    # via
+    #   anyio
+    #   httpx
+    #   requests
 importlib-metadata==6.8.0
     # via streamlit
 jinja2==3.1.2
@@ -93,7 +109,16 @@ six==1.16.0
     # via python-dateutil
 smmap==5.0.0
     # via gitdb
+sniffio==1.3.0
+    # via
+    #   anyio
+    #   httpcore
+    #   httpx
 streamlit==1.25.0
+    # via
+    #   YFDashboard (pyproject.toml)
+    #   streamlit-oauth
+streamlit-oauth==0.0.1
     # via YFDashboard (pyproject.toml)
 tenacity==8.2.3
     # via streamlit

src/Home.py

@@ -1,12 +1,13 @@
 import streamlit as st
 
 from config import DEFAULT_ICON, LEAGUE_NAME
+from login_component import get_authorization_button
 
 
 def get_app():
     keeper_title = f"{LEAGUE_NAME}"
     st.set_page_config(page_title=keeper_title, page_icon=DEFAULT_ICON)
-
+    get_authorization_button()
     st.markdown(
         f"""
     Welcome {LEAGUE_NAME}!

src/login_component.py

@@ -0,0 +1,45 @@
+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)
+
+
+def get_authorization_button():
+    # Check if token exists in session state
+    if "token" not in st.session_state:
+        # If not, show authorize button
+        result = oauth2.authorize_button("Login", 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.experimental_rerun()
+    else:
+        # # If token exists in session state, allow logout
+        st.session_state["token"]
+        if st.button("Logout"):
+            del st.session_state.token
+            st.experimental_rerun()
+        # # If token exists in session state, show the token
+        # token = st.session_state["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.experimental_rerun()

src/pages/1_Keepers.py

@@ -3,6 +3,7 @@ import pandas as pd
 import streamlit as st
 
 from config import DEFAULT_ICON, LEAGUE_NAME, LEAGUE_NUMBER_TEAMS
+from login_component import get_authorization_button
 from streamlit_filter import filter_dataframe
 
 
@@ -135,6 +136,7 @@ def filtered_keeper_dataframe(data: pd.DataFrame, teams_list: list[str]):
 def get_keeper_app():
     keeper_title = f"{LEAGUE_NAME} Keeper Options"
     st.set_page_config(page_title=keeper_title, page_icon=DEFAULT_ICON, layout="wide")
+    get_authorization_button()
     st.title(keeper_title)
     data, teams_list = load_data()
 

src/pages/2_Rules.py

@@ -1,10 +1,12 @@
 import streamlit as st
 
 from config import DEFAULT_ICON, LEAGUE_NAME
+from login_component import get_authorization_button
 
 
 page_title = f"{LEAGUE_NAME} Keeper Rules"
 st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON)
+get_authorization_button()
 st.title(page_title)
 st.markdown(
     """