coollsd commited on
Commit
6b98612
·
verified ·
1 Parent(s): b1434c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -974,7 +974,7 @@ async def handle_upload(file: UploadFile = File(...)):
974
 
975
  cookies = await get_cookies()
976
  if 'csrftoken' not in cookies or 'sessionid' not in cookies:
977
- return JSONResponse(content={"error": "Failed"}, status_code=500)
978
 
979
  original_extension = os.path.splitext(file.filename)[1][1:]
980
  supported_types = ['mp4', 'png', 'jpg', 'jpeg', 'gif', 'mp3', 'pdf', 'txt']
@@ -988,12 +988,12 @@ async def handle_upload(file: UploadFile = File(...)):
988
 
989
  upload_result = await initiate_upload(cookies, temp_filename, content_type)
990
  if not upload_result or 'upload_url' not in upload_result:
991
- return JSONResponse(content={"error": "Failed to upload"}, status_code=500)
992
 
993
  file_content = await file.read()
994
  upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
995
  if not upload_success:
996
- return JSONResponse(content={"error": "FAILED GOD MAN AFTER alot of attempts"}, status_code=500)
997
 
998
  original_url = upload_result['serving_url']
999
  mirrored_url = f"/rbxg/{original_url.split('/pbxt/')[1]}"
@@ -1114,7 +1114,7 @@ async def upload_file(upload_url: str, file_content: bytes, content_type: str) -
1114
  return False
1115
 
1116
  async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
1117
- while True:
1118
  try:
1119
  success = await upload_file(upload_url, file_content, content_type)
1120
  if success:
 
974
 
975
  cookies = await get_cookies()
976
  if 'csrftoken' not in cookies or 'sessionid' not in cookies:
977
+ return JSONResponse(content={"error": "Failed to get cookies"}, status_code=500)
978
 
979
  original_extension = os.path.splitext(file.filename)[1][1:]
980
  supported_types = ['mp4', 'png', 'jpg', 'jpeg', 'gif', 'mp3', 'pdf', 'txt']
 
988
 
989
  upload_result = await initiate_upload(cookies, temp_filename, content_type)
990
  if not upload_result or 'upload_url' not in upload_result:
991
+ return JSONResponse(content={"error": "Failed to initiate upload"}, status_code=500)
992
 
993
  file_content = await file.read()
994
  upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
995
  if not upload_success:
996
+ return JSONResponse(content={"error": "Failed to upload after multiple attempts"}, status_code=500)
997
 
998
  original_url = upload_result['serving_url']
999
  mirrored_url = f"/rbxg/{original_url.split('/pbxt/')[1]}"
 
1114
  return False
1115
 
1116
  async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
1117
+ for _ in range(max_retries):
1118
  try:
1119
  success = await upload_file(upload_url, file_content, content_type)
1120
  if success: