This commit is contained in:
parent
d554d0ef14
commit
30303969bd
14
main.py
14
main.py
|
@ -29,17 +29,17 @@ async def create_s3_client():
|
|||
|
||||
@login_required
|
||||
async def upload_handler(request: Request):
|
||||
logging.debug("Received upload request")
|
||||
logger.debug("Received upload request")
|
||||
form = await request.form()
|
||||
file = form.get("file")
|
||||
|
||||
if file is None:
|
||||
logging.error("No file uploaded")
|
||||
logger.error("No file uploaded")
|
||||
return JSONResponse({"error": "No file uploaded"}, status_code=400)
|
||||
|
||||
file_name, file_extension = os.path.splitext(file.filename)
|
||||
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:
|
||||
with tempfile.NamedTemporaryFile() as tmp_file:
|
||||
|
@ -50,22 +50,22 @@ async def upload_handler(request: Request):
|
|||
tmp_file.write(chunk)
|
||||
tmp_file.flush()
|
||||
|
||||
logging.debug("Starting file upload to S3")
|
||||
logger.debug("Starting file upload to S3")
|
||||
await s3.upload_file(
|
||||
Filename=tmp_file.name,
|
||||
Bucket=STORJ_BUCKET_NAME,
|
||||
Key=key,
|
||||
ExtraArgs={"ContentType": file.content_type},
|
||||
)
|
||||
logging.debug("File upload completed")
|
||||
logger.debug("File upload completed")
|
||||
|
||||
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})
|
||||
|
||||
|
||||
async def home(request: Request):
|
||||
logging.debug("Home route called")
|
||||
logger.debug("Home route called")
|
||||
return JSONResponse({"message": "Hello World!"})
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user