65 lines
5.4 KiB
Python
65 lines
5.4 KiB
Python
from pathlib import Path
|
|
import shutil
|
|
|
|
root = Path('C:/wechat_rpa')
|
|
target = root / 'protocol_engine.py'
|
|
text = target.read_text(encoding='utf-8')
|
|
|
|
def replace(old, new):
|
|
global text
|
|
assert text.count(old) == 1, (old[:100], text.count(old))
|
|
text = text.replace(old, new)
|
|
|
|
replace("def _unanswered_manual_reason(context):", """MANUAL_MEDIA_REASON = '当前未回复消息包含图片、语音或其他非文本内容，协议版需人工处理'
|
|
MANUAL_BATCH_REASON = '连续未回复消息超过 100 条，无法保证问题完整，请人工处理后继续'
|
|
MANUAL_NON_TEXT_REASON = '当前协议版只自动处理文本消息'
|
|
|
|
|
|
def _restore_review_details(state):
|
|
# Existing persisted queues predate the desktop's explicit review fields.
|
|
# Recover presentation metadata without granting approval or replaying sends.
|
|
if state.get('awaiting_review') and not state.get('approved'):
|
|
state['review_reason'] = str(state.get('review_reason') or state.get('stage_detail') or '人工审核后点击“通过并发送”')
|
|
if state.get('stage') == 'error' and state.get('last_error') in (
|
|
MANUAL_MEDIA_REASON, MANUAL_BATCH_REASON, MANUAL_NON_TEXT_REASON):
|
|
state.update(manual_required=True, manual_reason=state['last_error'])
|
|
|
|
|
|
def _unanswered_manual_reason(context):""")
|
|
replace("if not _is_text_message(message):return '当前未回复消息包含图片、语音或其他非文本内容，协议版需人工处理'", "if not _is_text_message(message):return MANUAL_MEDIA_REASON")
|
|
replace("if count>100:return '连续未回复消息超过 100 条，无法保证问题完整，请人工处理后继续'", "if count>100:return MANUAL_BATCH_REASON")
|
|
replace(" for key,state in list(self._pending.items()):\n lookup", " for key,state in list(self._pending.items()):\n _restore_review_details(state)\n lookup")
|
|
replace(" if not _is_text_message(state):manual_reason=manual_reason or '当前协议版只自动处理文本消息'\n if manual_reason:\n with self._lock:state.update(reply_text='',staged_reply_text='',approved=False,awaiting_review=False)\n self._stage(key,'error','当前消息需要人工处理',manual_reason);return", """ if not _is_text_message(state):manual_reason=manual_reason or MANUAL_NON_TEXT_REASON
|
|
if manual_reason:
|
|
lines=[]
|
|
for message in reversed((context or {}).get('messages') or []):
|
|
if message.get('is_self') or system_contact_status(message)=='restored':break
|
|
lines.append(str(message.get('content') or '（非文本消息）'))
|
|
if len(lines)>=101:break
|
|
with self._lock:
|
|
state.update(reply_text='',staged_reply_text='',approved=False,awaiting_review=False,
|
|
review_reason='',manual_required=True,manual_reason=manual_reason,
|
|
chat_text=str((context or {}).get('text') or ''),last_lines=list(reversed(lines)))
|
|
self._stage(key,'error','当前消息需要人工处理',manual_reason);return""")
|
|
replace(" with self._lock:state['awaiting_review']=True\n self._stage(key,'awaiting_review',reason)", " with self._lock:state.update(awaiting_review=True,review_reason=reason,manual_required=False,manual_reason='')\n self._stage(key,'awaiting_review',reason)")
|
|
replace(" if s is None:result['missing'].append(key);continue\n if s.get('send_state')", " if s is None:result['missing'].append(key);continue\n if s.get('account')!=self.account:result['protected'].append(key);continue\n if s.get('send_state')")
|
|
replace(" if s.get('awaiting_review') and s.get('reply_text'):s.update(approved=True,awaiting_review=False);result['approved'].append(key)", """ if s.get('awaiting_review') and s.get('reply_text') and not s.get('manual_required'):
|
|
now=time.time()
|
|
s.update(approved=True,awaiting_review=False,review_reason='',stage='ready_to_send',
|
|
stage_detail='审核已通过，等待发送前核对',stage_started_at=now,updated_at=now)
|
|
result['approved'].append(key)""")
|
|
replace(" if s.get('stage')=='error':s.update(stage='queued',last_error='',ready_at=time.time(),reply_text='',approved=False);result['retried'].append(key)", """ if s.get('stage')=='error':
|
|
now=time.time()
|
|
s.update(stage='queued',stage_detail='等待重新处理',stage_started_at=now,updated_at=now,
|
|
last_error='',ready_at=now,reply_text='',staged_reply_text='',approved=False,
|
|
awaiting_review=False,review_reason='',manual_required=False,manual_reason='')
|
|
result['retried'].append(key)""")
|
|
compile(text, str(target), 'exec')
|
|
backup = root / 'backups/review-visibility-20260916'
|
|
backup.mkdir(parents=True, exist_ok=True)
|
|
assert not (backup / target.name).exists(), 'Backup already exists; do not rerun'
|
|
shutil.copy2(target, backup / target.name)
|
|
target.write_text(text, encoding='utf-8')
|
|
shutil.copy2(Path(__file__).with_name('test_protocol_review_visibility.py'), root / 'test_protocol_review_visibility.py')
|
|
print('Updated protocol review metadata and isolated tests; backup:', backup)
|