Refactor query construction in DK seed frame function to streamline player and team locking/removal logic. Consolidated query conditions and improved execution flow for better performance and readability.

src/streamlit_app.py

@@ -70,45 +70,47 @@ def init_DK_seed_frames(sharp_split, player_lock_var1, team_lock_var1, player_re
     names_dict = dict(zip(raw_data['key'], raw_data['value']))
     
     collection = db["DK_NFL_seed_frame"] 
+    
+    # Build the query
+    query = {}
     if player_lock_var1 != [] or team_lock_var1 != []:
         # Build query to check if locked players exist in any position column
-        player_conditions = []
+        query_conditions = []
+        
         if player_lock_var1:
             position_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST']
+            player_conditions = []
             for column in position_columns:
                 player_conditions.append({column: {'$in': player_lock_var1}})
-        
-        query_conditions = []
-        if player_conditions:
             query_conditions.append({'$or': player_conditions})
+        
         if team_lock_var1:
             query_conditions.append({'Team': {'$in': team_lock_var1}})
         
         if len(query_conditions) == 1:
-            collection = collection.find(query_conditions[0])
+            query = query_conditions[0]
         else:
-            collection = collection.find({'$and': query_conditions})
+            query = {'$and': query_conditions}
+            
     elif player_remove_var1 != [] or team_remove_var1 != []:
         # Build query to exclude lineups containing removed players
-        exclusion_conditions = []
+        query_conditions = []
+        
         if player_remove_var1:
             position_columns = ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST']
             for column in position_columns:
-                exclusion_conditions.append({column: {'$nin': player_remove_var1}})
+                query_conditions.append({column: {'$nin': player_remove_var1}})
         
-        query_conditions = []
-        if exclusion_conditions:
-            query_conditions.extend(exclusion_conditions)
         if team_remove_var1:
             query_conditions.append({'Team': {'$nin': team_remove_var1}})
         
         if len(query_conditions) == 1:
-            collection = collection.find(query_conditions[0])
+            query = query_conditions[0]
         else:
-            collection = collection.find({'$and': query_conditions})
-    else:
-        collection = collection.find()
-    cursor = collection.find().limit(sharp_split)
+            query = {'$and': query_conditions}
+    
+    # Execute the query once
+    cursor = collection.find(query).limit(sharp_split)
 
     raw_display = pd.DataFrame(list(cursor))
     raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]