Update app.py
Browse files
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.
|
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.
|
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.
|
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]
|