feat: 优化部署脚本
This commit is contained in:
丹尼尔
179
backend/db.py
Normal file
179
backend/db.py
Normal file
@@ -0,0 +1,179 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""SQLite 数据库：表结构初始化与连接，数据目录由 DATA_DIR 决定（可挂载到宿主机）。"""
|
||||
import json
|
||||
import os
|
||||
import sqlite3
|
||||
|
||||
_DATA_DIR = os.getenv("DATA_DIR") or os.path.join(os.path.dirname(__file__), "data")
|
||||
_DB_PATH = os.path.join(_DATA_DIR, "wechat.db")
|
||||
|
||||
def get_db_path() -> str:
|
||||
return _DB_PATH
|
||||
|
||||
def get_conn() -> sqlite3.Connection:
|
||||
os.makedirs(_DATA_DIR, exist_ok=True)
|
||||
conn = sqlite3.connect(_DB_PATH, check_same_thread=False)
|
||||
conn.row_factory = sqlite3.Row
|
||||
return conn
|
||||
|
||||
def init_schema(conn: sqlite3.Connection) -> None:
|
||||
cur = conn.cursor()
|
||||
# 客户档案
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS customers (
|
||||
id TEXT PRIMARY KEY,
|
||||
key TEXT NOT NULL,
|
||||
wxid TEXT NOT NULL,
|
||||
remark_name TEXT,
|
||||
region TEXT,
|
||||
age TEXT,
|
||||
gender TEXT,
|
||||
level TEXT,
|
||||
tags TEXT
|
||||
)
|
||||
""")
|
||||
cur.execute("CREATE INDEX IF NOT EXISTS idx_customers_key ON customers(key)")
|
||||
# 定时问候任务
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS greeting_tasks (
|
||||
id TEXT PRIMARY KEY,
|
||||
key TEXT NOT NULL,
|
||||
name TEXT,
|
||||
send_time TEXT,
|
||||
customer_tags TEXT,
|
||||
template TEXT,
|
||||
use_qwen INTEGER DEFAULT 0,
|
||||
enabled INTEGER DEFAULT 1,
|
||||
executed_at TEXT
|
||||
)
|
||||
""")
|
||||
cur.execute("CREATE INDEX IF NOT EXISTS idx_greeting_tasks_key ON greeting_tasks(key)")
|
||||
# 商品标签
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS product_tags (
|
||||
id TEXT PRIMARY KEY,
|
||||
key TEXT NOT NULL,
|
||||
name TEXT
|
||||
)
|
||||
""")
|
||||
# 推送群组
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS push_groups (
|
||||
id TEXT PRIMARY KEY,
|
||||
key TEXT NOT NULL,
|
||||
name TEXT,
|
||||
customer_ids TEXT,
|
||||
tag_ids TEXT
|
||||
)
|
||||
""")
|
||||
# 推送任务
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS push_tasks (
|
||||
id TEXT PRIMARY KEY,
|
||||
key TEXT NOT NULL,
|
||||
product_tag_id TEXT,
|
||||
group_id TEXT,
|
||||
content TEXT,
|
||||
send_at TEXT,
|
||||
status TEXT,
|
||||
created_at TEXT
|
||||
)
|
||||
""")
|
||||
# 同步消息（WS 拉取 + 发出记录）
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS sync_messages (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
key TEXT NOT NULL,
|
||||
create_time INTEGER DEFAULT 0,
|
||||
payload TEXT
|
||||
)
|
||||
""")
|
||||
cur.execute("CREATE INDEX IF NOT EXISTS idx_sync_messages_key ON sync_messages(key)")
|
||||
# 模型配置
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS models (
|
||||
id TEXT PRIMARY KEY,
|
||||
name TEXT,
|
||||
provider TEXT,
|
||||
api_key TEXT,
|
||||
base_url TEXT,
|
||||
model_name TEXT,
|
||||
is_current INTEGER DEFAULT 0
|
||||
)
|
||||
""")
|
||||
# AI 回复配置（白名单 / 超级管理员）
|
||||
cur.execute("""
|
||||
CREATE TABLE IF NOT EXISTS ai_reply_config (
|
||||
key TEXT PRIMARY KEY,
|
||||
super_admin_wxids TEXT,
|
||||
whitelist_wxids TEXT
|
||||
)
|
||||
""")
|
||||
conn.commit()
|
||||
_migrate_json_if_needed(conn)
|
||||
|
||||
|
||||
def _migrate_json_if_needed(conn: sqlite3.Connection) -> None:
|
||||
"""若表为空且存在同名 JSON 文件，则从 JSON 迁移一次。"""
|
||||
cur = conn.cursor()
|
||||
# (table, filename, columns, json_columns)
|
||||
tables_files = [
|
||||
("customers", "customers.json", ["id", "key", "wxid", "remark_name", "region", "age", "gender", "level", "tags"], ["tags"]),
|
||||
("greeting_tasks", "greeting_tasks.json", ["id", "key", "name", "send_time", "customer_tags", "template", "use_qwen", "enabled", "executed_at"], ["customer_tags"]),
|
||||
("product_tags", "product_tags.json", ["id", "key", "name"], []),
|
||||
("push_groups", "push_groups.json", ["id", "key", "name", "customer_ids", "tag_ids"], ["customer_ids", "tag_ids"]),
|
||||
("push_tasks", "push_tasks.json", ["id", "key", "product_tag_id", "group_id", "content", "send_at", "status", "created_at"], []),
|
||||
("models", "models.json", ["id", "name", "provider", "api_key", "base_url", "model_name", "is_current"], []),
|
||||
("ai_reply_config", "ai_reply_config.json", ["key", "super_admin_wxids", "whitelist_wxids"], ["super_admin_wxids", "whitelist_wxids"]),
|
||||
]
|
||||
for table, filename, columns, json_cols in tables_files:
|
||||
json_cols_set = set(json_cols)
|
||||
cur.execute(f"SELECT COUNT(*) FROM {table}")
|
||||
if cur.fetchone()[0] > 0:
|
||||
continue
|
||||
path = os.path.join(_DATA_DIR, filename)
|
||||
if not os.path.isfile(path):
|
||||
continue
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
rows = json.load(f)
|
||||
except Exception:
|
||||
continue
|
||||
if not rows:
|
||||
continue
|
||||
for r in rows:
|
||||
if not isinstance(r, dict):
|
||||
continue
|
||||
vals = []
|
||||
for c in columns:
|
||||
v = r.get(c)
|
||||
if c in json_cols_set and isinstance(v, (list, dict)):
|
||||
v = json.dumps(v, ensure_ascii=False)
|
||||
elif isinstance(v, bool):
|
||||
v = 1 if v else 0
|
||||
vals.append(v)
|
||||
placeholders = ",".join("?" * len(columns))
|
||||
cur.execute(f"INSERT OR IGNORE INTO {table} ({','.join(columns)}) VALUES ({placeholders})", vals)
|
||||
# sync_messages: 按 key + payload 迁移
|
||||
cur.execute("SELECT COUNT(*) FROM sync_messages")
|
||||
if cur.fetchone()[0] == 0:
|
||||
path = os.path.join(_DATA_DIR, "sync_messages.json")
|
||||
if os.path.isfile(path):
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
rows = json.load(f)
|
||||
for r in rows:
|
||||
if not isinstance(r, dict):
|
||||
continue
|
||||
key = r.get("key", "")
|
||||
ct = int(r.get("CreateTime") or 0) if isinstance(r.get("CreateTime"), (int, float)) else 0
|
||||
cur.execute("INSERT INTO sync_messages (key, create_time, payload) VALUES (?,?,?)", (key, ct, json.dumps(r, ensure_ascii=False)))
|
||||
except Exception:
|
||||
pass
|
||||
conn.commit()
|
||||
|
||||
|
||||
def _conn():
|
||||
c = get_conn()
|
||||
init_schema(c)
|
||||
return c
|
||||
Reference in New Issue
Block a user