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]
|