Add login connected to gsheets

dev-requirements.txt

@@ -30,8 +30,8 @@ babel==2.13.1
     # via jupyterlab-server
 beautifulsoup4==4.12.2
     # via
-    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   nbconvert
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
 black==23.11.0
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
 bleach==6.1.0
@@ -39,7 +39,9 @@ bleach==6.1.0
 blinker==1.7.0
     # via streamlit
 cachetools==5.3.2
-    # via streamlit
+    # via
+    #   google-auth
+    #   streamlit
 certifi==2023.11.17
     # via
     #   httpcore
@@ -60,11 +62,15 @@ comm==0.2.0
 debugpy==1.8.0
     # via ipykernel
 decorator==5.1.1
-    # via ipython
+    # via
+    #   gspread-pandas
+    #   ipython
 defusedxml==0.7.1
     # via nbconvert
 duckdb==0.9.2
-    # via nfl-playoff-challenge-steamlit (pyproject.toml)
+    # via
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
 executing==2.0.1
     # via stack-data
 fastjsonschema==2.19.0
@@ -75,6 +81,27 @@ gitdb==4.0.11
     # via gitpython
 gitpython==3.1.40
     # via streamlit
+google-auth==2.26.1
+    # via
+    #   google-auth-oauthlib
+    #   gspread
+    #   gspread-pandas
+google-auth-oauthlib==1.2.0
+    # via
+    #   gspread
+    #   gspread-pandas
+gspread==5.12.4
+    # via
+    #   gspread-dataframe
+    #   gspread-formatting
+    #   gspread-pandas
+    #   st-gsheets-connection
+gspread-dataframe==3.3.1
+    # via st-gsheets-connection
+gspread-formatting==1.1.2
+    # via st-gsheets-connection
+gspread-pandas==3.2.3
+    # via st-gsheets-connection
 h11==0.14.0
     # via httpcore
 html5lib==1.1
@@ -222,13 +249,15 @@ notebook-shim==0.2.3
     #   notebook
 numpy==1.26.2
     # via
-    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   altair
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   pandas
     #   pandas-stubs
     #   pyarrow
     #   pydeck
     #   streamlit
+oauthlib==3.2.2
+    # via requests-oauthlib
 overrides==7.4.0
     # via jupyter-server
 packaging==23.2
@@ -244,10 +273,13 @@ packaging==23.2
     #   qtconsole
     #   qtpy
     #   streamlit
-pandas==2.1.3
+pandas==1.5.3
     # via
-    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   altair
+    #   gspread-dataframe
+    #   gspread-pandas
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
     #   streamlit
 pandas-stubs==2.1.1.230928
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
@@ -285,6 +317,12 @@ pure-eval==0.2.2
     # via stack-data
 pyarrow==14.0.1
     # via streamlit
+pyasn1==0.5.1
+    # via
+    #   pyasn1-modules
+    #   rsa
+pyasn1-modules==0.3.0
+    # via google-auth
 pycparser==2.21
     # via cffi
 pydantic==2.5.2
@@ -335,7 +373,10 @@ referencing==0.31.1
 requests==2.31.0
     # via
     #   jupyterlab-server
+    #   requests-oauthlib
     #   streamlit
+requests-oauthlib==1.3.1
+    # via google-auth-oauthlib
 rfc3339-validator==0.1.4
     # via
     #   jsonschema
@@ -350,6 +391,8 @@ rpds-py==0.13.2
     # via
     #   jsonschema
     #   referencing
+rsa==4.9
+    # via google-auth
 ruff==0.1.6
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
 send2trash==1.8.2
@@ -358,6 +401,7 @@ six==1.16.0
     # via
     #   asttokens
     #   bleach
+    #   gspread-dataframe
     #   html5lib
     #   python-dateutil
     #   rfc3339-validator
@@ -370,11 +414,18 @@ sniffio==1.3.0
     #   httpx
 soupsieve==2.5
     # via beautifulsoup4
