diff --git a/modules/fulfillment_base/repositories/deal_product.py b/modules/fulfillment_base/repositories/deal_product.py index 8a237a8..e168048 100644 --- a/modules/fulfillment_base/repositories/deal_product.py +++ b/modules/fulfillment_base/repositories/deal_product.py @@ -5,7 +5,6 @@ from sqlalchemy.orm import joinedload, selectinload from modules.fulfillment_base.models import DealProductService from modules.fulfillment_base.models.deal_product import DealProduct -from modules.fulfillment_base.models.service import ServicesKit from modules.fulfillment_base.schemas.deal_product import ( UpdateDealProductSchema, CreateDealProductSchema, @@ -61,16 +60,3 @@ class DealProductRepository( async def delete(self, obj: DealProduct): await self.session.delete(obj) await self.session.commit() - - async def add_services_kit( - self, deal_product: DealProduct, services_kit: ServicesKit - ): - for service in services_kit.services: - deal_product_service = DealProductService( - deal_id=deal_product.deal_id, - product_id=deal_product.product_id, - service_id=service.id, - price=service.price, - ) - self.session.add(deal_product_service) - await self.session.commit() diff --git a/modules/fulfillment_base/repositories/product_service.py b/modules/fulfillment_base/repositories/product_service.py index ec650bc..dbd5e9e 100644 --- a/modules/fulfillment_base/repositories/product_service.py +++ b/modules/fulfillment_base/repositories/product_service.py @@ -4,7 +4,8 @@ from sqlalchemy import select, delete from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.orm import joinedload -from modules.fulfillment_base.models import DealProductService +from modules.fulfillment_base.models import DealProductService, DealProduct +from modules.fulfillment_base.models.service import ServicesKit from modules.fulfillment_base.schemas.product_service import * from repositories.base import BaseRepository from repositories.mixins import RepUpdateMixin @@ -89,3 +90,16 @@ class ProductServiceRepository( ) self.session.add(product_service) await self.session.commit() + + async def add_services_kit( + self, deal_product: DealProduct, services_kit: ServicesKit + ): + for service in services_kit.services: + deal_product_service = DealProductService( + deal_id=deal_product.deal_id, + product_id=deal_product.product_id, + service_id=service.id, + price=service.price, + ) + self.session.add(deal_product_service) + await self.session.commit()