Add contest data download functionality in app.py

- Introduced a new function to download contest data from DraftKings, handling zip files and CSV extraction.
- Updated the user interface to allow selection between uploading a file or downloading contest data directly, enhancing user experience and flexibility.
- Improved error handling for download failures, ensuring users receive appropriate feedback.

app.py

@@ -7,6 +7,9 @@ from collections import Counter
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 from datetime import datetime
+import requests
+import zipfile
+import io
 
 def init_conn():
         
@@ -54,6 +57,38 @@ def grab_contest_player_info(db, sport, type, contest_date, contest_name, contes
     
     return player_info, info_maps
 
+def download_draftkings_contest(contest_id):
+    try:
+        # Construct the URL
+        url = f"https://www.draftkings.com/contest/exportfullstandingscsv/{contest_id}"
+        
+        # Download the file
+        response = requests.get(url)
+        response.raise_for_status()  # Raise an exception for bad status codes
+        
+        # Create a BytesIO object from the content
+        zip_file = io.BytesIO(response.content)
+        
+        # Open the zip file
+        with zipfile.ZipFile(zip_file) as z:
+            # Get the first CSV file in the zip (there should only be one)
+            csv_filename = z.namelist()[0]
+            # Read the CSV file into a pandas DataFrame
+            with z.open(csv_filename) as f:
+                df = pd.read_csv(f)
+        
+        return df
+    
+    except requests.exceptions.RequestException as e:
+        st.error(f"Error downloading contest data: {str(e)}")
+        return None
+    except zipfile.BadZipFile:
+        st.error("The downloaded file is not a valid zip file")
+        return None
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+        return None
+
 db = init_conn()
 
 ## import global functions
@@ -116,19 +151,29 @@ 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]}#/")
+        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]}")
         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')
+                download_option = st.radio("Choose input method:", ["Upload File", "Download from DraftKings"], key='download_option')
+                
+                if download_option == "Upload File":
+                    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')
+                else:
+                    if st.button("Download Contest Data"):
+                        with st.spinner("Downloading contest data..."):
+                            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)
+                            st.session_state['Contest_file'] = download_draftkings_contest(contest_id_map[contest_name_var])
+                            if st.session_state['Contest_file'] is not None:
+                                st.success("Contest data downloaded successfully!")
             else:
                 pass