"""
强制重置问卷数据库
"""
import paramiko, time

NAS_HOST = "100.85.152.112"
NAS_PORT = 22
NAS_USER = "simon"
NAS_PASS = "Nas82829330"
NAS_DEPLOY_PATH = "/volume1/docker/gaokao-portal"
DOCKER = f"echo '{NAS_PASS}' | sudo -S /usr/local/bin/docker"

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect(NAS_HOST, NAS_PORT, NAS_USER, NAS_PASS, timeout=20)

def run(cmd, desc=""):
    if desc:
        print(f"[{desc}]")
    stdin, stdout, stderr = c.exec_command(cmd, timeout=30)
    time.sleep(1)
    ch = stdout.channel
    out = b""
    while not ch.exit_status_ready():
        if ch.recv_ready():
            out += ch.recv(4096)
        time.sleep(0.2)
    while ch.recv_ready():
        out += ch.recv(4096)
    return out.decode().strip()

db_path = f"{NAS_DEPLOY_PATH}/backend/data/gaokao.db"

# 1. 删除旧数据库
run(f"rm -f {db_path}", "删除旧数据库")

# 2. 重启后端容器（会触发 seed）
run(f"cd {NAS_DEPLOY_PATH} && {DOCKER} compose restart backend", "重启后端")
print("等待容器启动...")
time.sleep(10)

# 3. 检查新问卷
result = run(f"sqlite3 {db_path} 'SELECT id, title FROM questionnaires;'", "问卷列表")
print(f"\n新问卷:\n{result}")

# 4. 验证API
run("curl -s http://localhost:8080/api/questionnaires | head -c 800", "问卷API")

c.close()
print("\n完成！刷新问卷页面。")