recycle-docker/backend/routers/admin_template.py

import json
from fastapi import APIRouter, Depends, Request, Form
from fastapi.responses import HTMLResponse, RedirectResponse
from sqlalchemy import text  # 新增这一行
from sqlalchemy.orm import Session
from database import SessionLocal
from services.template_builder import build_estimate_packet

router = APIRouter(prefix="/admin/template", tags=["template"])


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


@router.get("/preview", response_class=HTMLResponse)
def preview_template(
    request: Request,
    base_template_id: int,
    db: Session = Depends(get_db)
):
    packet = build_estimate_packet(db, base_template_id)

    json_data = {
        "templateId": base_template_id,
        "template": packet
    }

    # ------- HTML 渲染 -------
    html = f"""
    <h2>模板预览（Template ID：{base_template_id}）</h2>
    <div style="display:flex;gap:30px">

      <div style="width:55%">
    """

    for step in packet:
        html += f"""
        <div style="border:1px solid #ccc;padding:10px;margin-bottom:10px">
          <h3>Step {step['step']} - {step['stepName']}</h3>
        """

        for prop in step["properties"]:
            html += f"""
            <div style="margin-left:20px">
              <b>{prop['name']}</b>
              {"(多选)" if prop.get("isMulti") else ""}
              {"(必填)" if prop.get("required") else ""}
              <ul>
            """

            for v in prop["values"]:
                html += f"""
                <li>
                  {v['valueText']}
                  {"✅" if v.get("isNormal") else "❌"}
                </li>
                """

            html += "</ul></div>"

        html += "</div>"

    html += f"""
      </div>

      <div style="width:45%">
        <h3>JSON 输出</h3>
        <pre style="background:#f6f6f6;padding:10px;max-height:800px;overflow:auto">
{json.dumps(json_data, ensure_ascii=False, indent=2)}
        </pre>
      </div>

    </div>
    """

    return html
# ---------------- 页面 ----------------
@router.get("/generate", response_class=HTMLResponse)
def generate_page(request: Request, db: Session = Depends(get_db)):
    machines = db.execute(text("""
      SELECT machine_id, brand_name, name
      FROM t_machine
      ORDER BY brand_name, name
      LIMIT 200
    """)).fetchall()

    templates = [
        (99181, "iPhone 国行基础模板"),
        (99198, "安卓通用模板"),
        (99197, "折叠屏模板"),
    ]

    html = """
    <h2>生成机型模板</h2>

    <form method="post">
      <h3>① 选择机型</h3>
      <select name="machine_id">
    """

    for m in machines:
        html += f'<option value="{m.machine_id}">{m.brand_name} - {m.name}</option>'

    html += """
      </select>

      <h3>② 选择基础模板</h3>
      <select name="base_template_id">
    """

    for tid, tname in templates:
        html += f'<option value="{tid}">{tid} - {tname}</option>'

    html += """
      </select>

      <h3>③ Step1 补充配置（手动输入，逗号分隔）</h3>

      容量（capacity）：<br>
      <input name="capacity" placeholder="128G,256G,512G"><br><br>

      颜色（color）：<br>
      <input name="color" placeholder="黑色,白色,蓝色"><br><br>

      保修（warranty）：<br>
      <input name="warranty" placeholder="无保修,官方保修"><br><br>

      <button type="submit">④ 生成模板</button>
    </form>
    """

    return html

@router.post("/generate")
def generate_submit(
    machine_id: int = Form(...),
    base_template_id: int = Form(...),
    capacity: str = Form(""),
    color: str = Form(""),
    warranty: str = Form(""),
    db: Session = Depends(get_db)
):
    # 1️⃣ 基础模板（step1_attr + release_option）
    packet = build_estimate_packet(db, base_template_id)

    # 2️⃣ 补充 step1（人工输入）
    step1 = packet[0]["properties"]

    def append_property(key, name, values):
        if not values:
            return
        step1.append({
            "id": key,
            "name": name,
            "required": False,
            "isMulti": False,
            "values": [
                {
                    "valueId": f"manual_{v.strip()}",
                    "valueText": v.strip(),
                    "isNormal": True
                }
                for v in values.split(",") if v.strip()
            ]
        })

    append_property("capacity", "容量", capacity)
    append_property("color", "颜色", color)
    append_property("warranty", "保修", warranty)

    # 3️⃣ 保存模板
    db.execute(text("""
      INSERT INTO machine_temp(machine_id,temp_type,estimate_packet)
      VALUES (:mid,'00',:json)
      ON DUPLICATE KEY UPDATE estimate_packet=:json
    """), {
        "mid": machine_id,
        "json": json.dumps({
            "templateId": base_template_id,
            "template": packet
        }, ensure_ascii=False)
    })

    db.commit()
    return RedirectResponse("/admin/template/generate", status_code=302)