补全任务与测试

etl_billiards/tasks/tables_task.py

@@ -1,4 +1,85 @@
+# -*- coding: utf-8 -*-
+"""台桌档案任务"""
+
+import json
+
+from .base_task import BaseTask
+from loaders.dimensions.table import TableLoader
+from models.parsers import TypeParser
+
+
 class TablesTask(BaseTask):
-    def get_task_code(self) -> str:  # 返回 "TABLES"
-    def execute(self) -> dict:       # 拉取 /Table/GetSiteTables
-    def _parse_table(self, raw: dict) -> dict | None:
+    """同步门店台桌列表"""
+
+    def get_task_code(self) -> str:
+        return "TABLES"
+
+    def execute(self) -> dict:
+        self.logger.info("开始执行 TABLES 任务")
+        params = {"storeId": self.config.get("app.store_id")}
+
+        try:
+            records, _ = self.api.get_paginated(
+                endpoint="/Table/GetSiteTables",
+                params=params,
+                page_size=self.config.get("api.page_size", 200),
+                data_path=("data", "siteTables"),
+            )
+
+            parsed = []
+            for raw in records:
+                mapped = self._parse_table(raw)
+                if mapped:
+                    parsed.append(mapped)
+
+            loader = TableLoader(self.db)
+            inserted, updated, skipped = loader.upsert_tables(parsed)
+
+            self.db.commit()
+            counts = {
+                "fetched": len(records),
+                "inserted": inserted,
+                "updated": updated,
+                "skipped": skipped,
+                "errors": 0,
+            }
+            self.logger.info(f"TABLES 完成: {counts}")
+            return self._build_result("SUCCESS", counts)
+        except Exception:
+            self.db.rollback()
+            self.logger.error("TABLES 失败", exc_info=True)
+            raise
+
+    def _parse_table(self, raw: dict) -> dict | None:
+        table_id = TypeParser.parse_int(raw.get("id"))
+        if not table_id:
+            self.logger.warning("跳过缺少 table_id 的台桌记录: %s", raw)
+            return None
+
+        store_id = self.config.get("app.store_id")
+        return {
+            "store_id": store_id,
+            "table_id": table_id,
+            "site_id": TypeParser.parse_int(raw.get("site_id") or raw.get("siteId")),
+            "area_id": TypeParser.parse_int(
+                raw.get("site_table_area_id") or raw.get("siteTableAreaId")
+            ),
+            "area_name": raw.get("areaName") or raw.get("site_table_area_name"),
+            "table_name": raw.get("table_name") or raw.get("tableName"),
+            "table_price": TypeParser.parse_decimal(
+                raw.get("table_price") or raw.get("tablePrice")
+            ),
+            "table_status": raw.get("table_status") or raw.get("tableStatus"),
+            "table_status_name": raw.get("tableStatusName"),
+            "light_status": raw.get("light_status"),
+            "is_rest_area": raw.get("is_rest_area"),
+            "show_status": raw.get("show_status"),
+            "virtual_table": raw.get("virtual_table"),
+            "charge_free": raw.get("charge_free"),
+            "only_allow_groupon": raw.get("only_allow_groupon"),
+            "is_online_reservation": raw.get("is_online_reservation"),
+            "created_time": TypeParser.parse_timestamp(
+                raw.get("create_time") or raw.get("createTime"), self.tz
+            ),
+            "raw_data": json.dumps(raw, ensure_ascii=False),
+        }