67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
import math
|
|
|
|
from fastapi import UploadFile
|
|
|
|
from external.s3_uploader import S3Uploader
|
|
from modules.fulfillment_base.models import Product
|
|
from modules.fulfillment_base.repositories import ProductRepository
|
|
from modules.fulfillment_base.schemas.product import *
|
|
from schemas.base import PaginationSchema, PaginationInfoSchema
|
|
from services.mixins import *
|
|
|
|
|
|
class ProductService(
|
|
ServiceCreateMixin[Product, CreateProductRequest, ProductSchema],
|
|
ServiceUpdateMixin[Product, UpdateProductRequest],
|
|
ServiceDeleteMixin[Product],
|
|
):
|
|
schema_class = ProductSchema
|
|
entity_deleted_msg = "Товар успешно удален"
|
|
entity_updated_msg = "Товар успешно обновлен"
|
|
entity_created_msg = "Товар успешно создан"
|
|
|
|
def __init__(self, session: AsyncSession):
|
|
self.repository = ProductRepository(session)
|
|
|
|
async def get_all(
|
|
self,
|
|
pagination: PaginationSchema,
|
|
*filters,
|
|
) -> GetProductsResponse:
|
|
products, total_items = await self.repository.get_all(
|
|
pagination.page,
|
|
pagination.items_per_page,
|
|
*filters,
|
|
)
|
|
|
|
total_pages = 1
|
|
if pagination.items_per_page:
|
|
total_pages = math.ceil(total_items / pagination.items_per_page)
|
|
|
|
return GetProductsResponse(
|
|
items=[ProductSchema.model_validate(product) for product in products],
|
|
pagination_info=PaginationInfoSchema(
|
|
total_pages=total_pages, total_items=total_items
|
|
),
|
|
)
|
|
|
|
async def upload_image(
|
|
self, product_id: int, upload_file: UploadFile
|
|
) -> ProductUploadImageResponse:
|
|
try:
|
|
product: Product = await self.repository.get_by_id(product_id)
|
|
s3_uploader = S3Uploader()
|
|
|
|
for image in product.images:
|
|
s3_key = image.image_url.split("/")[-1]
|
|
await s3_uploader.delete_image(s3_key)
|
|
await self.repository.delete_images(product.images)
|
|
|
|
image_url = await s3_uploader.upload_from_upload_file_obj(upload_file)
|
|
await self.repository.create_image(product_id, image_url)
|
|
return ProductUploadImageResponse(
|
|
ok=True, message="Изображение успешно загружено", image_url=image_url
|
|
)
|
|
except Exception as e:
|
|
return ProductUploadImageResponse(ok=False, message=str(e))
|