调整格式

2026-05-09 14:11:02 +08:00
parent 7de3bf123c
commit 7ea35eaa9c
4 changed files with 100 additions and 61 deletions
@@ -1,6 +1,8 @@
 from __future__ import annotations

 import logging
+from collections import OrderedDict
+from collections.abc import Sequence
 from typing import Any

 from inventory_analytics.logging_utils import log_event as emit_log_event
@@ -8,7 +10,7 @@ from inventory_analytics.logging_utils import log_event as emit_log_event
 from .feishu import FeishuApiError, FeishuClient
 from .postgres import query_postgres
 from .settings import AppConfig
-from .specs import BRAND_SYNC_TARGETS, FIELD_SPECS, PRIMARY_FIELD_NAME, PRIMARY_FIELD_SOURCE, BrandSyncTarget
+from .specs import BRAND_SYNC_TARGETS, FIELD_SPECS, PRIMARY_FIELD_NAME, PRIMARY_FIELD_SOURCE, BrandSyncTarget, FieldSpec


 logger = logging.getLogger("inventory_sync.service")
@@ -22,6 +24,7 @@ LOG_FIELD_LABELS = {
    "rows": "行数",
    "source_rows": "源数据行数",
    "stage": "阶段",
+    "shop_count": "店铺数量",
    "table_count": "表数量",
    "table_name": "表名",
    "target_count": "目标数量",
@@ -32,17 +35,20 @@ def log_event(level: int, event: str, **fields: Any) -> None:
    emit_log_event(logger, LOG_FIELD_LABELS, level, event, **fields)


-def build_feishu_record(row: dict[str, Any]) -> dict[str, Any]:
+def build_feishu_record(row: dict[str, Any], field_specs: Sequence[FieldSpec] = FIELD_SPECS) -> dict[str, Any]:
    if PRIMARY_FIELD_SOURCE not in row:
        raise KeyError(f"映射缺少主字段来源列: {PRIMARY_FIELD_SOURCE}")

    fields: dict[str, Any] = {
        PRIMARY_FIELD_NAME: "" if row.get(PRIMARY_FIELD_SOURCE) is None else str(row.get(PRIMARY_FIELD_SOURCE)),
    }
-    for spec in FIELD_SPECS:
+    for spec in field_specs:
        if spec.source_name not in row:
-            raise KeyError(f"映射缺少 SQL 列: {spec.source_name} -> {spec.feishu_name}")
-        value = row.get(spec.source_name)
+            if spec.source_name in {base_spec.source_name for base_spec in FIELD_SPECS}:
+                raise KeyError(f"映射缺少 SQL 列: {spec.source_name} -> {spec.feishu_name}")
+            value = 0
+        else:
+            value = row.get(spec.source_name)
        if value is None or value == "":
            fields[spec.feishu_name] = None
        elif spec.field_type == 2:
@@ -53,6 +59,54 @@ def build_feishu_record(row: dict[str, Any]) -> dict[str, Any]:
    return {"fields": fields}


+def _to_number(value: Any) -> float:
+    if value is None or value == "":
+        return 0.0
+    return float(value)
+
+
+def build_shop_stock_field_specs(shop_names: Sequence[str]) -> list[FieldSpec]:
+    return [FieldSpec(shop_name, shop_name, 2, "0", 0) for shop_name in shop_names]
+
+
+def build_channel_inventory_field_specs(shop_names: Sequence[str]) -> list[FieldSpec]:
+    shop_stock_specs = build_shop_stock_field_specs(shop_names)
+    insert_after_field = "零售价"
+    field_specs: list[FieldSpec] = []
+    for spec in FIELD_SPECS:
+        field_specs.append(spec)
+        if spec.feishu_name == insert_after_field:
+            field_specs.extend(shop_stock_specs)
+    return field_specs
+
+
+def pivot_shop_stock_rows(rows: list[dict[str, Any]]) -> tuple[list[dict[str, Any]], list[str]]:
+    shop_names = list(OrderedDict.fromkeys(str(row.get("店铺名称") or "").strip() for row in rows if row.get("店铺名称")))
+    rows_by_code: OrderedDict[str, dict[str, Any]] = OrderedDict()
+    for row in rows:
+        code = "" if row.get(PRIMARY_FIELD_SOURCE) is None else str(row.get(PRIMARY_FIELD_SOURCE))
+        if code not in rows_by_code:
+            rows_by_code[code] = {
+                PRIMARY_FIELD_SOURCE: row.get(PRIMARY_FIELD_SOURCE),
+                "品牌名称": row.get("品牌名称"),
+                "货品简称": row.get("货品简称"),
+                "一级分类": row.get("一级分类"),
+                "商品名称": row.get("商品名称"),
+                "条形码": row.get("条形码"),
+                "零售价": row.get("零售价"),
+                "总库存": row.get("总库存"),
+                "可用总库存": 0,
+            }
+        shop_name = str(row.get("店铺名称") or "").strip()
+        if not shop_name:
+            continue
+        stock = _to_number(row.get("店铺挂单量"))
+        current_stock = _to_number(rows_by_code[code].get(shop_name))
+        rows_by_code[code][shop_name] = current_stock + stock
+        rows_by_code[code]["可用总库存"] = _to_number(rows_by_code[code].get("可用总库存")) + stock
+    return list(rows_by_code.values()), shop_names
+
+
 def run_feishu_summary(client: FeishuClient) -> dict[str, Any]:
    log_event(logging.INFO, "开始获取飞书访问令牌", stage="feishu_auth")
    access_token = client.get_tenant_access_token()
@@ -102,8 +156,10 @@ def sync_brand_target(
    if not table_id:
        raise FeishuApiError(f"目标飞书表缺少 table_id: {target.table_name}")

-    target_rows = postgres_rows or []
-    available_fields = client.ensure_target_fields(access_token, app_token, table_id)
+    source_rows = postgres_rows or []
+    target_rows, shop_names = pivot_shop_stock_rows(source_rows)
+    field_specs = build_channel_inventory_field_specs(shop_names)
+    available_fields = client.ensure_target_fields(access_token, app_token, table_id, field_specs)
    existing_records = client.list_records(access_token, app_token, table_id)
    existing_record_ids = [record.get("record_id") for record in existing_records if record.get("record_id")]
    log_event(
@@ -119,7 +175,7 @@ def sync_brand_target(
        access_token,
        app_token,
        table_id,
-        [build_feishu_record(row) for row in target_rows],
+        [build_feishu_record(row, field_specs) for row in target_rows],
    ) if target_rows else 0
    log_event(
        logging.INFO,
@@ -129,6 +185,7 @@ def sync_brand_target(
        table_name=target.table_name,
        deleted_records=deleted_count,
        created_records=created_count,
+        shop_count=len(shop_names),
    )

    return {
@@ -137,7 +194,7 @@ def sync_brand_target(
        "target_table_name": target.table_name,
        "target_table_id": table_id,
        "available_fields": available_fields,
-        "source_row_count": len(target_rows),
+        "source_row_count": len(source_rows),
        "filtered_row_count": len(target_rows),
        "deleted_record_count": deleted_count,
        "created_record_count": created_count,