232 lines
8.6 KiB
Python
232 lines
8.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
inkos 终极修复方案
|
||
直接修复数据库中的数据错误
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import shutil
|
||
import sqlite3
|
||
import subprocess
|
||
import time
|
||
|
||
def stop_inkos():
|
||
"""停止inkos所有进程"""
|
||
print("1. 停止inkos所有进程...")
|
||
subprocess.run(["pkill", "-9", "-f", "node.*inkos"], capture_output=True)
|
||
subprocess.run(["pkill", "-9", "-f", "inkos"], capture_output=True)
|
||
time.sleep(3)
|
||
print(" ✅ inkos已停止")
|
||
|
||
def backup_data():
|
||
"""备份所有数据"""
|
||
print("2. 备份数据...")
|
||
backup_dir = f"/root/.openclaw/workspace/backups/inkos_ultimate_fix_{time.strftime('%Y%m%d_%H%M%S')}"
|
||
os.makedirs(backup_dir, exist_ok=True)
|
||
|
||
# 备份重要文件和目录
|
||
backup_items = [
|
||
"/root/.openclaw/workspace/tomato-novel/inkos.json",
|
||
"/root/.openclaw/workspace/tomato-novel/tomato-novel/books/末日重生-开局囤货十亿物资",
|
||
"/root/.openclaw/workspace/tomato-novel/books/末日重生-开局囤货十亿物资"
|
||
]
|
||
|
||
for item in backup_items:
|
||
if os.path.exists(item):
|
||
if os.path.isfile(item):
|
||
shutil.copy2(item, backup_dir)
|
||
else:
|
||
shutil.copytree(item, os.path.join(backup_dir, os.path.basename(item)), dirs_exist_ok=True)
|
||
print(f" 已备份: {os.path.basename(item)}")
|
||
|
||
print(f" 备份位置: {backup_dir}")
|
||
return backup_dir
|
||
|
||
def fix_database():
|
||
"""修复数据库文件"""
|
||
print("3. 修复数据库文件...")
|
||
|
||
# 找到并修复所有数据库文件
|
||
db_files = [
|
||
"/root/.openclaw/workspace/tomato-novel/tomato-novel/books/末日重生-开局囤货十亿物资/story/memory.db",
|
||
"/root/.openclaw/workspace/tomato-novel/books/末日重生-开局囤货十亿物资/story/memory.db"
|
||
]
|
||
|
||
for db_file in db_files:
|
||
if os.path.exists(db_file):
|
||
print(f" 检查数据库: {db_file}")
|
||
try:
|
||
# 备份数据库
|
||
shutil.copy2(db_file, f"{db_file}.backup")
|
||
|
||
# 尝试连接到数据库
|
||
conn = sqlite3.connect(db_file)
|
||
cursor = conn.cursor()
|
||
|
||
# 获取所有表名
|
||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
|
||
tables = cursor.fetchall()
|
||
print(f" 数据库表: {[t[0] for t in tables]}")
|
||
|
||
# 查找可能包含章节数据的表
|
||
for table in tables:
|
||
table_name = table[0]
|
||
if 'chapter' in table_name.lower() or 'row' in table_name.lower():
|
||
print(f" 检查表: {table_name}")
|
||
|
||
# 获取表结构
|
||
cursor.execute(f"PRAGMA table_info({table_name});")
|
||
columns = cursor.fetchall()
|
||
print(f" 列: {[col[1] for col in columns]}")
|
||
|
||
# 查找title列
|
||
title_col = None
|
||
for col in columns:
|
||
if 'title' in col[1].lower():
|
||
title_col = col[1]
|
||
break
|
||
|
||
if title_col:
|
||
# 查找空标题的行
|
||
cursor.execute(f"SELECT rowid, {title_col} FROM {table_name} WHERE {title_col} IS NULL OR {title_col} = '' OR LENGTH(TRIM({title_col})) = 0;")
|
||
empty_titles = cursor.fetchall()
|
||
|
||
if empty_titles:
|
||
print(f" 找到 {len(empty_titles)} 个空标题")
|
||
for rowid, title in empty_titles:
|
||
# 修复空标题
|
||
new_title = f"第{rowid}章"
|
||
cursor.execute(f"UPDATE {table_name} SET {title_col} = ? WHERE rowid = ?", (new_title, rowid))
|
||
print(f" 修复行 {rowid}: '{new_title}'")
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
print(f" ✅ 数据库修复完成: {os.path.basename(db_file)}")
|
||
|
||
except Exception as e:
|
||
print(f" ❌ 修复数据库时出错: {e}")
|
||
# 如果数据库损坏严重，删除它让inkos重新创建
|
||
print(f" 删除损坏的数据库: {db_file}")
|
||
os.remove(db_file)
|
||
print(f" ✅ 已删除，inkos将重新创建数据库")
|
||
|
||
def clean_state_files():
|
||
"""清理状态文件，让inkos重新开始"""
|
||
print("4. 清理状态文件...")
|
||
|
||
# 删除可能损坏的状态文件
|
||
state_dirs = [
|
||
"/root/.openclaw/workspace/tomato-novel/tomato-novel/books/末日重生-开局囤货十亿物资/story/state",
|
||
"/root/.openclaw/workspace/tomato-novel/tomato-novel/books/末日重生-开局囤货十亿物资/story/runtime",
|
||
"/root/.openclaw/workspace/tomato-novel/books/末日重生-开局囤货十亿物资/story/state",
|
||
"/root/.openclaw/workspace/tomato-novel/books/末日重生-开局囤货十亿物资/story/runtime"
|
||
]
|
||
|
||
for state_dir in state_dirs:
|
||
if os.path.exists(state_dir):
|
||
# 只删除特定文件，保留章节文件
|
||
for file in os.listdir(state_dir):
|
||
if file.endswith('.json') or file.endswith('.db'):
|
||
file_path = os.path.join(state_dir, file)
|
||
try:
|
||
os.remove(file_path)
|
||
print(f" 删除: {file}")
|
||
except:
|
||
pass
|
||
|
||
print(" ✅ 状态文件已清理")
|
||
|
||
def create_simple_config():
|
||
"""创建最简单的配置"""
|
||
print("5. 创建简单配置...")
|
||
|
||
config = {
|
||
"name": "tomato-novel-fixed",
|
||
"version": "0.1.0",
|
||
"language": "zh",
|
||
"llm": {
|
||
"provider": "custom",
|
||
"baseUrl": "https://ark.cn-beijing.volces.com/api/coding/v3",
|
||
"model": "deepseek-v3.2"
|
||
},
|
||
"daemon": {
|
||
"schedule": {
|
||
"writeCron": "0 */2 * * *" # 每2小时一次
|
||
},
|
||
"maxConcurrentBooks": 1
|
||
},
|
||
"quality": {
|
||
"enforcement": "none" # 完全关闭质量检查
|
||
}
|
||
}
|
||
|
||
config_path = "/root/.openclaw/workspace/tomato-novel/inkos.json"
|
||
with open(config_path, 'w', encoding='utf-8') as f:
|
||
json.dump(config, f, ensure_ascii=False, indent=2)
|
||
|
||
print(" ✅ 简单配置已创建")
|
||
|
||
def start_inkos():
|
||
"""启动inkos"""
|
||
print("6. 启动inkos...")
|
||
|
||
os.chdir("/root/.openclaw/workspace/tomato-novel")
|
||
|
||
# 启动inkos
|
||
with open("/tmp/inkos_ultimate_start.log", "w") as log_file:
|
||
process = subprocess.Popen(
|
||
["nohup", "inkos", "up"],
|
||
stdout=log_file,
|
||
stderr=subprocess.STDOUT,
|
||
start_new_session=True
|
||
)
|
||
|
||
time.sleep(10)
|
||
|
||
# 检查是否启动成功
|
||
result = subprocess.run(["ps", "aux"], capture_output=True, text=True)
|
||
if "inkos up" in result.stdout:
|
||
pid_line = [line for line in result.stdout.split('\n') if "inkos up" in line][0]
|
||
pid = pid_line.split()[1]
|
||
print(f" ✅ inkos 已启动 (PID: {pid})")
|
||
return True
|
||
else:
|
||
print(" ❌ inkos 启动失败")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("=" * 60)
|
||
print("inkos 终极修复方案")
|
||
print("=" * 60)
|
||
|
||
stop_inkos()
|
||
backup_dir = backup_data()
|
||
fix_database()
|
||
clean_state_files()
|
||
create_simple_config()
|
||
|
||
if start_inkos():
|
||
print("\n" + "=" * 60)
|
||
print("✅ 修复完成!")
|
||
print(f"备份位置: {backup_dir}")
|
||
print("=" * 60)
|
||
print("\ninkos 已重新启动，配置为:")
|
||
print("- 质量检查: 关闭")
|
||
print("- 创作频率: 每2小时")
|
||
print("- 日志文件: /tmp/inkos_ultimate_start.log")
|
||
print("\n等待5-10分钟后，使用以下命令测试:")
|
||
print(" inkos status")
|
||
print(" inkos draft 末日重生-开局囤货十亿物资")
|
||
else:
|
||
print("\n" + "=" * 60)
|
||
print("❌ 修复失败!")
|
||
print("=" * 60)
|
||
print("\n请检查:")
|
||
print("1. inkos 是否安装: which inkos")
|
||
print("2. 日志文件: tail -100 /tmp/inkos_ultimate_start.log")
|
||
print("3. 可能需要重新安装 inkos")
|
||
|
||
if __name__ == "__main__":
|
||
main() |