Add support for multiple league data sources in app.py

app.py

@@ -40,17 +40,20 @@ def init_conn():
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
         db = client["League_of_Legends"]
      
-        LOL_Data = st.secrets['LOL_Data']
+        LCK_Data = st.secrets['LOL_Data']
+        LEC_Data = st.secrets['LEC_Data']
+        LCS_Data = st.secrets['LCS_Data']
 
         gc = gspread.service_account_from_dict(credentials)
         gc2 = gspread.service_account_from_dict(credentials2)
 
-        return gc, gc2, db, LOL_Data
+        return gc, gc2, db, LCK_Data, LEC_Data, LCS_Data
     
-gcservice_account, gcservice_account2, db, LOL_Data = init_conn()
+gcservice_account, gcservice_account2, db, LCK_Data, LEC_Data, LCS_Data = init_conn()
 
 percentages_format = {'Exposure': '{:.2%}'}
 freq_format = {'Exposure': '{:.2%}', 'Proj Own': '{:.2%}', 'Edge': '{:.2%}'}
+
 dk_columns = ['CPT', 'TOP', 'JNG', 'MID', 'ADC', 'SUP', 'TEAM', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 fd_columns = ['CPT', 'TOP', 'JNG', 'MID', 'ADC', 'SUP', 'TEAM', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']
 
@@ -74,12 +77,25 @@ def init_DK_seed_frames():
         return DK_seed
 
 @st.cache_data(ttl = 599)
-def init_baselines():
-    try:
-        sh = gcservice_account.open_by_url(LOL_Data)
-    except:
-        sh = gcservice_account2.open_by_url(LOL_Data)
+def init_baselines(league_data):
+
+    if league_data == 'LCK':
+        try:
+            sh = gcservice_account.open_by_url(LCK_Data)
+        except:
+            sh = gcservice_account2.open_by_url(LCK_Data)
+    elif league_data == 'LEC':
+        try:
+            sh = gcservice_account.open_by_url(LEC_Data)
+        except:
+            sh = gcservice_account2.open_by_url(LEC_Data)
+    elif league_data == 'LCS':
+        try:
+            sh = gcservice_account.open_by_url(LCS_Data)
+        except:
+            sh = gcservice_account2.open_by_url(LCS_Data)
     
+
     worksheet = sh.worksheet('ROO')
     load_display = pd.DataFrame(worksheet.get_all_records())
     load_display.replace('', np.nan, inplace=True)
@@ -143,7 +159,7 @@ def sim_contest(Sim_size, seed_frame, maps_dict, sharp_split, Contest_Size):
     return Sim_Winners
 
 DK_seed = init_DK_seed_frames()
-dk_raw = init_baselines()
+dk_raw = init_baselines('LCK')
 
 tab1, tab2 = st.tabs(['Contest Sims', 'Data Export'])
 with tab2:
@@ -154,12 +170,12 @@ with tab2:
               for key in st.session_state.keys():
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
-              dk_raw = init_baselines()
+              dk_raw = init_baselines('LCK')
               
-        slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'))
+        slate_var1 = st.radio("Which data are you loading?", ('LCK', 'LEC', 'LCS'))
         site_var1 = st.radio("What site are you working with?", ('Draftkings'))
         if site_var1 == 'Draftkings':
-            raw_baselines = dk_raw
+            raw_baselines = init_baselines(slate_var1)
             column_names = dk_columns
             
             team_var1 = st.radio("Do you want a frame with specific teams?", ('Full Slate', 'Specific Teams'), key='team_var1')
@@ -208,11 +224,12 @@ with tab1:
               for key in st.session_state.keys():
                   del st.session_state[key]
               DK_seed = init_DK_seed_frames()
-              dk_raw = init_baselines()
-        sim_slate_var1 = st.radio("Which data are you loading?", ('Main Slate', 'Other Main Slate'), key='sim_slate_var1')
+              dk_raw = init_baselines('LCK')
+        sim_slate_var1 = st.radio("Which data are you loading?", ('LCK', 'LEC', 'LCS'), key='sim_slate_var1')
+
         sim_site_var1 = st.radio("What site are you working with?", ('Draftkings'), key='sim_site_var1')
         if sim_site_var1 == 'Draftkings':
-            raw_baselines = dk_raw
+            raw_baselines = init_baselines(sim_slate_var1)
             column_names = dk_columns
             
         contest_var1 = st.selectbox("What contest size are you simulating?", ('Small', 'Medium', 'Large', 'Custom'))