Set to admin only feature

src/login_component.py

@@ -20,6 +20,7 @@ REDIRECT_URI = os.environ.get("REDIRECT_URI")
 SCOPE = os.environ.get("SCOPE")
 
 ENABLE_LOGIN = os.environ.get("ENABLE_LOGIN", False)
+ADMIN_GUID = os.environ.get("ADMIN_GUID")
 
 # Create OAuth2Component instance
 oauth2 = OAuth2Component(CLIENT_ID, CLIENT_SECRET, AUTHORIZE_URL, TOKEN_URL, REFRESH_TOKEN_URL, REVOKE_TOKEN_URL)
@@ -42,6 +43,7 @@ def get_authorization_button():
             yahoo_con = YahooFantasyClient(oauth2, st.session_state.token)
             st.session_state.logged_in_guid = yahoo_con.get_guid_for_logged_in_user()
             st.session_state.yahoo_client = yahoo_con
+            st.session_state.user_admin = st.session_state.logged_in_guid == ADMIN_GUID
             st.rerun()
     else:
         # # If token exists in session state, allow logout
@@ -50,6 +52,7 @@ def get_authorization_button():
             del st.session_state.token
             del st.session_state.logged_in_guid
             del st.session_state.yahoo_client
+            del st.session_state.user_admin
             st.rerun()
         # # If token exists in session state, show the token
         # token = st.session_state["token"]

src/pages/3_Draft_View.py

@@ -96,11 +96,8 @@ def get_page():
     common_page_config()
     st.title(page_title)
 
-    if not is_token_in_session():
-        st.write(
-            "You must authorize the application to access your account in order to use this feature."
-            " Please click Login button above."
-        )
+    if not (is_token_in_session() and st.session_state.get("user_admin")):
+        st.write("Exclusive feature")
 
     else:
         selected_season = st.selectbox("Select Season", list(range(SEASON, 2012, -1)))