+sql-metadata==2.10.0
+    # via st-gsheets-connection
+sqlparse==0.4.4
+    # via sql-metadata
+st-gsheets-connection==0.0.3
+    # via nfl-playoff-challenge-steamlit (pyproject.toml)
 stack-data==0.6.3
     # via ipython
 streamlit==1.29.0
     # via
     #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
     #   streamlit-oauth
 streamlit-oauth==0.1.5
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
@@ -432,8 +483,6 @@ typing-extensions==4.8.0
     #   pydantic
     #   pydantic-core
     #   streamlit
-tzdata==2023.3
-    # via pandas
 tzlocal==5.2
     # via streamlit
 uri-template==1.3.0
@@ -443,7 +492,9 @@ urllib3==2.1.0
     #   requests
     #   types-requests
 validators==0.22.0
-    # via streamlit
+    # via
+    #   st-gsheets-connection
+    #   streamlit
 watchdog==3.0.0
     # via streamlit
 wcwidth==0.2.12

pyproject.toml

@@ -15,6 +15,7 @@ dependencies = [
     "pandas",
     "pydantic",
     "streamlit-oauth",
+    "st-gsheets-connection",
     ]
 
 [project.optional-dependencies]

requirements.txt

@@ -19,7 +19,9 @@ beautifulsoup4==4.12.2
 blinker==1.7.0
     # via streamlit
 cachetools==5.3.2
-    # via streamlit
+    # via
+    #   google-auth
+    #   streamlit
 certifi==2023.11.17
     # via
     #   httpcore
@@ -29,12 +31,37 @@ charset-normalizer==3.3.2
     # via requests
 click==8.1.7
     # via streamlit
+decorator==5.1.1
+    # via gspread-pandas
 duckdb==0.9.2
-    # via nfl-playoff-challenge-steamlit (pyproject.toml)
+    # via
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
 gitdb==4.0.11
     # via gitpython
 gitpython==3.1.40
     # via streamlit
+google-auth==2.26.1
+    # via
+    #   google-auth-oauthlib
+    #   gspread
+    #   gspread-pandas
+google-auth-oauthlib==1.2.0
+    # via
+    #   gspread
+    #   gspread-pandas
+gspread==5.12.4
+    # via
+    #   gspread-dataframe
+    #   gspread-formatting
+    #   gspread-pandas
+    #   st-gsheets-connection
+gspread-dataframe==3.3.1
+    # via st-gsheets-connection
+gspread-formatting==1.1.2
+    # via st-gsheets-connection
+gspread-pandas==3.2.3
+    # via st-gsheets-connection
 h11==0.14.0
     # via httpcore
 html5lib==1.1
@@ -70,20 +97,25 @@ mdurl==0.1.2
     # via markdown-it-py
 numpy==1.26.2
     # via
-    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   altair
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   pandas
     #   pyarrow
     #   pydeck
     #   streamlit
+oauthlib==3.2.2
+    # via requests-oauthlib
 packaging==23.2
     # via
     #   altair
     #   streamlit
-pandas==2.1.3
+pandas==1.5.3
     # via
-    #   nfl-playoff-challenge-steamlit (pyproject.toml)
     #   altair
+    #   gspread-dataframe
+    #   gspread-pandas
+    #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
     #   streamlit
 pillow==10.1.0
     # via streamlit
@@ -91,6 +123,12 @@ protobuf==4.25.1
     # via streamlit
 pyarrow==14.0.1
     # via streamlit
+pyasn1==0.5.1
+    # via
+    #   pyasn1-modules
+    #   rsa
+pyasn1-modules==0.3.0
+    # via google-auth
 pydantic==2.5.2
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
 pydantic-core==2.14.5
@@ -112,15 +150,22 @@ referencing==0.31.1
     #   jsonschema
     #   jsonschema-specifications
 requests==2.31.0
-    # via streamlit
+    # via
+    #   requests-oauthlib
+    #   streamlit
+requests-oauthlib==1.3.1
+    # via google-auth-oauthlib
 rich==13.7.0
     # via streamlit
 rpds-py==0.13.2
     # via
     #   jsonschema
     #   referencing
