57 lines
3.4 KiB
Python
57 lines
3.4 KiB
Python
from pathlib import Path
|
||
import ast
|
||
for root in (Path('C:/kefu/wechat_rpa'), Path('C:/wechat_rpa')):
|
||
p = root / 'wechat_bot.py'
|
||
text = p.read_text(encoding='utf-8').replace('_assistant_review_identity', '_review_assistant_identity')
|
||
start = text.index(' def _record_assistant_automated_send(self, state):')
|
||
end = text.index(' def _reply_review_decision(', start)
|
||
text = text[:start] + ''' def _record_assistant_automated_attempt(self, state, reply_text):
|
||
assistant = self._get_review_assistant(verify=True)
|
||
if assistant is None:
|
||
return True
|
||
try:
|
||
context = self._review_assistant_context(state or {})
|
||
if not context:
|
||
self._review_assistant_warn("自动发送前无法核对会话，暂缓本次发送")
|
||
return False
|
||
state["account"], state["conv_id"] = context["account"], context["conv_id"]
|
||
assistant.record_automated_attempt({**state, "reply_text": reply_text})
|
||
return True
|
||
except Exception as exc:
|
||
self._review_assistant_warn("自动发送留痕失败（" + type(exc).__name__ + "），暂缓本次发送")
|
||
return False
|
||
|
||
def _record_assistant_automated_send(self, state):
|
||
assistant = self._get_review_assistant()
|
||
if assistant is None or not isinstance(state, dict):
|
||
return
|
||
try:
|
||
# A verified visual receipt closes this local approval, but cannot
|
||
# grant blanket-mode resume or invent a server message identifier.
|
||
assistant.complete_local_review(state)
|
||
context = self._review_assistant_context(state)
|
||
if not context:
|
||
return
|
||
dispatched = float(state.get("send_dispatched_at") or 0)
|
||
reply = str(state.get("reply_text") or state.get("staged_reply_text") or "")
|
||
matches = [item for item in context.get("messages") or []
|
||
if item.get("account") == context["account"] and item.get("conv_id") == context["conv_id"]
|
||
and item.get("is_self") is True and str(item.get("sender_id") or "") == context["account"]
|
||
and str(item.get("content") or "").strip() == reply.strip()
|
||
and dispatched > 0 and float(item.get("send_time") or 0) >= dispatched - 1
|
||
and str(item.get("server_id") or "") not in {"", "0", "None"}]
|
||
if len(matches) == 1:
|
||
assistant.record_automated_send({**state, "account": context["account"],
|
||
"conv_id": context["conv_id"]}, {"status": "confirmed", "transport": "visual",
|
||
"accountId": context["account"], "conversationId": context["conv_id"],
|
||
"serverId": str(matches[0]["server_id"]), "confirmedAt": time.time()})
|
||
except Exception as exc:
|
||
self._review_assistant_warn("自动回复留痕暂不可用（" + type(exc).__name__ + "）")
|
||
|
||
''' + text[end:]
|
||
needle=' pending_state["send_state"] = "sending"\n pending_state["reply_text"] = reply_text'
|
||
assert text.count(needle)==1
|
||
text = text.replace(needle, ' if not self._record_assistant_automated_attempt(pending_state, reply_text):\n return self._deny_send("助理审核自动发送留痕未完成，本轮不发送")\n' + needle)
|
||
ast.parse(text)
|
||
p.write_text(text,encoding='utf-8',newline='\n')
|