fakz9/Fulfillment-Backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: tech specification for a group of deals

Browse Source
This commit is contained in:
AlexSserb
2024-11-10 15:32:58 +04:00
parent 30ec0528a4
commit 915a1d5f28
6 changed files with 186 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
generators/deal_pdf_generator/deal_data.py
View File

@@ -4,16 +4,22 @@ from models import DealProduct, Deal, DealStatusHistory
class DealTechSpecProductData(TypedDict):
deal_products: List[DealProduct]
deal: Deal
last_status: DealStatusHistory
total_one_product: int
quantity: int
additional_info: Optional[str]
# Поле для группировки товаров с одним артикулом и вывода таблицы [Штрихкод, Размер, Кол-во, Короба]
deal_products: List[DealProduct]
# Поле для группировки товаров из нескольких сделок и вывода таблицы [Склад отгрузки, Кол-во]
warehouses_and_quantities: List[Tuple[str, int]]
class DealTechSpecData(TypedDict):
deal: Deal
deals: List[Deal]
products: Dict[str, DealTechSpecProductData]
current_status_str: str
last_status: DealStatusHistory
product_images: Tuple[str]
product_images: Tuple
deal_ids_header: str
deal_status_str: list[str]

100
generators/deal_pdf_generator/generator.py
View File

@@ -8,13 +8,46 @@ from weasyprint import HTML, CSS
from constants import DEAL_STATUS_STR, ENV, APP_PATH
from generators.deal_pdf_generator.deal_data import DealTechSpecProductData, DealTechSpecData
from models import Deal, DealProduct, DealService as DealServiceModel, Product
from models import Deal, DealProduct, DealService as DealServiceModel, Product, DealGroup
from utils.images_fetcher import fetch_images
# Генерация ключа для группировки deal_product по артикулу и услугам
def _gen_key_for_product(deal_product: DealProduct) -> str:
return f"{deal_product.product.article} - " + ",".join(
str(service.service_id) for service in deal_product.services
)
# Генерация ключа для группировки deal_product из группы сделок по артикулу, сервисам, а также товарам
def _regen_key_for_product(product_data: DealTechSpecProductData) -> str:
if len(product_data['deal_products']) == 0:
return ""
article = product_data['deal_products'][0].product.article
services_ids = ",".join(str(service.service_id) for service in product_data['deal_products'][0].services)
if len(product_data['deal_products']) == 1:
products = product_data['deal_products'][0].product_id
else:
products = ",".join(
f"{deal_product.product_id}-{deal_product.quantity}" for deal_product in product_data['deal_products']
)
return f"{article}+{services_ids}+{products}"
class DealTechSpecPdfGenerator:
def __init__(self, session: AsyncSession):
self._session = session
self.deal_doc: DealTechSpecData = {
"deals": [],
"products": {},
"product_images": (),
"deal_ids_header": "",
"deal_status_str": DEAL_STATUS_STR,
}
self.deal: Deal
@staticmethod
async def _group_deal_products_by_products(deal_products: List[DealProduct]) -> Dict[str, DealTechSpecProductData]:
@@ -23,15 +56,15 @@ class DealTechSpecPdfGenerator:
for deal_product in deal_products:
# Для группировки по артикулу и услугам
key = f"{deal_product.product.article} - " + ",".join(
str(service.service_id) for service in deal_product.services
)
key = _gen_key_for_product(deal_product)
if key not in products:
products[key] = {
"deal": deal_product.deal,
"deal_products": [deal_product],
"quantity": deal_product.quantity,
"additional_info": deal_product.product.additional_info,
"warehouses_and_quantities": [],
}
else:
products[key]["deal_products"].append(deal_product)
@@ -50,39 +83,72 @@ class DealTechSpecPdfGenerator:
selectinload(Deal.products).selectinload(DealProduct.product).selectinload(Product.barcodes),
selectinload(Deal.services).selectinload(DealServiceModel.service),
selectinload(Deal.status_history),
selectinload(Deal.group).selectinload(DealGroup.deals),
joinedload(Deal.client),
joinedload(Deal.shipping_warehouse),
)
)
return deal
async def _get_deals_by_group_id(self, group_id: int) -> List[Deal]:
group: DealGroup | None = await self._session.scalar(
select(DealGroup)
.where(DealGroup.id == group_id)
.options(
selectinload(DealGroup.deals).selectinload(Deal.products).selectinload(DealProduct.services),
selectinload(DealGroup.deals).selectinload(Deal.products)
.selectinload(DealProduct.product).selectinload(Product.barcodes),
selectinload(DealGroup.deals).selectinload(Deal.services).selectinload(DealServiceModel.service),
selectinload(DealGroup.deals).selectinload(Deal.status_history),
selectinload(DealGroup.deals).selectinload(Deal.group).selectinload(DealGroup.deals),
selectinload(DealGroup.deals).joinedload(Deal.client),
selectinload(DealGroup.deals).joinedload(Deal.shipping_warehouse),
)
)
return group.deals if group else []
def _set_deals_ids_header(self):
self.deal_doc["deal_ids_header"] = f"ID: {self.deal.id}"
if self.deal.group:
self.deal_doc["deal_ids_header"] = "ID: " + ", ".join(str(d.id) for d in self.deal.group.deals)
async def _create_deal_tech_spec_document_html(self, deal_id: int):
deal = await self._get_deal_by_id(deal_id)
if not deal:
return ""
self.deal = deal
products = await self._group_deal_products_by_products(deal.products)
self._set_deals_ids_header()
if deal.group:
deals = await self._get_deals_by_group_id(deal.group.id)
for d in deals:
self.deal_doc["deals"].append(d)
grouped_products = await self._group_deal_products_by_products(d.products)
for product in grouped_products.values():
key = _regen_key_for_product(product)
if key not in self.deal_doc["products"]:
self.deal_doc["products"][key] = product
else:
self.deal_doc["products"][key]["quantity"] += product["quantity"]
self.deal_doc["products"][key]["warehouses_and_quantities"].append((
product["deal"].shipping_warehouse.name, product["quantity"],
))
else:
self.deal_doc["deals"] = [deal]
self.deal_doc["products"] = await self._group_deal_products_by_products(deal.products)
product_urls: List[Optional[str]] = []
for product in products.values():
for product in self.deal_doc["products"].values():
if len(product["deal_products"][0].product.images) > 0:
product_urls.append(product["deal_products"][0].product.images[0].image_url)
else:
product_urls.append(None)
product_images = await fetch_images(product_urls)
document_deal_data: DealTechSpecData = {
"deal": deal,
"products": products,
"current_status_str": DEAL_STATUS_STR[deal.current_status],
"last_status": max(deal.status_history, key=lambda status: status.changed_at),
"product_images": product_images,
}
self.deal_doc["product_images"] = await fetch_images(product_urls)
template = ENV.get_template("deal/deal-tech-spec.html")
result = template.render({"data": document_deal_data, "sign_place_text": "_" * 22})
result = template.render({"data": self.deal_doc, "sign_place_text": "_" * 22})
return result
async def create_deal_tech_spec_pdf(self, deal_id) -> BytesIO: