50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
|
"""
|
||
|
|
强制重置问卷数据库
|
||
|
|
"""
|
||
|
|
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完成！刷新问卷页面。")
|