61 lines
3.1 KiB
Python
61 lines
3.1 KiB
Python
from pathlib import Path
|
|
import hashlib
|
|
import json
|
|
import shutil
|
|
|
|
root = Path('C:/wechat_rpa')
|
|
audit = Path(__file__).resolve().parent
|
|
target = root / 'protocol_engine.py'
|
|
before = target.read_bytes()
|
|
text = before.decode('utf-8')
|
|
old = "def session_key(account,conversation):return hashlib.blake2b((account+'\\0'+conversation).encode(),digest_size=40).hexdigest()"
|
|
new = '''def _is_text_message(message):
|
|
try:return int(message.get('content_type') or 0) in (0,1,2)
|
|
except (TypeError,ValueError):return False
|
|
|
|
def _unanswered_requires_manual(context):
|
|
# A text caption after an image/voice is still a media question. Only inspect
|
|
# the unanswered batch: media already handled by a human must not block later text.
|
|
for message in reversed((context or {}).get('messages') or []):
|
|
if message.get('is_self'):break
|
|
if not _is_text_message(message):return True
|
|
return False
|
|
|
|
''' + old
|
|
assert text.count(old) == 1
|
|
text = text.replace(old, new)
|
|
old = " self._save()\n def _save(self):"
|
|
new = """ # A restart must preserve the same pause as an unknown live send result.
|
|
self.needs_attention=any(s.get('account')==self.account and s.get('send_state')=='uncertain' for s in self._pending.values())
|
|
self._save()
|
|
def _save(self):"""
|
|
assert text.count(old) == 1
|
|
text = text.replace(old,new)
|
|
old = """ if int(state.get('content_type') or 0) not in (0,1,2):
|
|
self._stage(key,'error','此消息类型需要人工处理','当前协议版只自动处理文本消息');return
|
|
if not state.get('reply_text'):
|
|
self._stage(key,'generating','根据当前会话生成回复')
|
|
context=self.db.get_conversation_context_by_id(self.account,state['conv_id'],limit=100)
|
|
"""
|
|
new = """ context=self.db.get_conversation_context_by_id(self.account,state['conv_id'],limit=100)
|
|
if not _is_text_message(state) or _unanswered_requires_manual(context):
|
|
with self._lock:state.update(reply_text='',staged_reply_text='',approved=False,awaiting_review=False)
|
|
self._stage(key,'error','此消息类型需要人工处理','当前未回复消息包含图片、语音或其他非文本内容，协议版需人工处理');return
|
|
if not state.get('reply_text'):
|
|
self._stage(key,'generating','根据当前会话生成回复')
|
|
"""
|
|
assert text.count(old) == 1
|
|
text = text.replace(old,new)
|
|
backup = root / 'backups/recognition-audit-20260916'
|
|
backup.mkdir(parents=True,exist_ok=True)
|
|
save = backup / target.name
|
|
assert not save.exists(), f'Backup already exists: {save}'
|
|
save.write_bytes(before)
|
|
target.write_text(text,encoding='utf-8')
|
|
shutil.copy2(audit/'test_protocol_audit_boundaries.py', root/'test_protocol_audit_boundaries.py')
|
|
result = {'target':str(target),'backup':str(save), 'before_sha256':hashlib.sha256(before).hexdigest(),
|
|
'after_sha256':hashlib.sha256(target.read_bytes()).hexdigest(),
|
|
'added_test':str(root/'test_protocol_audit_boundaries.py')}
|
|
(audit/'protocol-fix-manifest.json').write_text(json.dumps(result,indent=2),encoding='utf-8')
|
|
print(json.dumps(result))
|