From 0bdf5ac51ba2292bf939b7770603b3c8e105e427 Mon Sep 17 00:00:00 2001 From: shikong <919411476@qq.com> Date: Wed, 28 Aug 2024 04:00:12 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BE=AE=E4=BF=A1=20=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 3 ++- tasks/wechat/__init__.py | 10 ++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 4c6f1f9..993d16a 100644 --- a/main.py +++ b/main.py @@ -125,7 +125,8 @@ async def wechat_search_task(item: WechatSearchInput): tasks.wechat.click_search_btn(d) tasks.wechat.input_search_text(d, item.text) tasks.wechat.click_search_result(d) - tasks.wechat.input_chat_text(d, time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())) + tasks.wechat.input_chat_text(d, time.strftime("[AI] %Y-%m-%d %H:%M:%S", time.localtime())) + tasks.wechat.get_last_chat_text(d) if __name__ == '__main__': uvicorn.run('main:app', host='0.0.0.0', port=8000, reload=True) diff --git a/tasks/wechat/__init__.py b/tasks/wechat/__init__.py index ea3f353..28ea180 100644 --- a/tasks/wechat/__init__.py +++ b/tasks/wechat/__init__.py @@ -48,3 +48,13 @@ def input_chat_text(d: Device, text: str): selector = d.xpath('//*[@resource-id="com.tencent.mm:id/bql"]') selector.wait(timeout=1) selector.click() + + +def get_last_chat_text(d: Device): + selector = (d.xpath('//*[@resource-id="com.tencent.mm:id/bn1"]') + .child('//*[@resource-id="com.tencent.mm:id/bkl"]')) + elements = selector.all() + + if len(elements) > 0: + last_element = elements[-1] + print(last_element.text)