+rsa==4.9
+    # via google-auth
 six==1.16.0
     # via
+    #   gspread-dataframe
     #   html5lib
     #   python-dateutil
 smmap==5.0.1
@@ -132,9 +177,16 @@ sniffio==1.3.0
     #   httpx
 soupsieve==2.5
     # via beautifulsoup4
+sql-metadata==2.10.0
+    # via st-gsheets-connection
+sqlparse==0.4.4
+    # via sql-metadata
+st-gsheets-connection==0.0.3
+    # via nfl-playoff-challenge-steamlit (pyproject.toml)
 streamlit==1.29.0
     # via
     #   nfl-playoff-challenge-steamlit (pyproject.toml)
+    #   st-gsheets-connection
     #   streamlit-oauth
 streamlit-oauth==0.1.5
     # via nfl-playoff-challenge-steamlit (pyproject.toml)
@@ -151,14 +203,14 @@ typing-extensions==4.8.0
     #   pydantic
     #   pydantic-core
     #   streamlit
-tzdata==2023.3
-    # via pandas
 tzlocal==5.2
     # via streamlit
 urllib3==2.1.0
     # via requests
 validators==0.22.0
-    # via streamlit
+    # via
+    #   st-gsheets-connection
+    #   streamlit
 watchdog==3.0.0
     # via streamlit
 webencodings==0.5.1

src/login.py

@@ -0,0 +1,68 @@
+import hmac
+import streamlit as st
+from streamlit_gsheets import GSheetsConnection
+
+
+conn = st.connection("gsheets", type=GSheetsConnection)
+
+
+def get_user_id_from_email_if_exists(email: str) -> int:
+    # returns -1 if id not found
+    df_email = conn.read(
+    worksheet="users",
+    usecols=[0, 1],)
+    df_email = df_email[df_email.email.notna()]
+    for row in df_email.itertuples():
+        if row.email.lower() == email.lower():
+            return row.id
+    return -1
+
+
+def get_password_from_user_id(user_id: int) -> str | None:
+    try:
+        df_pass = conn.read(
+        worksheet=f"user-{user_id}",
+        usecols=[0],)
+        if df_pass:
+            return df_pass.password[0]
+    except:
+        return None
+
+
+def check_password():
+    """Returns `True` if the user had a correct password."""
+
+    def login_form():
+        """Form with widgets to collect user information"""
+        with st.form("Credentials"):
+            st.text_input("Email", key="email")
+            st.text_input("Password", type="password", key="password")
+            st.form_submit_button("Log in", on_click=password_entered)
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        # check if email exists
+        if user_id:=get_user_id_from_email_if_exists(st.session_state["email"]):
+            if password:=get_password_from_user_id(user_id):
+                st.write(password)
+                if hmac.compare_digest(
+                    st.session_state["password"],
+                    password,
+                ):
+                    st.session_state["password_correct"] = True
+                    del st.session_state["password"]  # Don't store the email or password.
+                    del st.session_state["email"]
+                    del password
+                    return
+        st.session_state["password_correct"] = False
+
+
+    # Return True if the email + password is validated.
+    if st.session_state.get("password_correct", False):
+        return True
+
+    # Show inputs for email + password.
+    login_form()
+    if "password_correct" in st.session_state:
+        st.error("😕 User not known or password incorrect")
+    return False

src/pages/10_Set_Your_Lineup.py

@@ -5,6 +5,7 @@ from config import DEFAULT_ICON
 from shared_page import common_page_config
 
 from queries.nflverse.github_data import get_weekly_rosters
+from login import check_password
 
 
 @dataclass
@@ -75,6 +76,8 @@ def get_page():
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
     common_page_config()
     st.title(page_title)
+    if not check_password:
+        st.write("Sorry, you must be logged in first to play")
 
     qb_options, wr_options, rb_options, te_options, k_options, df_rosters = load_options()