321 lines
11 KiB
Python
321 lines
11 KiB
Python
 
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""战备(SQLite)数据存取层。
|
|||
|
|
|
|||
|
|
把原来写在 data/equipment.json 的战备数据迁移到 SQLite，供
|
|||
|
|
- equipment.py 随机/组合抽取
|
|||
|
|
- 战备管理后台(stratagem_admin.py) 的读/写/导入/导出
|
|||
|
|
|
|||
|
|
三层语义弱化为单表 + is_custom 标记：
|
|||
|
|
- 从 data/equipment.json 首次播种的记录 is_custom=0（基础库）
|
|||
|
|
- 管理后台手动新增的记录 is_custom=1
|
|||
|
|
- 手动编辑都是直接改行，不另建覆盖层（当前不做 API 同步，因此单表即可）
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import sqlite3
|
|||
|
|
import time
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Any
|
|||
|
|
|
|||
|
|
from nonebot import logger
|
|||
|
|
|
|||
|
|
basic_path = Path(__file__).resolve().parent
|
|||
|
|
DATA_DIR = basic_path / "data"
|
|||
|
|
DB_PATH = DATA_DIR / "equipment.db"
|
|||
|
|
SEED_PATH = DATA_DIR / "archive" / "equipment.json"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _connect() -> sqlite3.Connection:
|
|||
|
|
conn = sqlite3.connect(DB_PATH)
|
|||
|
|
conn.row_factory = sqlite3.Row
|
|||
|
|
return conn
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _now() -> str:
|
|||
|
|
return time.strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _load_seed() -> list[dict[str, Any]]:
|
|||
|
|
"""读取随包的 equipment.json 作为初始基础库。"""
|
|||
|
|
if not SEED_PATH.exists():
|
|||
|
|
return []
|
|||
|
|
try:
|
|||
|
|
data = json.loads(SEED_PATH.read_text(encoding="utf-8"))
|
|||
|
|
return data if isinstance(data, list) else []
|
|||
|
|
except Exception as e: # noqa: BLE001
|
|||
|
|
logger.warning(f"读取 equipment.json 失败，跳过播种: {e}")
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
|
|||
|
|
def init_db() -> None:
|
|||
|
|
"""建表（WAL 便于网页与 bot 并发读写），空表时从 equipment.json 播种。"""
|
|||
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|||
|
|
with _connect() as conn:
|
|||
|
|
conn.execute("PRAGMA journal_mode=WAL")
|
|||
|
|
conn.execute(
|
|||
|
|
"""
|
|||
|
|
CREATE TABLE IF NOT EXISTS stratagems (
|
|||
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|||
|
|
name TEXT NOT NULL,
|
|||
|
|
name_en TEXT NOT NULL DEFAULT '',
|
|||
|
|
type TEXT NOT NULL DEFAULT 'blue',
|
|||
|
|
backpack INTEGER NOT NULL DEFAULT 0,
|
|||
|
|
icon TEXT NOT NULL DEFAULT '',
|
|||
|
|
alias TEXT NOT NULL DEFAULT '[]',
|
|||
|
|
is_custom INTEGER NOT NULL DEFAULT 0,
|
|||
|
|
created_at TEXT,
|
|||
|
|
updated_at TEXT,
|
|||
|
|
UNIQUE(name)
|
|||
|
|
)
|
|||
|
|
"""
|
|||
|
|
)
|
|||
|
|
cur = conn.execute("SELECT COUNT(*) AS n FROM stratagems")
|
|||
|
|
if cur.fetchone()["n"] == 0:
|
|||
|
|
now = _now()
|
|||
|
|
for item in _load_seed():
|
|||
|
|
if not isinstance(item, dict):
|
|||
|
|
continue
|
|||
|
|
name = (item.get("name") or "").strip()
|
|||
|
|
if not name:
|
|||
|
|
continue
|
|||
|
|
alias = item.get("alias") or []
|
|||
|
|
if isinstance(alias, str):
|
|||
|
|
alias = [a.strip() for a in alias.split(",") if a.strip()]
|
|||
|
|
conn.execute(
|
|||
|
|
"""
|
|||
|
|
INSERT OR IGNORE INTO stratagems
|
|||
|
|
(name, name_en, type, backpack, icon, alias, is_custom, created_at, updated_at)
|
|||
|
|
VALUES (?,?,?,?,?,?,?,?,?)
|
|||
|
|
""",
|
|||
|
|
(
|
|||
|
|
name,
|
|||
|
|
item.get("name_en") or "",
|
|||
|
|
item.get("type") or "blue",
|
|||
|
|
1 if item.get("backpack") else 0,
|
|||
|
|
item.get("path") or item.get("icon") or "",
|
|||
|
|
json.dumps(alias, ensure_ascii=False),
|
|||
|
|
0,
|
|||
|
|
now,
|
|||
|
|
now,
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
conn.commit()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _row_to_item(row: sqlite3.Row) -> dict[str, Any]:
|
|||
|
|
try:
|
|||
|
|
alias = json.loads(row["alias"] or "[]")
|
|||
|
|
if not isinstance(alias, list):
|
|||
|
|
alias = []
|
|||
|
|
except Exception: # noqa: BLE001
|
|||
|
|
alias = []
|
|||
|
|
return {
|
|||
|
|
"id": row["id"],
|
|||
|
|
"name": row["name"],
|
|||
|
|
"name_en": row["name_en"] or "",
|
|||
|
|
"type": row["type"] or "blue",
|
|||
|
|
"backpack": bool(row["backpack"]),
|
|||
|
|
"path": row["icon"] or "",
|
|||
|
|
"icon": row["icon"] or "",
|
|||
|
|
"alias": alias,
|
|||
|
|
"is_custom": bool(row["is_custom"]),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _normalize_alias(value: Any) -> list[str]:
|
|||
|
|
if isinstance(value, str):
|
|||
|
|
parts = [v.strip() for v in value.replace("，", ",").replace("、", ",").replace("\n", ",").split(",") if v.strip()]
|
|||
|
|
elif isinstance(value, list):
|
|||
|
|
parts = [v.strip() for v in value if isinstance(v, str) and v.strip()]
|
|||
|
|
else:
|
|||
|
|
parts = []
|
|||
|
|
seen: set[str] = set()
|
|||
|
|
out: list[str] = []
|
|||
|
|
for p in parts:
|
|||
|
|
if p not in seen:
|
|||
|
|
seen.add(p)
|
|||
|
|
out.append(p)
|
|||
|
|
return out
|
|||
|
|
|
|||
|
|
|
|||
|
|
def list_stratagems() -> list[dict[str, Any]]:
|
|||
|
|
init_db()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
rows = conn.execute("SELECT * FROM stratagems ORDER BY id").fetchall()
|
|||
|
|
return [_row_to_item(r) for r in rows]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def count_stratagems() -> int:
|
|||
|
|
init_db()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
return int(conn.execute("SELECT COUNT(*) AS n FROM stratagems").fetchone()["n"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def list_stratagems_page(page: int = 1, page_size: int = 20) -> tuple[list[dict[str, Any]], int]:
|
|||
|
|
init_db()
|
|||
|
|
total = count_stratagems()
|
|||
|
|
offset = (page - 1) * page_size
|
|||
|
|
with _connect() as conn:
|
|||
|
|
rows = conn.execute(
|
|||
|
|
"SELECT * FROM stratagems ORDER BY id LIMIT ? OFFSET ?",
|
|||
|
|
(page_size, offset),
|
|||
|
|
).fetchall()
|
|||
|
|
return [_row_to_item(r) for r in rows], total
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_stratagem(seed: int) -> dict[str, Any] | None:
|
|||
|
|
init_db()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
row = conn.execute("SELECT * FROM stratagems WHERE id=?", (seed,)).fetchone()
|
|||
|
|
return _row_to_item(row) if row else None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def create_stratagem(item: dict[str, Any]) -> dict[str, Any]:
|
|||
|
|
init_db()
|
|||
|
|
name = (item.get("name") or "").strip()
|
|||
|
|
if not name:
|
|||
|
|
raise ValueError("name 不能为空")
|
|||
|
|
now = _now()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
cursor = conn.execute(
|
|||
|
|
"""
|
|||
|
|
INSERT INTO stratagems (name, name_en, type, backpack, icon, alias, is_custom, created_at, updated_at)
|
|||
|
|
VALUES (?,?,?,?,?,?,?,?,?)
|
|||
|
|
""",
|
|||
|
|
(
|
|||
|
|
name,
|
|||
|
|
item.get("name_en") or "",
|
|||
|
|
item.get("type") or "blue",
|
|||
|
|
1 if item.get("backpack") else 0,
|
|||
|
|
item.get("icon") or item.get("path") or "",
|
|||
|
|
json.dumps(_normalize_alias(item.get("alias")), ensure_ascii=False),
|
|||
|
|
1,
|
|||
|
|
now,
|
|||
|
|
now,
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
conn.commit()
|
|||
|
|
new_id = cursor.lastrowid
|
|||
|
|
row = get_stratagem(new_id)
|
|||
|
|
assert row is not None
|
|||
|
|
return row
|
|||
|
|
|
|||
|
|
|
|||
|
|
def update_stratagem(seed: int, fields: dict[str, Any]) -> dict[str, Any] | None:
|
|||
|
|
init_db()
|
|||
|
|
if not fields:
|
|||
|
|
return get_stratagem(seed)
|
|||
|
|
# 只允许安全字段
|
|||
|
|
allowed = {"name", "name_en", "type", "backpack", "icon", "path", "alias"}
|
|||
|
|
updates: dict[str, Any] = {}
|
|||
|
|
for k in allowed:
|
|||
|
|
if k in fields:
|
|||
|
|
updates[k] = fields[k]
|
|||
|
|
if "alias" in updates:
|
|||
|
|
updates["alias"] = json.dumps(_normalize_alias(updates["alias"]), ensure_ascii=False)
|
|||
|
|
if "path" in updates and "icon" not in updates:
|
|||
|
|
updates["icon"] = updates.pop("path")
|
|||
|
|
elif "icon" in updates:
|
|||
|
|
updates.pop("path", None)
|
|||
|
|
if not updates:
|
|||
|
|
return get_stratagem(seed)
|
|||
|
|
updates["updated_at"] = _now()
|
|||
|
|
sets = ", ".join(f"{k}=?" for k in updates)
|
|||
|
|
params = list(updates.values()) + [seed]
|
|||
|
|
with _connect() as conn:
|
|||
|
|
cur = conn.execute(f"UPDATE stratagems SET {sets} WHERE id=?", params)
|
|||
|
|
conn.commit()
|
|||
|
|
if cur.rowcount == 0:
|
|||
|
|
return None
|
|||
|
|
return get_stratagem(seed)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def delete_stratagem(seed: int) -> bool:
|
|||
|
|
init_db()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
cur = conn.execute("DELETE FROM stratagems WHERE id=?", (seed,))
|
|||
|
|
conn.commit()
|
|||
|
|
return cur.rowcount > 0
|
|||
|
|
|
|||
|
|
|
|||
|
|
def import_items(items: list[dict[str, Any]]) -> dict[str, int]:
|
|||
|
|
"""按 id(存在则更新) 或 name(存在则更新) 合并导入；新增的记录标记为 custom。
|
|||
|
|
|
|||
|
|
返回 {added, updated, skipped}
|
|||
|
|
"""
|
|||
|
|
init_db()
|
|||
|
|
added = 0
|
|||
|
|
updated = 0
|
|||
|
|
skipped = 0
|
|||
|
|
now = _now()
|
|||
|
|
with _connect() as conn:
|
|||
|
|
for item in items:
|
|||
|
|
if not isinstance(item, dict):
|
|||
|
|
skipped += 1
|
|||
|
|
continue
|
|||
|
|
name = (item.get("name") or "").strip()
|
|||
|
|
if not name:
|
|||
|
|
skipped += 1
|
|||
|
|
continue
|
|||
|
|
seed = item.get("id")
|
|||
|
|
existing: dict[str, Any] | None = None
|
|||
|
|
if isinstance(seed, int):
|
|||
|
|
row = conn.execute("SELECT * FROM stratagems WHERE id=?", (seed,)).fetchone()
|
|||
|
|
existing = _row_to_item(row) if row else None
|
|||
|
|
if existing is None:
|
|||
|
|
row = conn.execute("SELECT * FROM stratagems WHERE name=?", (name,)).fetchone()
|
|||
|
|
existing = _row_to_item(row) if row else None
|
|||
|
|
if existing is None:
|
|||
|
|
conn.execute(
|
|||
|
|
"""
|
|||
|
|
INSERT INTO stratagems (name, name_en, type, backpack, icon, alias, is_custom, created_at, updated_at)
|
|||
|
|
VALUES (?,?,?,?,?,?,?,?,?)
|
|||
|
|
""",
|
|||
|
|
(
|
|||
|
|
name,
|
|||
|
|
item.get("name_en") or "",
|
|||
|
|
item.get("type") or "blue",
|
|||
|
|
1 if item.get("backpack") else 0,
|
|||
|
|
item.get("icon") or item.get("path") or "",
|
|||
|
|
json.dumps(_normalize_alias(item.get("alias")), ensure_ascii=False),
|
|||
|
|
1,
|
|||
|
|
now,
|
|||
|
|
now,
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
added += 1
|
|||
|
|
else:
|
|||
|
|
fields = {
|
|||
|
|
"name": name,
|
|||
|
|
"name_en": (item.get("name_en") or existing.get("name_en") or ""),
|
|||
|
|
"type": (item.get("type") or existing.get("type") or "blue"),
|
|||
|
|
"backpack": bool(item.get("backpack", existing.get("backpack"))),
|
|||
|
|
"icon": (item.get("icon") or item.get("path") or existing.get("icon") or ""),
|
|||
|
|
"alias": item.get("alias", existing.get("alias")),
|
|||
|
|
}
|
|||
|
|
conn.execute(
|
|||
|
|
"""
|
|||
|
|
UPDATE stratagems
|
|||
|
|
SET name=?, name_en=?, type=?, backpack=?, icon=?, alias=?, updated_at=?
|
|||
|
|
WHERE id=?
|
|||
|
|
""",
|
|||
|
|
(
|
|||
|
|
fields["name"],
|
|||
|
|
fields["name_en"],
|
|||
|
|
fields["type"],
|
|||
|
|
1 if fields["backpack"] else 0,
|
|||
|
|
fields["icon"],
|
|||
|
|
json.dumps(_normalize_alias(fields["alias"]), ensure_ascii=False),
|
|||
|
|
now,
|
|||
|
|
existing["id"],
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
updated += 1
|
|||
|
|
conn.commit()
|
|||
|
|
return {"added": added, "updated": updated, "skipped": skipped}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def export_items() -> list[dict[str, Any]]:
|
|||
|
|
return list_stratagems()
|