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

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

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

14
main.py
View File

@ -29,17 +29,17 @@ async def create_s3_client():
@login_required @login_required
async def upload_handler(request: Request): async def upload_handler(request: Request):
logging.debug("Received upload request") logger.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:
logging.error("No file uploaded") logger.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
logging.debug(f"Generated file key: {key}") logger.debug(f"Generated file key: {key}")
async with create_s3_client() as s3: async with create_s3_client() as s3:
with tempfile.NamedTemporaryFile() as tmp_file: with tempfile.NamedTemporaryFile() as tmp_file:
@ -50,22 +50,22 @@ async def upload_handler(request: Request):
tmp_file.write(chunk) tmp_file.write(chunk)
tmp_file.flush() tmp_file.flush()
logging.debug("Starting file upload to S3") logger.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},
) )
logging.debug("File upload completed") logger.debug("File upload completed")
url = f"http://{CDN_DOMAIN}/{key}" url = f"http://{CDN_DOMAIN}/{key}"
logging.info(f"File uploaded successfully: {url}") logger.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):
logging.debug("Home route called") logger.debug("Home route called")
return JSONResponse({"message": "Hello World!"}) return JSONResponse({"message": "Hello World!"})