discours.io/uploader
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

await-fix
All checks were successful
Deploy on Push / deploy (push) Successful in 31s
Details

Browse Source
This commit is contained in:
Untone 2024-05-06 14:29:27 +03:00
parent 30303969bd
commit aa49f26689
1 changed files with 24 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
main.py
View File

@ -1,3 +1,4 @@
import logging
import os import os
import tempfile import tempfile
import uuid import uuid
@ -7,7 +8,6 @@ from starlette.responses import JSONResponse
from starlette.routing import Route from starlette.routing import Route
from starlette.requests import Request from starlette.requests import Request
from auth import login_required from auth import login_required
from logger import root_logger as logger
from settings import ( from settings import (
PORT, PORT,
@ -26,22 +26,21 @@ async def create_s3_client():
endpoint_url=STORJ_END_POINT, endpoint_url=STORJ_END_POINT,
) )
@login_required @login_required
async def upload_handler(request: Request): async def upload_handler(request: Request):
logger.debug("Received upload request") logging.debug("Received upload request")
form = await request.form() form = await request.form()
file = form.get("file") file = form.get("file")
if file is None: if file is None:
logger.error("No file uploaded") logging.error("No file uploaded")
return JSONResponse({"error": "No file uploaded"}, status_code=400) return JSONResponse({"error": "No file uploaded"}, status_code=400)
file_name, file_extension = os.path.splitext(file.filename) file_name, file_extension = os.path.splitext(file.filename)
key = str(uuid.uuid4()) + file_extension key = str(uuid.uuid4()) + file_extension
logger.debug(f"Generated file key: {key}") logging.debug(f"Generated file key: {key}")
async with create_s3_client() as s3: s3 = await create_s3_client()
with tempfile.NamedTemporaryFile() as tmp_file: with tempfile.NamedTemporaryFile() as tmp_file:
while True: while True:
chunk = await file.read(8192) chunk = await file.read(8192)
@ -50,22 +49,22 @@ async def upload_handler(request: Request):
tmp_file.write(chunk) tmp_file.write(chunk)
tmp_file.flush() tmp_file.flush()
logger.debug("Starting file upload to S3") logging.debug("Starting file upload to S3")
await s3.upload_file( await s3.upload_file(
Filename=tmp_file.name, Filename=tmp_file.name,
Bucket=STORJ_BUCKET_NAME, Bucket=STORJ_BUCKET_NAME,
Key=key, Key=key,
ExtraArgs={"ContentType": file.content_type}, ExtraArgs={"ContentType": file.content_type},
) )
logger.debug("File upload completed") logging.debug("File upload completed")
url = f"http://{CDN_DOMAIN}/{key}" url = f"http://{CDN_DOMAIN}/{key}"
logger.info(f"File uploaded successfully: {url}") logging.info(f"File uploaded successfully: {url}")
return JSONResponse({"url": url, "originalFilename": file.filename}) return JSONResponse({"url": url, "originalFilename": file.filename})
async def home(request: Request): async def home(request: Request):
logger.debug("Home route called") logging.debug("Home route called")
return JSONResponse({"message": "Hello World!"}) return JSONResponse({"message": "Hello World!"})
@ -75,4 +74,3 @@ routes = [
] ]
app = Starlette(debug=True, routes=routes) app = Starlette(debug=True, routes=routes)