first commit
This commit is contained in:
40
app/api/v1/product/images.py
Normal file
40
app/api/v1/product/images.py
Normal file
@ -0,0 +1,40 @@
|
||||
import os
|
||||
from time import time
|
||||
|
||||
from fastapi import APIRouter, UploadFile, File, HTTPException, Request
|
||||
|
||||
from app import mongo
|
||||
from app.utils.response_util import response
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
STATIC_FOLDER = "static"
|
||||
PRODUCTS_FOLDER = os.path.join(STATIC_FOLDER, "images")
|
||||
os.makedirs(PRODUCTS_FOLDER, exist_ok=True)
|
||||
|
||||
|
||||
@router.post("/upload/{product_id}", tags=["Products"])
|
||||
async def upload_product_image(
|
||||
request: Request,
|
||||
product_id: int,
|
||||
upload_file: UploadFile = File(...)
|
||||
):
|
||||
start_time = time()
|
||||
|
||||
product = await mongo.products_collection.find_one({"id": product_id})
|
||||
if not product:
|
||||
raise HTTPException(status_code=404, detail="Product not found")
|
||||
|
||||
extension = os.path.splitext(upload_file.filename)[1] or ".jpg"
|
||||
filename = f"{product_id}{extension}"
|
||||
file_path = os.path.join(PRODUCTS_FOLDER, filename)
|
||||
|
||||
with open(file_path, "wb") as file:
|
||||
file.write(await upload_file.read())
|
||||
|
||||
image_url = f"{str(request.base_url).rstrip('/')}/api/files/images/{filename}"
|
||||
return response({
|
||||
"imageUrl": image_url,
|
||||
"message": "Фото успешно загружено!",
|
||||
"ok": True
|
||||
}, start_time=start_time)
|
||||
Reference in New Issue
Block a user