Wire up login component to save user guid

src/login_component.py

@@ -3,6 +3,8 @@ import streamlit as st
 from streamlit_oauth import OAuth2Component
 import os
 
+from yahoo_client import YahooFantasyClient
+
 # # Load environment variables from .env file
 # from dotenv import load_dotenv
 # load_dotenv()
@@ -37,12 +39,15 @@ def get_authorization_button():
         if result and "token" in result:
             # If authorization successful, save token in session state
             st.session_state.token = result.get("token")
+            yahoo_con = YahooFantasyClient(oauth2, st.session_state.token)
+            st.session_state.logged_in_guid = yahoo_con.get_guid_for_logged_in_user()
             st.rerun()
     else:
         # # If token exists in session state, allow logout
         st.session_state["token"]
         if st.button("Logout"):
             del st.session_state.token
+            del st.session_state.logged_in_guid
             st.rerun()
         # # If token exists in session state, show the token
         # token = st.session_state["token"]

src/yahoo_client.py

@@ -1,9 +1,10 @@
+import asyncio
 import pandas as pd
 from pydantic import BaseModel
 import re
 from typing import List, Mapping, Tuple, Union
 import xml.etree.ElementTree as ET
-from yahoo_oauth import OAuth2
+from streamlit_oauth import OAuth2Component
 
 
 class LeagueSettings(BaseModel):
@@ -50,20 +51,22 @@ class LeagueSettings(BaseModel):
 
 
 class YahooFantasyClient:
-    def get_new_oauth_from_json(self) -> OAuth2:
-        return OAuth2(None, None, from_file=self.client_json_path)
-
-    def __init__(self, client_json_path: str):
-        self.client_json_path = client_json_path
-        self.oauth: OAuth2 = self.get_new_oauth_from_json()
+    def __init__(self, oauth: OAuth2Component, token):
+        self.oauth: OAuth2Component = oauth
+        self.token = token
 
     def authorize_yahoo_from_client_json(self) -> None:
-        if not self.oauth.token_is_valid():
-            self.oauth.refresh_access_token()
+        self.token = self.oauth.refresh_token(self.token)
 
     def yahoo_request_to_xml(self, url: str) -> Tuple[ET.Element, str]:
         self.authorize_yahoo_from_client_json()
-        r = self.oauth.session.get(url)
+        client = self.oauth.client.get_httpx_client()
+        r = asyncio.run(
+            client.get(
+                url,
+                headers={"Authorization": f"Bearer {self.token.get('access_token')}"},
+            )
+        )
 
         xmlstring = r.text
         xmlstring = re.sub(' xmlns="[^"]+"', "", xmlstring, count=1)
@@ -78,6 +81,12 @@ class YahooFantasyClient:
         )
         return league_keys
 
+    def get_guid_for_logged_in_user(self) -> str:
+        url = "https://fantasysports.yahooapis.com/fantasy/v2/users;use_login=1/games;game_keys=nfl/teams"
+        root, _ = self.yahoo_request_to_xml(url)
+        user_guid = root.findtext("./users/user/guid")
+        return user_guid
+
     def parse_matchup(self, matchup: ET.Element, match_index: int) -> List[Mapping[str, Union[str, float]]]:
         matchup_info = Matchup.from_xml(matchup, match_index)
         return matchup_info.to_list_team_dict()