Spaces:
Sleeping
Sleeping
File size: 2,039 Bytes
0643755 324bb3c 0643755 9ec2d3e 324bb3c 9ec2d3e 324bb3c 9ec2d3e 0643755 324bb3c 0643755 9ec2d3e 324bb3c 0643755 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from fastapi import FastAPI, File, UploadFile, BackgroundTasks
from fastapi.responses import HTMLResponse, FileResponse
import shutil
import os
import time
app = FastAPI()
# HTML form for file upload
html_content = """
<!DOCTYPE html>
<html>
<head>
<title>File Upload</title>
</head>
<body>
<h2>Upload a File</h2>
<form action="/uploadfile/" method="post" enctype="multipart/form-data">
<input name="file" type="file">
<input type="submit">
</form>
<h2>Uploaded Files</h2>
<ul>
{file_list}
</ul>
</body>
</html>
"""
@app.get("/", response_class=HTMLResponse)
async def read_root():
file_list = get_uploaded_files()
return html_content.format(file_list=file_list)
@app.post("/uploadfile/")
async def upload_file(file: UploadFile = File(...), background_tasks: BackgroundTasks):
# Save the uploaded file to the uploads directory
upload_directory = "uploads"
os.makedirs(upload_directory, exist_ok=True) # Create the directory if it doesn't exist
file_location = f"{upload_directory}/{file.filename}"
with open(file_location, "wb") as buffer:
shutil.copyfileobj(file.file, buffer)
# Schedule the file deletion after 10 minutes
background_tasks.add_task(delete_file_after_delay, file_location, 600) # 600 seconds = 10 minutes
return {"filename": file.filename}
def get_uploaded_files():
upload_directory = "uploads"
files = os.listdir(upload_directory)
file_links = ""
for file in files:
file_links += f'<li><a href="/files/{file}">{file}</a></li>'
return file_links
@app.get("/files/{filename}")
async def get_file(filename: str):
return FileResponse(f"uploads/{filename}")
def delete_file_after_delay(file_path: str, delay: int):
time.sleep(delay) # Wait for the specified delay
if os.path.exists(file_path):
os.remove(file_path) # Delete the file
print(f"Deleted file: {file_path}") # Log the deletion (optional)
|