mirror of
https://github.com/primedigitaltech/market-assistant.git
synced 2026-07-24 01:01:40 +08:00
6280e436d8
- 统一 csv_schema 与搜索/合并/评价/商详导出为纯中文列名，入库与视图按中文表头解析 - 竞品分析报告兼容新旧表头；新增 csv_header_rewrite 与 rewrite_pipeline_csv_headers 管理命令 - 调整 pipeline 至 jd、llm、reporting、demos 子包并更新任务与测试引用 - 新增购买者优惠摘要抽取、合并表 regen/ingest 命令、0016 迁移及相关测试 Made-with: Cursor
129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
将磁盘上已存在的流水线批次目录（``data/JD/pipeline_runs/...``）导入数据库：
|
||
|
||
- ``pc_search_export.csv`` / ``detail_ware_export.csv`` / ``comments_flat.csv``
|
||
- ``keyword_pipeline_merged.csv``（合并宽表 + JdProduct / JdProductSnapshot）
|
||
|
||
用法（在 ``backend`` 目录下）::
|
||
|
||
python manage.py ingest_pipeline_dataset --run-dir pipeline_runs/20260413_104252_低GI
|
||
|
||
或绝对路径（仍须在 ``data/JD`` 下）::
|
||
|
||
python manage.py ingest_pipeline_dataset --run-dir "D:/.../data/JD/pipeline_runs/xxx"
|
||
|
||
绑定已有 ``PipelineJob``::
|
||
|
||
python manage.py ingest_pipeline_dataset --job-id 12 --run-dir pipeline_runs/xxx
|
||
|
||
新建任务并入库（关键词优先读 ``run_meta.json``）::
|
||
|
||
python manage.py ingest_pipeline_dataset --create --run-dir pipeline_runs/xxx --keyword 低GI
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
from pathlib import Path
|
||
|
||
from django.conf import settings
|
||
from django.core.management.base import BaseCommand, CommandError
|
||
|
||
from pipeline.ingest import FILE_MERGED_CSV, ingest_job_full, resolve_and_validate_run_dir
|
||
from pipeline.models import JobStatus, PipelineJob
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = "将已有 pipeline run 目录下的 CSV 导入数据库（搜索/详情/评价/合并表与商品快照）。"
|
||
|
||
def add_arguments(self, parser) -> None:
|
||
parser.add_argument(
|
||
"--run-dir",
|
||
type=str,
|
||
required=True,
|
||
help="相对 data/JD 的子路径，或位于 data/JD 下的绝对路径",
|
||
)
|
||
parser.add_argument(
|
||
"--job-id",
|
||
type=int,
|
||
default=None,
|
||
help="绑定到已有 PipelineJob；未给则须配合 --create",
|
||
)
|
||
parser.add_argument(
|
||
"--create",
|
||
action="store_true",
|
||
help="新建 PipelineJob（success）并写入 run_dir 后入库",
|
||
)
|
||
parser.add_argument(
|
||
"--keyword",
|
||
type=str,
|
||
default="",
|
||
help="与 --create 合用；默认尝试从 run_meta.json 读取 keyword",
|
||
)
|
||
|
||
def handle(self, *args, **options):
|
||
if not (settings.LOW_GI_PROJECT_ROOT or "").strip():
|
||
raise CommandError("请在 .env 中配置 LOW_GI_PROJECT_ROOT")
|
||
|
||
raw = str(options["run_dir"] or "").strip()
|
||
try:
|
||
run_path = resolve_and_validate_run_dir(raw)
|
||
except ValueError as e:
|
||
raise CommandError(str(e)) from e
|
||
|
||
merged = run_path / FILE_MERGED_CSV
|
||
if not merged.is_file():
|
||
self.stdout.write(
|
||
self.style.WARNING(
|
||
f"缺少 {FILE_MERGED_CSV}，仍将尝试导入搜索/详情/评价（合并表与快照会跳过或报错）。"
|
||
)
|
||
)
|
||
|
||
job_id = options.get("job_id")
|
||
create = bool(options.get("create"))
|
||
kw_in = (options.get("keyword") or "").strip()
|
||
|
||
if job_id and create:
|
||
raise CommandError("请只使用 --job-id 或 --create 之一")
|
||
|
||
if create:
|
||
meta_kw = ""
|
||
meta_path = run_path / "run_meta.json"
|
||
if meta_path.is_file():
|
||
try:
|
||
meta = json.loads(meta_path.read_text(encoding="utf-8"))
|
||
if isinstance(meta, dict):
|
||
meta_kw = str(meta.get("keyword") or "").strip()
|
||
except (json.JSONDecodeError, OSError):
|
||
pass
|
||
keyword = kw_in or meta_kw or "imported"
|
||
job = PipelineJob.objects.create(
|
||
platform="jd",
|
||
keyword=keyword[:256],
|
||
status=JobStatus.SUCCESS,
|
||
run_dir=str(run_path),
|
||
)
|
||
self.stdout.write(self.style.NOTICE(f"已创建任务 id={job.id} keyword={job.keyword!r}"))
|
||
elif job_id:
|
||
job = PipelineJob.objects.filter(pk=job_id).first()
|
||
if not job:
|
||
raise CommandError(f"找不到 PipelineJob id={job_id}")
|
||
job.run_dir = str(run_path)
|
||
job.save(update_fields=["run_dir", "updated_at"])
|
||
self.stdout.write(self.style.NOTICE(f"已更新任务 id={job.id} 的 run_dir"))
|
||
else:
|
||
raise CommandError("请指定 --job-id 绑定已有任务，或使用 --create 新建任务")
|
||
|
||
try:
|
||
stats = ingest_job_full(job)
|
||
except FileNotFoundError as e:
|
||
raise CommandError(str(e)) from e
|
||
|
||
self.stdout.write(self.style.SUCCESS(json.dumps(stats, ensure_ascii=False, indent=2)))
|
||
self.stdout.write(
|
||
self.style.NOTICE(
|
||
f"完成。前端可打开任务 {job.id}，数据集接口："
|
||
f"/api/pipeline/jobs/{job.id}/dataset/summary/ 等。"
|
||
)
|
||
)
|