Refactor database connection in app.py to remove unused database reference and streamline data retrieval for DraftKings and FanDuel lineups, enhancing code clarity and maintainability.

app.py

@@ -11,11 +11,10 @@ def init_conn():
         uri = st.secrets['mongo_uri']
         client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
         db = client["MLB_Database"]
-        db2 = client["MLB_DFS"]
 
-        return db, db2
+        return db
     
-db, db2 = init_conn()
+db = init_conn()
 
 game_format = {'Win Percentage': '{:.2%}','First Inning Lead Percentage': '{:.2%}', 'Top Score': '{:.2%}',
               'Fifth Inning Lead Percentage': '{:.2%}', '8+ runs': '{:.2%}', 'DK LevX': '{:.2%}', 'FD LevX': '{:.2%}'}
@@ -143,19 +142,19 @@ def init_DK_lineups(type_var, slate_var):
             raw_display[dict_columns] = raw_display[dict_columns].apply(lambda x: x.map(names_dict))
     elif type_var == 'Showdown':
         if slate_var == 'Main':
-            collection = db2['DK_MLB_SD1_seed_frame']
+            collection = db['DK_MLB_SD1_seed_frame']
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
             raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
         elif slate_var == 'Secondary':
-            collection = db2['DK_MLB_SD2_seed_frame']
+            collection = db['DK_MLB_SD2_seed_frame']
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
             raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
         elif slate_var == 'Auxiliary':
-            collection = db2['DK_MLB_SD3_seed_frame']
+            collection = db['DK_MLB_SD3_seed_frame']
             cursor = collection.find().limit(10000)
 
             raw_display = pd.DataFrame(list(cursor))
@@ -214,19 +213,19 @@ def init_FD_lineups(type_var,slate_var):
 
         elif type_var == 'Showdown':
             if slate_var == 'Main':
-                collection = db2['FD_MLB_SD1_seed_frame']
+                collection = db['FD_MLB_SD1_seed_frame']
                 cursor = collection.find().limit(10000)
 
                 raw_display = pd.DataFrame(list(cursor))
                 raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
             elif slate_var == 'Secondary':
-                collection = db2['FD_MLB_SD2_seed_frame']
+                collection = db['FD_MLB_SD2_seed_frame']
                 cursor = collection.find().limit(10000)
 
                 raw_display = pd.DataFrame(list(cursor))
                 raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
             elif slate_var == 'Auxiliary':
-                collection = db2['FD_MLB_SD3_seed_frame']
+                collection = db['FD_MLB_SD3_seed_frame']
                 cursor = collection.find().limit(10000)
             
                 raw_display = pd.DataFrame(list(cursor))