diff --git a/main.py b/main.py index fbd085c..4c628ec 100644 --- a/main.py +++ b/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!"})