76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
清理流程测试产生的数据：persons（测试自动创建_、测试新人物_）、chapters（t 开头的 6 位数字 id）
|
|||
|
|
用法：cd scripts/test && python process/cleanup_test_data.py
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
import re
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|||
|
|
|
|||
|
|
import requests
|
|||
|
|
|
|||
|
|
API_BASE = os.environ.get("SOUL_API_BASE", "http://localhost:8080").rstrip("/")
|
|||
|
|
ADMIN_USER = os.environ.get("SOUL_ADMIN_USERNAME", "admin")
|
|||
|
|
ADMIN_PASS = os.environ.get("SOUL_ADMIN_PASSWORD", "admin123")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
r = requests.post(
|
|||
|
|
f"{API_BASE}/api/admin",
|
|||
|
|
json={"username": ADMIN_USER, "password": ADMIN_PASS},
|
|||
|
|
timeout=10,
|
|||
|
|
)
|
|||
|
|
if not r.json().get("success") or not r.json().get("token"):
|
|||
|
|
print("登录失败")
|
|||
|
|
return 1
|
|||
|
|
token = r.json()["token"]
|
|||
|
|
headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
|||
|
|
|
|||
|
|
# 1. 删除测试 Person
|
|||
|
|
rp = requests.get(f"{API_BASE}/api/db/persons", headers=headers, timeout=10)
|
|||
|
|
if rp.json().get("success"):
|
|||
|
|
persons = rp.json().get("persons") or []
|
|||
|
|
for p in persons:
|
|||
|
|
name = p.get("name") or ""
|
|||
|
|
if name.startswith("测试自动创建_") or name.startswith("测试新人物_"):
|
|||
|
|
pid = p.get("personId")
|
|||
|
|
rd = requests.delete(
|
|||
|
|
f"{API_BASE}/api/db/persons?personId={pid}",
|
|||
|
|
headers=headers,
|
|||
|
|
timeout=10,
|
|||
|
|
)
|
|||
|
|
if rd.json().get("success"):
|
|||
|
|
print(f" 已删除 Person: {name} ({pid})")
|
|||
|
|
else:
|
|||
|
|
print(f" 删除 Person 失败: {name} - {rd.json()}")
|
|||
|
|
|
|||
|
|
# 2. 删除测试 Chapter（id 形如 t123456）
|
|||
|
|
rl = requests.get(
|
|||
|
|
f"{API_BASE}/api/db/book?action=list",
|
|||
|
|
headers=headers,
|
|||
|
|
timeout=10,
|
|||
|
|
)
|
|||
|
|
if rl.json().get("success"):
|
|||
|
|
sections = rl.json().get("sections") or []
|
|||
|
|
for s in sections:
|
|||
|
|
sid = s.get("id") or ""
|
|||
|
|
if re.match(r"^t\d{6}$", sid):
|
|||
|
|
rd = requests.delete(
|
|||
|
|
f"{API_BASE}/api/db/book?id={sid}",
|
|||
|
|
headers=headers,
|
|||
|
|
timeout=10,
|
|||
|
|
)
|
|||
|
|
if rd.json().get("success"):
|
|||
|
|
print(f" 已删除 Chapter: {sid} ({s.get('title', '')})")
|
|||
|
|
else:
|
|||
|
|
print(f" 删除 Chapter 失败: {sid} - {rd.json()}")
|
|||
|
|
|
|||
|
|
print("清理完成")
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
sys.exit(main())
|