mirror of
https://mirror.skon.top/github.com/cft0808/edict
synced 2026-04-21 05:10:27 +08:00
c436aadc1a
**问题一：无效备份不断堆积** 当 pending_model_changes.json 中的变更与 openclaw.json 现有值相同时 （即 model 已经是目标值），脚本仍会创建备份文件并重写配置， 导致每 15 秒产生一个内容完全相同的 .bak.model-* 文件。 修复：先比较新旧配置的 JSON 内容，只有真正发生变化时才备份和写入。 **问题二：log_data.extend() AttributeError 导致 pending 永不清空** CHANGE_LOG 文件损坏或内容为 dict 时，rj() 返回 dict 而非 list， 调用 .extend() 抛出 AttributeError，使函数在第 75 行中断。 后续的 atomic_json_write(PENDING, []) 永远不会执行， pending_model_changes.json 始终非空，造成每次循环都重复处理。 修复：在 extend 前加 isinstance 检查，非 list 时重置为空列表。 Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com> Co-authored-by: cft0808 <41196455+cft0808@users.noreply.github.com>
121 lines
4.3 KiB
Python
121 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
||
"""应用 data/pending_model_changes.json → openclaw.json，并重启 Gateway"""
|
||
import json, pathlib, subprocess, datetime, shutil, logging, glob
|
||
from file_lock import atomic_json_write, atomic_json_read
|
||
|
||
log = logging.getLogger('model_change')
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s [%(name)s] %(message)s', datefmt='%H:%M:%S')
|
||
|
||
BASE = pathlib.Path(__file__).parent.parent
|
||
DATA = BASE / 'data'
|
||
OPENCLAW_CFG = pathlib.Path.home() / '.openclaw' / 'openclaw.json'
|
||
PENDING = DATA / 'pending_model_changes.json'
|
||
CHANGE_LOG = DATA / 'model_change_log.json'
|
||
MAX_BACKUPS = 10
|
||
|
||
|
||
def rj(path, default):
|
||
try:
|
||
return json.loads(path.read_text())
|
||
except Exception:
|
||
return default
|
||
|
||
|
||
def cleanup_backups():
|
||
"""只保留最近 MAX_BACKUPS 个备份"""
|
||
pattern = str(OPENCLAW_CFG.parent / 'openclaw.json.bak.model-*')
|
||
baks = sorted(glob.glob(pattern))
|
||
for old in baks[:-MAX_BACKUPS]:
|
||
try:
|
||
pathlib.Path(old).unlink()
|
||
except OSError:
|
||
pass
|
||
|
||
|
||
def main():
|
||
if not PENDING.exists():
|
||
return
|
||
pending = rj(PENDING, [])
|
||
if not pending:
|
||
return
|
||
|
||
cfg = rj(OPENCLAW_CFG, {})
|
||
agents_list = cfg.get('agents', {}).get('list', [])
|
||
default_model = cfg.get('agents', {}).get('defaults', {}).get('model', {}).get('primary', '')
|
||
|
||
applied, errors = [], []
|
||
for change in pending:
|
||
ag_id = change.get('agentId', '').strip()
|
||
new_model = change.get('model', '').strip()
|
||
if not ag_id or not new_model:
|
||
errors.append({'change': change, 'error': 'missing fields'})
|
||
continue
|
||
found = False
|
||
for ag in agents_list:
|
||
if ag.get('id') == ag_id:
|
||
old = ag.get('model', default_model)
|
||
if new_model == default_model:
|
||
ag.pop('model', None)
|
||
else:
|
||
ag['model'] = new_model
|
||
applied.append({'at': datetime.datetime.now().isoformat(), 'agentId': ag_id, 'oldModel': old, 'newModel': new_model})
|
||
found = True
|
||
break
|
||
if not found:
|
||
errors.append({'change': change, 'error': f'agent {ag_id} not found'})
|
||
|
||
if applied:
|
||
# 只有内容真正变化时才备份和写入
|
||
new_cfg = dict(cfg)
|
||
new_cfg['agents'] = dict(cfg.get('agents', {}))
|
||
new_cfg['agents']['list'] = agents_list
|
||
old_text = json.dumps(cfg, ensure_ascii=False, sort_keys=True)
|
||
new_text = json.dumps(new_cfg, ensure_ascii=False, sort_keys=True)
|
||
if old_text != new_text:
|
||
bak = OPENCLAW_CFG.parent / f'openclaw.json.bak.model-{datetime.datetime.now().strftime("%Y%m%d-%H%M%S")}'
|
||
shutil.copy2(OPENCLAW_CFG, bak)
|
||
cleanup_backups()
|
||
atomic_json_write(OPENCLAW_CFG, new_cfg)
|
||
cfg = new_cfg
|
||
|
||
log_data = rj(CHANGE_LOG, [])
|
||
if not isinstance(log_data, list):
|
||
log_data = []
|
||
log_data.extend(applied)
|
||
if len(log_data) > 200:
|
||
log_data = log_data[-200:]
|
||
atomic_json_write(CHANGE_LOG, log_data)
|
||
|
||
for e in applied:
|
||
log.info(f'{e["agentId"]}: {e["oldModel"]} → {e["newModel"]}')
|
||
|
||
restart_ok = False
|
||
rollback = False
|
||
try:
|
||
r = subprocess.run(['openclaw', 'gateway', 'restart'], capture_output=True, text=True, timeout=30)
|
||
restart_ok = r.returncode == 0
|
||
log.info(f'gateway restart rc={r.returncode}')
|
||
except Exception as e:
|
||
log.error(f'gateway restart failed: {e}')
|
||
# 回滚配置
|
||
if bak.exists():
|
||
shutil.copy2(bak, OPENCLAW_CFG)
|
||
log.warning('rolled back openclaw.json from backup')
|
||
rollback = True
|
||
for a in applied:
|
||
a['rolledBack'] = True
|
||
|
||
atomic_json_write(PENDING, [])
|
||
atomic_json_write(DATA / 'last_model_change_result.json', {
|
||
'at': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
|
||
'applied': applied, 'errors': errors,
|
||
'gatewayRestarted': restart_ok, 'rolledBack': rollback,
|
||
})
|
||
elif errors:
|
||
log.warning(f'{len(errors)} changes failed, 0 applied')
|
||
atomic_json_write(PENDING, [])
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|