Add position filtering options to portfolio management, allowing users to filter by position and team with adjustable thresholds.

app.py

@@ -32,6 +32,9 @@ from global_func.recalc_diversity import recalc_diversity
 from database_queries import *
 from database import *
 
+pos_parse_options = ['Projection', 'Ownership', 'Salary', 'Position', 'Team']
+pos_parse_translate = ['proj_map', 'own_map', 'salary_map', 'pos_map', 'team_map']
+
 showdown_selections = ['Showdown #1', 'Showdown #2', 'Showdown #3', 'Showdown #4', 'Showdown #5', 'Showdown #6', 'Showdown #7', 'Showdown #8', 'Showdown #9', 'Showdown #10', 'Showdown #11', 'Showdown #12', 'Showdown #13', 'Showdown #14', 'Showdown #15']
 dk_db_nfl_showdown_selections = ['DK_NFL_SD_seed_frame_Showdown #1', 'DK_NFL_SD_seed_frame_Showdown #2', 'DK_NFL_SD_seed_frame_Showdown #3', 'DK_NFL_SD_seed_frame_Showdown #4', 'DK_NFL_SD_seed_frame_Showdown #5', 'DK_NFL_SD_seed_frame_Showdown #6',
 'DK_NFL_SD_seed_frame_Showdown #7', 'DK_NFL_SD_seed_frame_Showdown #8', 'DK_NFL_SD_seed_frame_Showdown #9', 'DK_NFL_SD_seed_frame_Showdown #10', 'DK_NFL_SD_seed_frame_Showdown #11', 'DK_NFL_SD_seed_frame_Showdown #12', 'DK_NFL_SD_seed_frame_Showdown #13',
@@ -1745,6 +1748,8 @@ if selected_tab == 'Manage Portfolio':
                     'Finish_percentile': st.session_state['working_frame']['Finish_percentile'].max(),
                     'Diversity': st.session_state['working_frame']['Diversity'].max()
                 }
+            
+            position_choices = [col for col in st.session_state['working_frame'].columns if col not in excluded_cols]
 
         with st.sidebar:
             if 'trimming_dict_maxes' not in st.session_state:
@@ -2125,6 +2130,37 @@ if selected_tab == 'Manage Portfolio':
                         st.session_state['export_base'] = parsed_frame.sort_values(by='median', ascending=False).reset_index(drop=True)
                         st.session_state['export_merge'] = st.session_state['export_base'].copy()
 
+            with st.expander('Position Filtering'):
+                with st.form(key='position_filtering_form'):
+                    position_choice = st.multiselect("Position(s) to filter", options=position_choices, default=[])
+                    position_filter = st.selectbox("Filter on:", options=pos_parse_options)
+                    if position_filter not in ['Position', 'Team']:
+                        position_low_threshold = st.number_input("Low Threshold", value=0.0, min_value=0.0, step=1.0)
+                        position_high_threshold = st.number_input("High Threshold", value=150.0, min_value=0.0, step=1.0)
+                        filter_keys = None
+                    else:
+                        filter_keys_choice = st.selectbox("Filter choice:", options=['Position', 'Team'])
+                        if filter_keys_choice == 'Position':
+                            filter_keys = st.multiselect("Position(s) to keep", options=position_choices, default=[])
+                        else:
+                            filter_keys = st.multiselect("Position(s) to keep", options=st.session_state['map_dict']['team_map'].keys(), default=[])
+                        position_low_threshold = None
+                        position_high_threshold = None
+                    submitted_col, export_col = st.columns(2)
+                    with submitted_col:
+                        reg_submitted = st.form_submit_button("Portfolio")
+                    with export_col:
+                        exp_submitted = st.form_submit_button("Export")
+                    if reg_submitted:
+                        parsed_frame = st.session_state['working_frame'].copy()
+                        parsed_frame = parse_portfolio_on_mapped(parsed_frame, st.session_state['map_dict'], pos_parse_translate[pos_parse_options.index(position_filter)], filter_keys, position_low_threshold, position_high_threshold, position_choice)
+                        st.session_state['working_frame'] = parsed_frame.sort_values(by='median', ascending=False)
+                        st.session_state['export_merge'] = st.session_state['working_frame'].copy()
+                    elif exp_submitted:
+                        parsed_frame = st.session_state['export_base'].copy()
+                        parsed_frame = parse_portfolio_on_mapped(parsed_frame, st.session_state['map_dict'], pos_parse_translate[pos_parse_options.index(position_filter)], filter_keys, position_low_threshold, position_high_threshold, position_choice)
+                        st.session_state['export_base'] = parsed_frame.sort_values(by='median', ascending=False)
+                        st.session_state['export_merge'] = st.session_state['export_base'].copy()
             with st.expander('Trimming Options'):
                 with st.form(key='trim_form'):
                     st.write("Sorting and trimming variables:")