from modules.fulfillment_base.models import Marketplace
from modules.fulfillment_base.repositories import MarketplaceRepository
from modules.fulfillment_base.schemas.marketplace import *
from services.mixins import *


class MarketplaceService(
    ServiceCrudMixin[
        Marketplace,
        MarketplaceSchema,
        CreateMarketplaceRequest,
        UpdateMarketplaceRequest,
    ]
):
    schema_class = MarketplaceSchema
    entity_deleted_msg = "Маркетплейс успешно удален"
    entity_updated_msg = "Маркетплейс успешно обновлен"
    entity_created_msg = "Маркетплейс успешно создан"

    def __init__(self, session: AsyncSession):
        self.repository = MarketplaceRepository(session)

    async def get_base_marketplaces(self) -> GetBaseMarketplacesResponse:
        mps = await self.repository.get_base_marketplaces()
        return GetBaseMarketplacesResponse(
            items=[BaseMarketplaceSchema.model_validate(mp) for mp in mps]
        )