Implement download option for contest data in app.py

- Added functionality to download contest data directly from DraftKings using user browser cookies, enhancing user experience by simplifying data retrieval.
- Introduced a manual upload option for contest files, allowing users to upload CSV or Excel files if needed.
- Updated the user interface to include separate columns for download and upload options, improving layout and accessibility.

app.py

@@ -7,6 +7,10 @@ from collections import Counter
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 from datetime import datetime
+import requests
+from io import BytesIO
+import zipfile
+import browser_cookie3
 
 def init_conn():
         
@@ -116,19 +120,62 @@ with tab1:
             else:
                 pass
     with col2:
-        st.info(f"If you are manually loading and do not have the results CSV for the contest you selected, you can find it here: https://www.draftkings.com/contest/gamecenter/{contest_id_map[contest_name_var]}#/, or you can initiate a download with this link: https://www.draftkings.com/contest/exportfullstandingscsv/{contest_id_map[contest_name_var]}")
+        st.info(f"If you are manually loading and do not have the results CSV for the contest you selected, you can find it here: https://www.draftkings.com/contest/gamecenter/{contest_id_map[contest_name_var]}#/")
+        
         if parse_type == 'Manual':
             if 'Contest_file_helper' in st.session_state:
                 del st.session_state['Contest_file_helper']
             if 'Contest_file' in st.session_state:
                 del st.session_state['Contest_file']
             if 'Contest_file' not in st.session_state:
-                st.session_state['Contest_upload'] = st.file_uploader("Upload Contest File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
-                st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_select, type_var, date_select, contest_name_var, contest_id_map)
-                try:
-                    st.session_state['Contest_file'] = pd.read_csv(st.session_state['Contest_upload'])
-                except:
-                    st.warning('Please upload a Contest CSV')
+                # Add download option
+                download_col, upload_col = st.columns(2)
+                with download_col:
+                    st.markdown("### Download Option")
+                    if st.button("Download from DraftKings"):
+                        try:
+                            # Get the download URL
+                            download_url = f"https://www.draftkings.com/contest/exportfullstandingscsv/{contest_id_map[contest_name_var]}"
+                            
+                            # Get cookies from the user's browser
+                            cookies = browser_cookie3.load(domain_name='.draftkings.com')
+                            cookie_dict = {cookie.name: cookie.value for cookie in cookies}
+                            
+                            # Attempt to download using requests with browser cookies
+                            headers = {
+                                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+                            }
+                            response = requests.get(download_url, cookies=cookie_dict, headers=headers)
+                            
+                            if response.status_code == 200:
+                                # Check if it's a zip file
+                                if 'application/zip' in response.headers.get('content-type', ''):
+                                    # Extract CSV from zip
+                                    with zipfile.ZipFile(BytesIO(response.content)) as zip_ref:
+                                        # Get the first CSV file in the zip
+                                        csv_filename = [f for f in zip_ref.namelist() if f.endswith('.csv')][0]
+                                        with zip_ref.open(csv_filename) as csv_file:
+                                            st.session_state['Contest_file'] = pd.read_csv(csv_file)
+                                    st.success("Successfully downloaded and loaded contest data!")
+                                else:
+                                    st.error("Downloaded file is not in the expected format. Please use manual upload instead.")
+                            else:
+                                st.error(f"Failed to download (Status code: {response.status_code}). You may need to log into DraftKings in your browser first.")
+                                st.info("Please use the manual upload option instead.")
+                        except Exception as e:
+                            st.error(f"Error during download: {str(e)}")
+                            st.info("Please use the manual upload option instead.")
+                
+                with upload_col:
+                    st.markdown("### Manual Upload Option")
+                    st.session_state['Contest_upload'] = st.file_uploader("Upload Contest File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
+                    st.session_state['player_info'], st.session_state['info_maps'] = grab_contest_player_info(db, sport_select, type_var, date_select, contest_name_var, contest_id_map)
+                    if st.session_state['Contest_upload'] is not None:
+                        try:
+                            st.session_state['Contest_file'] = pd.read_csv(st.session_state['Contest_upload'])
+                            st.success("Successfully loaded contest data!")
+                        except:
+                            st.warning('Please upload a valid Contest CSV')
             else:
                 pass
 

requirements.txt

@@ -6,4 +6,5 @@ rapidfuzz
 pulp
 docker
 scipy
-pymongo
+pymongo
+browser-cookie3