chore: import WonderQ project
This commit is contained in:
commit
f5a9e4403a
452 files changed
+34173
No files matched your search
@@ -0,0 +1,97 @@
|
||||
from datetime import datetime
|
||||
from sqlalchemy.inspection import inspect
|
||||
|
||||
|
||||
def encode_value(value):
|
||||
if isinstance(value, datetime):
|
||||
return value.isoformat()
|
||||
if isinstance(value, list):
|
||||
return [encode_value(item) for item in value]
|
||||
if isinstance(value, dict):
|
||||
return {key: encode_value(item) for key, item in value.items()}
|
||||
return value
|
||||
|
||||
|
||||
def model_dict(instance, include: dict[str, object] | None = None) -> dict:
|
||||
data = {column.key: encode_value(getattr(instance, column.key)) for column in inspect(instance).mapper.column_attrs}
|
||||
if include:
|
||||
for key, value in include.items():
|
||||
data[key] = encode_value(value)
|
||||
return data
|
||||
|
||||
|
||||
def _sorted_aliases(destination) -> list:
|
||||
return sorted(destination.aliases or [], key=lambda item: ((item.alias or "").lower(), item.id or ""))
|
||||
|
||||
|
||||
def _sorted_images(product) -> list:
|
||||
return sorted(product.images or [], key=lambda item: (item.sortOrder, item.id or "", item.url))
|
||||
|
||||
|
||||
def _product_images(product) -> list[dict]:
|
||||
return [model_dict(image) for image in _sorted_images(product)]
|
||||
|
||||
|
||||
def _detail_sections(product) -> list:
|
||||
return encode_value(product.detailSections or [])
|
||||
|
||||
|
||||
def admin_product_dict(product) -> dict:
|
||||
return model_dict(
|
||||
product,
|
||||
{
|
||||
"tags": list(product.tags or []),
|
||||
"detailSections": _detail_sections(product),
|
||||
"destination": destination_dict(product.destination) if product.destination else None,
|
||||
"images": _product_images(product),
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
def public_product_dict(product) -> dict:
|
||||
return {
|
||||
"id": product.id,
|
||||
"sourceId": product.sourceId,
|
||||
"title": product.title,
|
||||
"subtitle": product.subtitle,
|
||||
"destination": {"id": product.destination.id, "name": product.destination.name} if product.destination else None,
|
||||
"priceAmount": product.priceAmount,
|
||||
"priceUnit": product.priceUnit,
|
||||
"tags": list(product.tags or []),
|
||||
"coverImage": product.coverImage,
|
||||
"summary": product.summary,
|
||||
"images": _product_images(product),
|
||||
"detailSections": _detail_sections(product),
|
||||
"status": product.status,
|
||||
}
|
||||
|
||||
|
||||
def product_dict(product) -> dict:
|
||||
return admin_product_dict(product)
|
||||
|
||||
|
||||
def hotel_group_dict(group) -> dict:
|
||||
data = model_dict(group)
|
||||
if not data.get("coverImage"):
|
||||
data["coverImage"] = data.get("image")
|
||||
if not data.get("image"):
|
||||
data["image"] = data.get("coverImage")
|
||||
data["tags"] = list(group.tags or [])
|
||||
return data
|
||||
|
||||
|
||||
def destination_dict(destination, include_count: bool = False, product_count: int | None = None) -> dict:
|
||||
data = model_dict(destination, {"aliases": [model_dict(alias) for alias in _sorted_aliases(destination)]})
|
||||
if include_count:
|
||||
data["_count"] = {"products": product_count if product_count is not None else len(destination.products or [])}
|
||||
return data
|
||||
|
||||
|
||||
def lead_dict(lead) -> dict:
|
||||
return model_dict(
|
||||
lead,
|
||||
{
|
||||
"sourceProduct": {"id": lead.sourceProduct.id, "title": lead.sourceProduct.title} if lead.sourceProduct else None,
|
||||
"assignedUser": {"id": lead.assignedUser.id, "name": lead.assignedUser.name} if lead.assignedUser else None,
|
||||
},
|
||||
)
|
||||
Reference in new issue
Block a user