update

scripts/migration/import_shops.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+"""生成奇成代理商店铺批量导入 SQL，不连接或写入目标 PostgreSQL。"""
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parent))
+
+from lib.shop_import import load_config, load_plan, write_errors, write_results, write_summary  # noqa: E402
+from lib.shop_sql import build_shop_sql  # noqa: E402
+
+
+def main() -> int:
+    base = Path(__file__).resolve().parent
+    parser = argparse.ArgumentParser(description="生成店铺批量导入 SQL 和审核产物")
+    parser.add_argument("--csv", default=str(base / "副本代理商信息汇总表(3).csv"), help="代理商 CSV 路径；相对路径按当前目录解析")
+    parser.add_argument("--config", default=str(base / "config/shop_bulk_import.yaml"), help="本地导入配置路径；相对路径按当前目录解析")
+    parser.add_argument("--output-dir", default=str(base / "output"), help="SQL 和审核产物目录；相对路径按当前目录解析")
+    args = parser.parse_args()
+
+    csv_path = _resolve(args.csv)
+    config_path = _resolve(args.config)
+    output_dir = _resolve(args.output_dir)
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+    config, config_errors = load_config(config_path)
+    if config_errors:
+        error_path = output_dir / "shop_bulk_import_errors.csv"
+        write_errors(error_path, config_errors)
+        print(f"配置预检失败: {len(config_errors)} 项，错误清单: {error_path}")
+        return 1
+
+    plan = load_plan(csv_path, config)
+    prefix = f"shop_bulk_import_{config.batch_id}"
+    error_path = output_dir / f"{prefix}_errors.csv"
+    result_path = output_dir / f"{prefix}_results.csv"
+    summary_path = output_dir / f"{prefix}_summary.txt"
+    sql_path = output_dir / f"{prefix}.sql"
+    write_errors(error_path, plan.errors)
+    if plan.errors:
+        write_summary(summary_path, plan, sql_path)
+        print(f"CSV 预检失败: {len(plan.errors)} 项，错误清单: {error_path}")
+        return 1
+
+    try:
+        sql = build_shop_sql(plan)
+    except ModuleNotFoundError as exc:
+        if exc.name == "bcrypt":
+            print("缺少 bcrypt 依赖，请先执行: pip install -r requirements.txt")
+            return 1
+        raise
+    sql_path.write_text(sql, encoding="utf-8")
+    write_results(result_path, plan.results)
+    write_summary(summary_path, plan, sql_path)
+    print(f"SQL 已生成: {sql_path}")
+    print(f"审核结果: {result_path}")
+    print(f"摘要: {summary_path}")
+    print(f"计划导入: {len(plan.rows)} 条，父子关系: {sum(row.parent_line_no > 0 for row in plan.rows)} 条")
+    return 0
+
+
+def _resolve(value: str) -> Path:
+    return Path(value).resolve()
+
+
+if __name__ == "__main__":
+    sys.exit(main())