105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
"""SDS 100前缀刷新:备货中/待确认且 productionNo 以 100 开头的订单,工厂分配后用新数据替换。
|
|
|
|
对应 sds-refresh-100-prefix。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from prefect import flow, task
|
|
|
|
from api.sds.models import OrderPageRequest
|
|
from flows.shared import (
|
|
SyncDB, create_sds_client, get_channel_type_map,
|
|
refresh_channel_type_map, resolve_courier_source,
|
|
)
|
|
from flows.shared.mappers import map_sds_goods, map_sds_order
|
|
|
|
TARGET_STATUSES = ["备货中", "待确认"]
|
|
KEYWORDS_BATCH = 200
|
|
|
|
|
|
def _has_100_prefix_item(raw) -> bool:
|
|
for item in raw.items or []:
|
|
prod_no = (item.no or item.id) if hasattr(item, "no") else None
|
|
if prod_no and str(prod_no).startswith("100"):
|
|
return True
|
|
return False
|
|
|
|
|
|
@task(retries=3, retry_delay_seconds=[3, 6, 9])
|
|
def fetch_remote(order_nos: list[str]) -> dict:
|
|
client = create_sds_client()
|
|
try:
|
|
remote: dict[str, object] = {}
|
|
for i in range(0, len(order_nos), KEYWORDS_BATCH):
|
|
batch = order_nos[i : i + KEYWORDS_BATCH]
|
|
resp = client.get_orders(OrderPageRequest(keywords="\n".join(batch), all=True))
|
|
for o in resp.records:
|
|
if o.no:
|
|
remote[o.no] = o
|
|
return remote
|
|
finally:
|
|
client.close()
|
|
|
|
|
|
@task
|
|
def upsert_records(remote_map: dict, target_order_nos: list[str]) -> dict:
|
|
client = create_sds_client()
|
|
try:
|
|
channel_map = get_channel_type_map(client)
|
|
refreshed: set[int] = set()
|
|
|
|
order_inputs: list = []
|
|
goods_inputs: list = []
|
|
skipped_pending = skipped_unpaid = 0
|
|
|
|
for no in target_order_nos:
|
|
raw = remote_map.get(no)
|
|
if not raw:
|
|
continue
|
|
if _has_100_prefix_item(raw):
|
|
skipped_pending += 1
|
|
continue
|
|
courier = resolve_courier_source(raw.logistics_id, channel_map,
|
|
lambda: refresh_channel_type_map(client), refreshed)
|
|
try:
|
|
o = map_sds_order(raw, channel_map, courier)
|
|
except Exception as err: # noqa: BLE001
|
|
print(f"[SDS 100] 映射失败 {no}: {err}")
|
|
continue
|
|
if 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: # noqa: BLE001
|
|
pass
|
|
|
|
stats = SyncDB().sync_orders_and_goods(order_inputs, goods_inputs)
|
|
stats["skippedStillPending"] = skipped_pending
|
|
stats["skippedUnpaid"] = skipped_unpaid
|
|
return stats
|
|
finally:
|
|
client.close()
|
|
|
|
|
|
@flow(name="sds-refresh-100-prefix", log_prints=True)
|
|
def sds_refresh_100_prefix() -> dict:
|
|
db = SyncDB()
|
|
targets = db.get_orders_with_100_prefix("SDS", TARGET_STATUSES)
|
|
order_nos = list({o for o, _ in targets})
|
|
print(f"SDS 100前缀目标订单 {len(order_nos)} 个")
|
|
if not order_nos:
|
|
return {"targetCount": 0, "refreshed": 0}
|
|
|
|
remote = fetch_remote(order_nos)
|
|
stats = upsert_records(remote, order_nos)
|
|
print(f"完成: 刷新 {stats['orderCount']} / 仍待分配 {stats.get('skippedStillPending',0)} / 待付款 {stats.get('skippedUnpaid',0)}")
|
|
return {"targetCount": len(order_nos), **stats}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sds_refresh_100_prefix.serve(name="sds-refresh-100-prefix", cron="30 */2 * * *")
|