lidf
c5e2a58258
- 删除 VOC_DATA_DIR / get_voc_conn（不再跨云直读 SQLite） - 案例 DB 自带 comments 表，自包含所有数据 - 新增 POST /import-voc：通过 VOC 公网 API 导入评论 - VOC_API_BASE 环境变量控制 API 地址 - 新增 httpx 依赖
267 lines
8.7 KiB
Python
267 lines
8.7 KiB
Python
"""
|
||
黑手党提案 — 独立后端（阿里云内闭环）
|
||
|
||
FastAPI 服务，端口 8093。
|
||
VOC 数据通过公网 API 导入，不直读 VOC DB。
|
||
"""
|
||
import os
|
||
import logging
|
||
import httpx
|
||
|
||
from fastapi import FastAPI, Header, HTTPException, Query
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from pydantic import BaseModel
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
from db import get_case_conn, init_case_db, list_cases as _list_cases, DATA_DIR, VOC_API_BASE
|
||
|
||
logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(name)s] %(levelname)s %(message)s")
|
||
logger = logging.getLogger("mafia")
|
||
|
||
app = FastAPI(title="黑手党提案后端", version="2.0.0", description="独立后端：阿里云内闭环，VOC 通过 API 导入")
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
|
||
# ═══════════ Models ═══════════
|
||
|
||
class CreateCaseRequest(BaseModel):
|
||
brandName: str
|
||
category: str = ""
|
||
focusProduct: str = ""
|
||
competitors: str = "[]"
|
||
vocResearchId: str = None
|
||
|
||
|
||
class LinkVocRequest(BaseModel):
|
||
vocResearchId: str
|
||
|
||
|
||
# ═══════════ 案例管理 ═══════════
|
||
|
||
@app.post("/api/cases")
|
||
async def create_case(req: CreateCaseRequest):
|
||
case_id = init_case_db(
|
||
brand_name=req.brandName,
|
||
category=req.category,
|
||
focus_product=req.focusProduct,
|
||
competitors=req.competitors,
|
||
voc_research_id=req.vocResearchId,
|
||
)
|
||
return {"caseId": case_id}
|
||
|
||
|
||
@app.get("/api/cases")
|
||
async def get_cases():
|
||
return _list_cases()
|
||
|
||
|
||
@app.get("/api/cases/{case_id}")
|
||
async def get_case(case_id: str):
|
||
try:
|
||
with get_case_conn(case_id) as conn:
|
||
card = conn.execute("SELECT * FROM case_card LIMIT 1").fetchone()
|
||
comment_count = conn.execute("SELECT count(*) FROM comments").fetchone()[0]
|
||
ude_count = conn.execute("SELECT count(*) FROM ude_sentences").fetchone()[0]
|
||
cluster_count = conn.execute("SELECT count(*) FROM ude_clusters").fetchone()[0]
|
||
if not card:
|
||
raise HTTPException(404, "案例不存在")
|
||
return {
|
||
"caseId": case_id, **dict(card),
|
||
"commentCount": comment_count,
|
||
"udeCount": ude_count,
|
||
"clusterCount": cluster_count,
|
||
}
|
||
except FileNotFoundError:
|
||
raise HTTPException(404, "案例不存在")
|
||
|
||
|
||
@app.delete("/api/cases/{case_id}")
|
||
async def delete_case(case_id: str):
|
||
path = DATA_DIR / f"{case_id}.db"
|
||
if path.exists():
|
||
path.unlink()
|
||
return {"deleted": True}
|
||
raise HTTPException(404, "案例不存在")
|
||
|
||
|
||
# ═══════════ VOC 导入（跨云 API） ═══════════
|
||
|
||
@app.post("/api/cases/{case_id}/link-voc")
|
||
async def link_voc(case_id: str, req: LinkVocRequest):
|
||
"""关联 VOC 研究 ID"""
|
||
try:
|
||
with get_case_conn(case_id) as conn:
|
||
conn.execute("UPDATE case_card SET voc_research_id = ?", (req.vocResearchId,))
|
||
conn.commit()
|
||
except FileNotFoundError:
|
||
raise HTTPException(404, "案例不存在")
|
||
return {"linked": True, "vocResearchId": req.vocResearchId}
|
||
|
||
|
||
@app.post("/api/cases/{case_id}/import-voc")
|
||
async def import_voc(case_id: str, page: int = Query(1), pageSize: int = Query(100)):
|
||
"""从 VOC 公网 API 拉取评论数据，存入本地案例 DB"""
|
||
try:
|
||
with get_case_conn(case_id) as conn:
|
||
card = conn.execute("SELECT voc_research_id, voc_api_base FROM case_card LIMIT 1").fetchone()
|
||
except FileNotFoundError:
|
||
raise HTTPException(404, "案例不存在")
|
||
|
||
if not card or not card["voc_research_id"]:
|
||
raise HTTPException(400, "未关联 VOC 研究，请先调用 link-voc")
|
||
|
||
voc_rid = card["voc_research_id"]
|
||
api_base = card["voc_api_base"] or VOC_API_BASE
|
||
|
||
# 从 VOC API 拉取（只读，不需要 TikHub Key）
|
||
total_imported = 0
|
||
current_page = page
|
||
|
||
async with httpx.AsyncClient(timeout=30) as client:
|
||
while True:
|
||
url = f"{api_base}/{voc_rid}/voc-list?page={current_page}&page_size={pageSize}"
|
||
try:
|
||
resp = await client.get(url)
|
||
if resp.status_code != 200:
|
||
logger.warning(f"[Import] VOC API 返回 {resp.status_code}: {resp.text[:100]}")
|
||
break
|
||
data = resp.json()
|
||
except Exception as e:
|
||
logger.error(f"[Import] VOC API 请求失败: {e}")
|
||
break
|
||
|
||
items = data.get("items") or data.get("data") or []
|
||
if not items:
|
||
break
|
||
|
||
with get_case_conn(case_id) as conn:
|
||
for item in items:
|
||
text = item.get("text", "")
|
||
if len(text) < 10:
|
||
continue
|
||
try:
|
||
conn.execute(
|
||
"INSERT OR IGNORE INTO comments (voc_id, platform, text, like_count, published_at) VALUES (?,?,?,?,?)",
|
||
(
|
||
item.get("id"),
|
||
item.get("platform", ""),
|
||
text,
|
||
item.get("like_count", 0),
|
||
item.get("published_at", ""),
|
||
)
|
||
)
|
||
total_imported += 1
|
||
except Exception:
|
||
pass
|
||
conn.commit()
|
||
|
||
total = data.get("total", 0)
|
||
if current_page * pageSize >= total:
|
||
break
|
||
current_page += 1
|
||
|
||
# 更新统计
|
||
with get_case_conn(case_id) as conn:
|
||
local_count = conn.execute("SELECT count(*) FROM comments").fetchone()[0]
|
||
|
||
return {
|
||
"imported": total_imported,
|
||
"totalLocal": local_count,
|
||
"vocResearchId": voc_rid,
|
||
"pagesProcessed": current_page - page + 1,
|
||
}
|
||
|
||
|
||
@app.get("/api/cases/{case_id}/comments")
|
||
async def get_comments(case_id: str, page: int = 1, pageSize: int = 50):
|
||
"""查看本地导入的评论"""
|
||
try:
|
||
with get_case_conn(case_id) as conn:
|
||
total = conn.execute("SELECT count(*) FROM comments").fetchone()[0]
|
||
rows = conn.execute("""
|
||
SELECT id, voc_id, platform, text, like_count, published_at
|
||
FROM comments ORDER BY like_count DESC
|
||
LIMIT ? OFFSET ?
|
||
""", (pageSize, (page - 1) * pageSize)).fetchall()
|
||
except FileNotFoundError:
|
||
raise HTTPException(404, "案例不存在")
|
||
return {"total": total, "page": page, "items": [dict(r) for r in rows]}
|
||
|
||
|
||
# ═══════════ UDE 分析 ═══════════
|
||
|
||
@app.post("/api/cases/{case_id}/ude/extract")
|
||
async def extract_ude(case_id: str, limit: int = Query(0)):
|
||
from tools.ude_extract import run_ude_extraction
|
||
try:
|
||
result = await run_ude_extraction(case_id, limit)
|
||
except FileNotFoundError as e:
|
||
raise HTTPException(404, str(e))
|
||
return result
|
||
|
||
|
||
@app.post("/api/cases/{case_id}/ude/cluster")
|
||
async def cluster_ude(
|
||
case_id: str,
|
||
eps: float = Query(0.25),
|
||
minSamples: int = Query(3),
|
||
x_dashscope_key: str = Header(None),
|
||
):
|
||
from tools.ude_extract import run_clustering
|
||
key = x_dashscope_key or os.getenv("DASHSCOPE_API_KEY", "")
|
||
try:
|
||
result = run_clustering(case_id, eps, minSamples, dashscope_key=key)
|
||
except FileNotFoundError as e:
|
||
raise HTTPException(404, str(e))
|
||
return result
|
||
|
||
|
||
@app.get("/api/cases/{case_id}/ude/clusters")
|
||
async def get_clusters(case_id: str):
|
||
try:
|
||
with get_case_conn(case_id) as conn:
|
||
clusters = conn.execute(
|
||
"SELECT * FROM ude_clusters ORDER BY coverage DESC"
|
||
).fetchall()
|
||
except FileNotFoundError:
|
||
raise HTTPException(404, "案例不存在")
|
||
return [dict(r) for r in clusters]
|
||
|
||
|
||
@app.get("/api/cases/{case_id}/ude/coverage")
|
||
async def get_coverage(case_id: str):
|
||
from tools.ude_extract import run_coverage_scan
|
||
try:
|
||
result = run_coverage_scan(case_id)
|
||
except FileNotFoundError as e:
|
||
raise HTTPException(404, str(e))
|
||
return result
|
||
|
||
|
||
# ═══════════ 健康检查 ═══════════
|
||
|
||
@app.get("/api/health")
|
||
async def health():
|
||
return {
|
||
"status": "ok",
|
||
"version": "2.0.0",
|
||
"architecture": "independent (Aliyun self-contained)",
|
||
"vocApiBase": VOC_API_BASE,
|
||
"caseDataDir": str(DATA_DIR),
|
||
}
|
||
|
||
|
||
# ═══════════ 启动 ═══════════
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
port = int(os.getenv("PORT", "8093"))
|
||
uvicorn.run(app, host="0.0.0.0", port=port)
|