paola1 commited on
Commit
5832dd0
·
verified ·
1 Parent(s): 410351d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,7 +16,7 @@ async def upload_file(file: UploadFile = File(...)):
16
  'file_name': file.filename,
17
  'file_data': await file.read(),
18
  'mime_type': file.content_type,
19
- 'upload_time': datetime.datetime.utcnow()
20
  }
21
  return {"upload_id": upload_id, "message": "File uploaded successfully"}
22
 
@@ -34,14 +34,14 @@ async def upload_custom(file: UploadFile = File(...), download_link: str = Form(
34
  'file_name': file.filename,
35
  'file_data': await file.read(),
36
  'mime_type': file.content_type,
37
- 'upload_time': datetime.datetime.utcnow()
38
  }
39
  return {"download_link": download_link, "message": "File uploaded successfully"}
40
 
41
  @app.get("/download/{upload_id}")
42
  async def download_file(upload_id: str):
43
  # Remove expired files
44
- expire_time = datetime.datetime.utcnow() - timedelta(minutes=5)
45
  keys_to_remove = [key for key, value in data_storage.items() if value['upload_time'] < expire_time]
46
  for key in keys_to_remove:
47
  del data_storage[key]
 
16
  'file_name': file.filename,
17
  'file_data': await file.read(),
18
  'mime_type': file.content_type,
19
+ 'upload_time': datetime.utcnow()
20
  }
21
  return {"upload_id": upload_id, "message": "File uploaded successfully"}
22
 
 
34
  'file_name': file.filename,
35
  'file_data': await file.read(),
36
  'mime_type': file.content_type,
37
+ 'upload_time': datetime.utcnow()
38
  }
39
  return {"download_link": download_link, "message": "File uploaded successfully"}
40
 
41
  @app.get("/download/{upload_id}")
42
  async def download_file(upload_id: str):
43
  # Remove expired files
44
+ expire_time = datetime.utcnow() - timedelta(minutes=5)
45
  keys_to_remove = [key for key, value in data_storage.items() if value['upload_time'] < expire_time]
46
  for key in keys_to_remove:
47
  del data_storage[key]