59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
"""SDS 同步公共管道:映射订单(含内联 items)→ upsert 到 PG。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Callable
|
|
|
|
from flows.shared import resolve_courier_source
|
|
from flows.shared.db import SyncDB
|
|
from flows.shared.mappers import map_sds_goods, map_sds_order
|
|
|
|
|
|
def map_sds_records(
|
|
raw_orders: list[Any],
|
|
channel_map: dict[int, str | None],
|
|
channel_refresh: Callable[[], dict[int, str | None]] | None = None,
|
|
skip_unpaid: bool = True,
|
|
) -> tuple[list[Any], list[Any], int]:
|
|
"""映射 SDS 订单+商品。返回 (order_inputs, goods_inputs, skipped_unpaid)。"""
|
|
order_inputs: list[Any] = []
|
|
goods_inputs: list[Any] = []
|
|
refreshed: set[int] = set()
|
|
skipped_unpaid = 0
|
|
|
|
for raw in raw_orders:
|
|
courier = None
|
|
if channel_refresh is not None:
|
|
courier = resolve_courier_source(raw.logistics_id, channel_map, channel_refresh, refreshed)
|
|
try:
|
|
o = map_sds_order(raw, channel_map, courier)
|
|
except Exception as err: # noqa: BLE001
|
|
print(f"[SDS] 映射订单失败 {getattr(raw, 'no', '?')}: {err}")
|
|
continue
|
|
if skip_unpaid and o.systemStatus == "待付款":
|
|
skipped_unpaid += 1
|
|
continue
|
|
order_inputs.append(o)
|
|
for item in raw.items or []:
|
|
try:
|
|
goods_inputs.append(map_sds_goods(item, o.orderNo, o.createdAt))
|
|
except Exception as err: # noqa: BLE001
|
|
print(f"[SDS] 映射商品失败 order={o.orderNo}: {err}")
|
|
|
|
return order_inputs, goods_inputs, skipped_unpaid
|
|
|
|
|
|
def sync_sds_records(
|
|
raw_orders: list[Any], channel_map: dict[int, str | None], db: SyncDB,
|
|
channel_refresh: Callable[[], dict[int, str | None]] | None = None,
|
|
skip_unpaid: bool = True,
|
|
) -> dict[str, int]:
|
|
order_inputs, goods_inputs, skipped = map_sds_records(
|
|
raw_orders, channel_map, channel_refresh, skip_unpaid
|
|
)
|
|
stats = db.sync_orders_and_goods(order_inputs, goods_inputs)
|
|
stats["orderCount"] = len(order_inputs)
|
|
stats["goodsCount"] = len(goods_inputs)
|
|
stats["skippedUnpaid"] = skipped
|
|
return stats
|