fix: moved add_services_kit repo method
This commit is contained in:
@ -5,7 +5,6 @@ from sqlalchemy.orm import joinedload, selectinload
|
|||||||
|
|
||||||
from modules.fulfillment_base.models import DealProductService
|
from modules.fulfillment_base.models import DealProductService
|
||||||
from modules.fulfillment_base.models.deal_product import DealProduct
|
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 (
|
from modules.fulfillment_base.schemas.deal_product import (
|
||||||
UpdateDealProductSchema,
|
UpdateDealProductSchema,
|
||||||
CreateDealProductSchema,
|
CreateDealProductSchema,
|
||||||
@ -61,16 +60,3 @@ class DealProductRepository(
|
|||||||
async def delete(self, obj: DealProduct):
|
async def delete(self, obj: DealProduct):
|
||||||
await self.session.delete(obj)
|
await self.session.delete(obj)
|
||||||
await self.session.commit()
|
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()
|
|
||||||
|
|||||||
@ -4,7 +4,8 @@ from sqlalchemy import select, delete
|
|||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import joinedload
|
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 modules.fulfillment_base.schemas.product_service import *
|
||||||
from repositories.base import BaseRepository
|
from repositories.base import BaseRepository
|
||||||
from repositories.mixins import RepUpdateMixin
|
from repositories.mixins import RepUpdateMixin
|
||||||
@ -89,3 +90,16 @@ class ProductServiceRepository(
|
|||||||
)
|
)
|
||||||
self.session.add(product_service)
|
self.session.add(product_service)
|
||||||
await self.session.commit()
|
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()
|
||||||
|
|||||||
Reference in New Issue
Block a user