pyxllib 0.3.96__py3-none-any.whl → 0.3.197__py3-none-any.whl

pyxllib/autogui/wechat_msg.py

@@ -0,0 +1,421 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2024/11/05
+
+"""
+微信 消息框中不同消息结构的解析器 工具
+
+todo 这个文件很多值可以写成枚举类型更规范？
+"""
+
+import re
+import json
+
+# 定义全局字典
+msg_parsers = {}
+
+
+# 定义装饰器
+def register_tool(key):
+    def decorator(func):
+        # 将函数添加到全局字典中，键为传入的key参数
+        msg_parsers[key] = func
+        return func
+
+    return decorator
+
+
+def fmt_text(text):
+    return text.replace('\n', ' ')
+
+
+def parse_time_string(time_str):
+    """
+    解析不同格式的时间字符串为标准的 Python datetime 类型。
+
+    支持以下格式：
+    1. 完整的日期和时间，如 "2024年10月31日 22:03"，解析为对应的 datetime 对象。
+    2. 仅包含时间的字符串，如 "19:18"，会被解析为当天的时间点，日期部分设为当前日期。
+    3. 特殊字符串 "以下为新消息"，解析为当前时间的 datetime 对象。
+
+    示例输入：
+    - "2024年10月31日 22:03"
+    - "2024年11月1日 20:09"
+    - "2024年11月1日 21:08"
+    - "19:18"
+    - "20:16"
+    - "以下为新消息"
+    """
+    from datetime import datetime
+
+    # 定义当前日期
+    now = datetime.now()
+
+    # 定义不同的正则表达式模式
+    full_date_pattern = r"(\d{4})年(\d{1,2})月(\d{1,2})日\s+(\d{1,2}):(\d{2})"
+    partial_time_pattern = r"^(\d{1,2}):(\d{2})$"
+    new_message_pattern = r"^以下为新消息$"
+
+    # 解析完整的日期时间
+    match = re.match(full_date_pattern, time_str)
+    if match:
+        year, month, day, hour, minute = map(int, match.groups())
+        return datetime(year, month, day, hour, minute)
+
+    # 解析只有时间（小时:分钟），默认当天的日期
+    match = re.match(partial_time_pattern, time_str)
+    if match:
+        hour, minute = map(int, match.groups())
+        return datetime(now.year, now.month, now.day, hour, minute)
+
+    # 解析 "以下为新消息" 为当前时间
+    if re.match(new_message_pattern, time_str):
+        return now
+
+    # 如果未匹配到任何模式，返回 None 表示无法解析
+    return None
+
+
+@register_tool("1b2p3p3b3p")
+def 系统_查看更多消息(node):
+    node.msg_type = 'system'
+    node.content_type = 'button_more'
+    node.render_text = f'⚙️: 点击此处可 {node.text}'
+
+
+@register_tool("1l2t")
+def 系统_时间标签(node):
+    node.msg_type = 'system'
+    node.content_type = 'time'
+    node.time = parse_time_string(node.text)
+    node.render_text = f'⚙️: {node.text}'
+
+
+# 刚刚发送的特殊时间标签
+@register_tool("1l2p3p3p3t3p3p")
+def 系统_时间标签2(node):
+    node.msg_type = 'system'
+    node.content_type = 'time'
+    node.time = parse_time_string(node.text)
+    node.render_text = f'⚙️: {node.text}'
+
+
+@register_tool("1l2p3p3t3p")
+def 系统_撤回消息(node):
+    node.msg_type = 'system'
+
+    if node.text.endswith('撤回了一条消息'):
+        node.content_type = 'recall'
+        if node.text == '你撤回了一条消息':
+            node.user = node.user2 = '你'
+        else:
+            node.user = re.search(r'"(.+)"', node.text).group(1)
+            node.user2 = node.user
+        node.render_text = f'⚙️: {node.text}'
+    elif node.text.endswith('现在可以开始聊天了。'):
+        node.content_type = 'add_friend'
+        node.user = re.search(r'你已添加了(.+?)，现在可以开始聊天了。', node.text).group(1)
+        node.user2 = node.user
+        node.render_text = f'⚙️: {node.text}'
+    else:
+        raise ValueError
+
+
+@register_tool("1l2p3p3p4p5p6p7t3b")
+def 发送_文本(node):
+    node.msg_type = 'send'
+    node.content_type = 'text'
+    node.user = node[0][2].text
+    node.render_text = f'↑{node.user}: {fmt_text(node.text)}'
+
+
+@register_tool("1l2p3p3p4p5p6p7p7p8p8b3b")
+def 发送_图片(node):
+    node.msg_type = 'send'
+    node.content_type = 'image'
+    node.user = node[0][2].text
+
+
+@register_tool("1l2p3p3p4p5p6p7p7p8p9t9p9b8b3b")
+def 发送_视频(node):
+    node.msg_type = 'send'
+    node.content_type = 'video'
+    node.user = node[0][2].text
+
+
+@register_tool("1l2p3p3p4p5p6p7p7p8p9p10p11t6p7p8p9p10p11t7b3b")
+def 发送_文本_引用文本(node):
+    node.msg_type = 'send'
+    node.content_type = 'text'
+    node.user = node[0][2].text
+    quoted_node = node[0][1][0][0]  # 引用的层级结构
+    node.text = quoted_node[0][1][0][0][0][0].text
+    node.cite_text = quoted_node[1][0][0][0][0][0].text
+
+
+@register_tool("1l2p3p3p4p5p6p7p7p8p9p10p11t6p7p8p9p10p11t11t11p12b7b3b")
+def 发送_文本_引用图片(node):
+    node.msg_type = 'send'
+    node.content_type = 'text'
+    node.user = node[0][2].text
+    quoted_node = node[0][1][0][0]
+    node.text = quoted_node[0][1][0][0][0][0].text
+    node.cite_text = f'{quoted_node[1][0][0][0][0][0].text} : [图片]'
+
+
+# 引用文件
+@register_tool("1l2p3p3p4p5p6p7p7p8p9p10p11t6p7p8p9p10p11p12t11p12b7b3b")
+def 发送_文本_引用文件(node):
+    node.msg_type = 'send'
+    node.content_type = 'text'
+    node.user = node[0][2].text
+    quoted_node = node[0][1][0][0]
+    node.text = quoted_node[0][1][0][0][0][0].text
+    node.cite_text = f'[文件] {quoted_node[1][0][0][0][0][0][0].text}'
+
+
+# 发送文件
+@register_tool("1l2p3p3p4p5p6p7p8p9p10t10p11t11t9p10p10p8p9t7b7b3b")
+def 发送_文本_引用文件(node):
+    node.msg_type = 'send'
+    node.content_type = 'file'
+    node.user = node[0][2].text
+    file_node = node[0][1][0][0][0][0]
+    desc = {
+        'name': file_node[0][0][0].text,
+        'size': file_node[0][0][1][0].text,
+        'platform': file_node[1][0].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+# 发送链接
+@register_tool("1l2p3p3p4p5p6p7p8t8p9t9p9b8p9b9t7b3b")
+def 发送_链接(node):
+    node.msg_type = 'send'
+    node.content_type = 'link'
+    node.user = node[0][2].text
+    link_node = node[0][1][0][0]
+    desc = {
+        'title': link_node[0][0][0].text,
+        'head': link_node[0][0][1][0].text,
+        'author': link_node[0][0][2][1].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+# 转发消息
+@register_tool("1l2p3p3p4p5p6p7p8t8p9t9t9t8p9p9t7b3b")
+def 发送_转发消息(node):
+    node.msg_type = 'send'
+    node.content_type = 'messages'
+    node.user = node[0][2].text
+
+
+@register_tool("1l2p3b3p4p5p6p7t3p")
+def 接收_文本(node):
+    node.msg_type = 'receive'
+    node.content_type = 'text'
+    node.user = node[0][0].text
+    node.render_text = f'↓{node.user}: {fmt_text(node.text)}'
+
+
+@register_tool("1l2p3b3p4p5p6p7p7b3p")
+def 接收_图片(node):
+    node.msg_type = 'receive'
+    node.content_type = 'image'
+    node.user = node[0][0].text
+    node.render_text = f'↓{node.user}: [图片]'
+
+
+@register_tool("1l2p3b3p4p5p6p7p8p9p10t10p11t11t9p10p10p8p9t7b7b3p")
+def 接收_文件(node):
+    node.msg_type = 'receive'
+    node.content_type = 'file'
+    node.user = node[0][0].text
+    node.render_text = f'↓{node.user}: {fmt_text(node.text)}'
+
+
+@register_tool("1l2p3b3p4p5p6p7p8b8p8t7b3p")
+def 接收_语音(node):
+    node.msg_type = 'receive'
+    node.content_type = 'voice'
+    node.user = node[0][0].text
+
+
+@register_tool("1l2p3b3p4p5p6p7p8t8p9t9p9b8p9b9t7b3p")
+def 接收_链接(node):
+    node.msg_type = 'receive'
+    node.content_type = 'link'
+    node.user = node[0][0].text
+    link_node = node[0][1][0][0][0][0]  # 链接内容的层级结构
+    desc = {
+        'title': link_node[0].text,
+        'head': link_node[1][0].text,  # 文章开头部分
+        'author': link_node[2][1].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7t3p")
+def 群接收_文本(node):
+    node.msg_type = 'receive'
+    node.content_type = 'text'
+    node.user = node[0][0].text
+    node.user2 = node[0][1][0][0].text
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p7b3p")
+def 群接收_图片(node):
+    node.msg_type = 'receive'
+    node.content_type = 'image'
+    node.user = node[0][0].text
+    node.user2 = node[0][1][0][0].text
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8p9p10t10p11t11t9p10p10p7b7b3p")
+def 群接收_文件(node):
+    node.msg_type = 'receive'
+    node.content_type = 'file'
+    node.user = node[0][0].text
+    file_node = node[0][1]  # 文件信息的节点
+    node.user2 = file_node[0][0].text
+    file_info_node = file_node[1][0][0][0][0]
+    desc = {
+        'name': file_info_node[0][0].text,
+        'size': file_info_node[0][1][0].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8p9p10t10p11t11t9p10p10p8p9t7b7b3p")
+def 群接收_文件2(node):
+    node.msg_type = 'receive'
+    node.content_type = 'file'
+    node.user = node[0][0].text
+    node.user2 = node[0][1][0][0].text
+    file_info_node = node[0][1][1][0][0][0]
+    desc = {
+        'name': file_info_node[0][0][0].text,
+        'size': file_info_node[0][0][1][0].text,
+        'platform': file_info_node[1][0].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7b3p")
+def 群接收_动画表情(node):
+    node.msg_type = 'receive'
+    node.content_type = 'emoji'
+    node.user = node[0][0].text
+    node.user2 = node[0][1][0][0].text
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8t8p9t9p9b8p9b9t7b3p")
+def 群接收_链接_有公众号作者名(node):
+    node.msg_type = 'receive'
+    node.content_type = 'link'
+    node.user = node[0][0].text
+    link_node = node[0][1]
+    node.user2 = link_node[0][0].text
+    link_info_node = link_node[1][0][0][0]
+    desc = {
+        'title': link_info_node[0].text,
+        'head': link_info_node[1][0].text,  # 文章开头的一小部分
+        'author': link_info_node[2][0].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8t8p9t9p9b7b3p")
+def 群接收_链接_无公众号作者名(node):
+    node.msg_type = 'receive'
+    node.content_type = 'link'
+    node.user = node[0][0].text
+    link_node = node[0][1]
+    node.user2 = link_node[0][0].text
+    link_info_node = link_node[1][0][0][0]
+    desc = {
+        'title': link_info_node[0].text,
+        'head': link_info_node[1][0].text  # 文章开头的一小部分
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8b8p9p8p9p10p10t8p9p7b3p")
+def 群接收_视频(node):
+    node.msg_type = 'receive'
+    node.content_type = 'video'
+    node.user = node[0][0].text
+    video_node = node[0][1]
+    node.user2 = video_node[0][0].text
+    video_info_node = video_node[1][0][0][0]
+    desc = {
+        'author': video_info_node[2][0][1].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8t8p9t8t7b3p")
+def 群接收_视频2(node):
+    node.msg_type = 'receive'
+    node.content_type = 'video'
+    node.user = node[0][0].text
+    video_node = node[0][1]
+    node.user2 = video_node[0][0].text
+    video_info_node = video_node[1][0][0][0]
+    desc = {
+        'title': video_info_node[0].text,
+        'head': video_info_node[1][0].text,
+        'duration': video_info_node[2].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8p9b9t8t8b8p9t9t7b3p")
+def 群接收_小程序(node):
+    node.msg_type = 'receive'
+    node.content_type = 'applet'
+    node.user = node[0][0].text
+    applet_node = node[0][1]
+    node.user2 = applet_node[0][0].text
+    applet_info_node = applet_node[1][0][0][0]
+    desc = {
+        'name': applet_info_node[0][1].text,
+        'title': applet_info_node[1].text,
+        'footnote': applet_info_node[3][1].text
+    }
+    node.text = json.dumps(desc, ensure_ascii=False)
+
+
+@register_tool("1l2p3p3p4p5l6p7p7t7p3p")
+def 群接收_拍一拍(node):
+    node.msg_type = 'receive'
+    node.content_type = 'shake'
+    node.text = node[0][1][0][0].text
+    node.user = re.search(r'"(.+?)" 拍了拍 "', node.text).group(1)
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8t4p5p6p7p8p9t5b3p")
+def 群接收_引用消息(node):
+    node.msg_type = 'receive'
+    node.content_type = 'text'
+    node.user = node[0][0].text
+    quoted_node = node[0][1]
+    node.user2 = quoted_node[0][0].text
+    node.text = quoted_node[1][0][0][0][0].text
+    node.cite_text = quoted_node[2][0][0][0][0][0].text
+
+
+@register_tool("1l2p3b3p4p5t4p5p6p7p8t4p5p6p7p8p9p10t9p10b5b3p")
+def 群接收_引用文件(node):
+    node.msg_type = 'receive'
+    node.content_type = 'text'
+    node.user = node[0][0].text
+    quoted_node = node[0][1]
+    node.user2 = quoted_node[0][0].text
+    node.text = quoted_node[1][0][0][0][0].text
+    file_node = quoted_node[2][0][0][0][0][0][0]
+    node.cite_text = f'[文件] {file_node.text}'

pyxllib/autogui/wxautolib.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2024/12/16
+
+import sys
+
+from loguru import logger
+
+if sys.platform == 'win32':
+    try:  # 尝试加载VIP版
+        from wxautox import WeChat
+        from wxautox.elements import WxParam  # WxParam.DEFALUT_SAVEPATH 可以用来配置数据自动保存位置
+    except ModuleNotFoundError:  # 否则用贫民版
+        from wxauto import WeChat
+        from wxauto.elements import WxParam
+
+from pyxllib.prog.filelock import get_autoui_lock
+
+
+class WeChatSingletonLock:
+    """ 基于 get_autoui_lock 的微信全局唯一单例控制器，确保同一时间仅有一个微信自动化程序在操作 """
+
+    def __init__(self, lock_timeout=-1, *, init=True):
+        # 初始化全局锁
+        self.lock = get_autoui_lock(timeout=lock_timeout)
+        self.wx = WeChat() if init else None
+
+    def __enter__(self):
+        # 获取锁并激活微信窗口
+        self.lock.acquire()
+        if self.wx:
+            self.wx._show()
+            return self.wx
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        # 释放锁
+        self.lock.release()
+
+
+def wechat_lock_send(user, text=None, files=None, *, timeout=-1):
+    """ 使用全局唯一单例锁，确保同一时间仅有一个微信自动化程序在操作 """
+    with WeChatSingletonLock(timeout) as we:
+        # 241223周一12:27，今天可被这个默认2秒坑惨了，往错误群一直发骚扰消息
+        # 22:07，但我复测，感觉不可能找不到啊，为什么会找到禅宗考勤管理群呢，太离谱了
+        status = we.ChatWith(user, timeout=5)
+
+        if status != user:
+            raise ValueError(f'无法找到用户：{user}')
+
+        if text:
+            we.SendMsg(text, user)
+        if files:
+            we.SendFiles(files, user)
+
+
+def wechat_handler(message):
+    # 获取群名，如果没有指定，不使用此微信发送功能
+    user = message.record["extra"].get("wechat_user")
+    if user:
+        wechat_lock_send(user, message)
+
+
+if sys.platform == 'win32':
+    # 创建专用的微信日志记录器，不绑定默认群名
+    wechat_logger = logger.bind(wechat_user='文件传输助手')
+
+    # 添加专用的微信处理器
+    wechat_logger.add(wechat_handler,
+                      format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {name}:{function}:{line} - {message}")
+
+    """ 往微信发送特殊的日志格式报告
+    用法：wechat_logger.bind(wechat_user='文件传输助手').info(message)
+
+    或者：
+    # 先做好默认群名绑定
+    wechat_logger = wechat_logger.bind(wechat_user='考勤管理')
+    # 然后就能普通logger用法发送了
+    wechat_logger.info('测试')
+    """
+else:
+    # 降级为普通logger
+    wechat_logger = logger

pyxllib/cv/slidercaptcha.py

@@ -0,0 +1,137 @@
+import os
+import cv2
+import numpy as np
+from xlproject.code4101 import XlPath
+
+from pyxllib.file.specialist import download_file
+from pyxllib.prog.pupil import is_url
+
+
+class SliderCaptchaLocator:
+    """ 滑动验证码的位置定位
+
+    由于缺口一般是一个黑底图，整体颜色是偏暗的，可以利用这个特性来定位列所在位置
+    """
+
+    def __init__(self, background_image_path, slider_image_path=None):
+        """
+        :param background_image_path: 背景主图片
+        :param slider_image_path: 滑动条图片
+        """
+        # 支持输入是url的情况
+        if is_url(background_image_path):
+            background_image_path = download_file(background_image_path, temp=True, ext='.png')
+        if is_url(slider_image_path):
+            slider_image_path = download_file(slider_image_path, temp=True, ext='.png')
+        # 初始化图片等数据
+        self.image_path = background_image_path
+        self.image = cv2.imread(self.image_path)
+        self.darkest_rows = None
+        self.column_averages = None
+        self.enhanced_averages = None
+        self.radius = 24
+        if slider_image_path is not None:
+            # 先读取图片
+            template_image = cv2.imread(slider_image_path)
+            # 取图片宽度的40%作为radius
+            self.radius = int(template_image.shape[1] * 0.4)
+
+    def select_darkest_part_rows(self, fraction=0.1):
+        """
+        选择最暗的部分行，供下游计算列的平均暗度
+
+        :param float fraction: 最暗行的比例
+        :return: 最暗行的图像部分
+        """
+        if self.darkest_rows is None:
+            average_row_values = np.mean(self.image, axis=(1, 2))
+            threshold_index = int(len(average_row_values) * fraction)
+            darkest_indices = np.argsort(average_row_values)[:threshold_index]
+            self.darkest_rows = self.image[darkest_indices]
+        return self.darkest_rows
+
+    def calculate_column_averages(self):
+        """
+        计算列的平均强度。
+
+        :return: 列的平均强度
+        """
+        if self.column_averages is None:
+            rows = self.select_darkest_part_rows()
+            column_values = np.mean(rows, axis=0)
+            column_intensity = np.mean(column_values, axis=1)
+            min_val, max_val = np.min(column_intensity), np.max(column_intensity)
+            self.column_averages = ((column_intensity - min_val) / (max_val - min_val) * 255).astype(
+                np.uint8) if max_val != min_val else np.zeros_like(column_intensity)
+        return self.column_averages
+
+    def enhance_column_averages(self):
+        """
+        增强列平均值，利用验证码的黑度图的对称性，进行一个增强，使得其他非验证码的零散黑条变白
+
+        :param int radius: 对比半径，这个建议设成模版图宽度的40%~50%
+        :return: 增强后的列平均值
+        """
+        if self.enhanced_averages is None:
+            averages = self.calculate_column_averages()
+            length = len(averages)
+            enhanced = np.zeros_like(averages)
+            for i in range(length):
+                max_value = averages[i]
+                for j in range(1, self.radius // 2 + 1):
+                    if i - j >= 0 and i + j < length:
+                        max_value = max(max_value, abs(int(averages[i - j]) - int(averages[i + j])))
+                enhanced[i] = max_value
+            self.enhanced_averages = enhanced
+        return self.enhanced_averages
+
+    def find_captcha_position(self, threshold=50):
+        """
+        查找验证码的位置。
+
+        :param int threshold: 达到多少像素后，开始计算中位数
+        :return: 验证码的列中心位置
+        """
+        intensities = self.enhance_column_averages()
+        indices = []
+        for intensity in range(256):
+            new_indices = np.where(intensities == intensity)[0]
+            if len(indices) + len(new_indices) > threshold:
+                indices.extend(new_indices[:threshold - len(indices)])
+                break
+            indices.extend(new_indices)
+        return int(np.median(indices))
+
+    def draw_red_line(self, position, line_thickness=3):
+        """
+        在图像上绘制红线。
+
+        :param int position: 红线位置
+        :param int line_thickness: 线宽
+        :return: 绘制红线后的图像
+        """
+        gray_image = np.tile(self.calculate_column_averages(), (100, 1)).astype(np.uint8)
+        color_image = cv2.cvtColor(gray_image, cv2.COLOR_GRAY2BGR)
+        cv2.line(color_image, (position, 0), (position, gray_image.shape[0]), (0, 0, 255), line_thickness)
+        return color_image
+
+    def debug(self):
+        position = self.find_captcha_position()
+        final_image = self.draw_red_line(position)
+        cv2.imshow('Image with Red Line', final_image)
+        cv2.waitKey(0)
+        cv2.destroyAllWindows()
+
+
+def main():
+    os.chdir(XlPath.desktop())
+    locator = SliderCaptchaLocator('cap_union_new_getcapbysig.png')
+    position = locator.find_captcha_position()
+    final_image = locator.draw_red_line(position)
+    cv2.imshow('Image with Red Line', final_image)
+    cv2.waitKey(0)
+    cv2.destroyAllWindows()
+
+
+if __name__ == '__main__':
+    main()