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

pyxllib/ext/webhook.py

@@ -1,101 +1,124 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# @Author : 陈坤泽
-# @Email  : 877362867@qq.com
-# @Date   : 2020/10/21 09:22
-
-
-import time
-import hmac
-import hashlib
-import base64
-import urllib.parse
-
-import requests
-
-
-class WeixinRobot:
-    """ 企业微信 机器人 """
-
-    def __init__(self, url):
-        self.url = url
-
-    def push_text(self, s):
-        msgtype = 'text'
-        try:
-            headers = {"Content-Type": "text/plain"}
-            t = {"content": s} if isinstance(s, str) else s
-            data = {"msgtype": msgtype, msgtype: t}  # msgtype: text、markdown
-            requests.post(url=self.url, headers=headers, json=data)
-        except requests.exceptions.ConnectionError:  # 没网发送失败的时候也不报错
-            pass
-
-
-class DingtalkRobot:
-    """ 钉钉 自定义webhook机器人
-
-    https://ding-doc.dingtalk.com/doc#/serverapi2/qf2nxq
-    """
-
-    def __init__(self, url, secret):
-        self.base_url = url  # 原始url，不包含时间戳和签名
-        self.secret = secret
-        self.headers = {"Content-Type": "application/json"}
-        self.last_timestamp = 0  # 初始化上次时间戳
-        self.update_url()  # 初始化时更新URL
-
-    def update_url(self):
-        """更新URL中的时间戳和签名"""
-        self.url = self.base_url + self.add_secret(self.secret)
-        self.last_timestamp = round(time.time())  # 更新上次时间戳
-
-    @classmethod
-    def add_secret(cls, secret):
-        """ 钉钉机器人需要加签，确保安全性 """
-        timestamp = str(round(time.time() * 1000))
-        secret_enc = secret.encode('utf-8')
-        string_to_sign = '{}\n{}'.format(timestamp, secret)
-        string_to_sign_enc = string_to_sign.encode('utf-8')
-        hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
-        sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
-        return f'&timestamp={timestamp}&sign={sign}'
-
-    def send_data(self, data):
-        # 检查当前时间是否超过30分钟
-        if round(time.time()) - self.last_timestamp > 1800:  # 30 * 60
-            self.update_url()  # 更新URL
-        try:
-            requests.post(url=self.url, headers=self.headers, json=data)
-        except requests.exceptions.ConnectionError as e:  # 没网发送失败的时候也不报错
-            raise e
-
-    def send_text(self, content):
-        msgtype = 'text'
-        d = {}
-        if content: d['content'] = content
-        data = {"msgtype": msgtype, msgtype: d}
-        self.send_data(data)
-
-    def send_link(self, text='', title='', pic_url='', message_url=''):
-        msgtype = 'link'
-        d = {}
-        if text: d['text'] = text
-        if title: d['title'] = title
-        if pic_url: d['picUrl'] = pic_url
-        if message_url: d['messageUrl'] = message_url
-        data = {"msgtype": msgtype, msgtype: d}
-        self.send_data(data)
-
-    def send_markdown(self, text='', title=''):
-        msgtype = 'link'
-        d = {}
-        if text: d['text'] = text
-        if title: d['title'] = title
-        data = {"msgtype": msgtype, msgtype: d}
-        self.send_data(data)
-
-    def send_actioncard(self, text='', title='', siggle_url='', siggle_title='', btn_orientation='0'):
-        raise NotImplementedError
-
-    def send_feedcard(self):
-        raise NotImplementedError
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2020/10/21 09:22
+
+import os
+import time
+import hmac
+import hashlib
+import base64
+import urllib.parse
+
+import requests
+
+
+class WeixinRobot:
+    """ 企业微信 机器人 """
+
+    def __init__(self, url):
+        self.url = url
+
+    def push_text(self, s):
+        msgtype = 'text'
+        try:
+            headers = {"Content-Type": "text/plain"}
+            t = {"content": s} if isinstance(s, str) else s
+            data = {"msgtype": msgtype, msgtype: t}  # msgtype: text、markdown
+            requests.post(url=self.url, headers=headers, json=data)
+        except requests.exceptions.ConnectionError:  # 没网发送失败的时候也不报错
+            pass
+
+
+class DingtalkRobot:
+    """ 钉钉 自定义webhook机器人
+
+    https://ding-doc.dingtalk.com/doc#/serverapi2/qf2nxq
+    """
+
+    def __init__(self, access_token=None, secret=None):
+        access_token = access_token or os.getenv('DINGTALK_ROBOT_SECRET')
+        self.url = f'https://oapi.dingtalk.com/robot/send?access_token={access_token}'
+        self.url += self.add_secret(secret or os.getenv('DINGTALK_ROBOT_SECRET'))
+        self.headers = {"Content-Type": "application/json"}
+
+    @classmethod
+    def add_secret(cls, secret):
+        """ 钉钉机器人需要加签，确保安全性 """
+        timestamp = str(round(time.time() * 1000))
+        secret_enc = secret.encode('utf-8')
+        string_to_sign = '{}\n{}'.format(timestamp, secret)
+        string_to_sign_enc = string_to_sign.encode('utf-8')
+        hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
+        sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
+        return f'&timestamp={timestamp}&sign={sign}'
+
+    def send_data(self, data):
+        try:
+            requests.post(url=self.url, headers=self.headers, json=data)
+        except requests.exceptions.ConnectionError as e:  # 没网发送失败的时候也不报错
+            raise e
+
+    def send_text(self, content):
+        msgtype = 'text'
+        d = {}
+        if content: d['content'] = content
+        data = {"msgtype": msgtype, msgtype: d}
+        self.send_data(data)
+
+    def send_link(self, text='', title='', pic_url='', message_url=''):
+        msgtype = 'link'
+        d = {}
+        if text: d['text'] = text
+        if title: d['title'] = title
+        if pic_url: d['picUrl'] = pic_url
+        if message_url: d['messageUrl'] = message_url
+        data = {"msgtype": msgtype, msgtype: d}
+        self.send_data(data)
+
+    def send_markdown(self, text='', title=''):
+        msgtype = 'link'
+        d = {}
+        if text: d['text'] = text
+        if title: d['title'] = title
+        data = {"msgtype": msgtype, msgtype: d}
+        self.send_data(data)
+
+    def send_actioncard(self, text='', title='', siggle_url='', siggle_title='', btn_orientation='0'):
+        raise NotImplementedError
+
+    def send_feedcard(self):
+        raise NotImplementedError
+
+
+class DingtalkRobot2(DingtalkRobot):
+
+    def __init__(self, title=None):
+        super().__init__()
+        self.title = title
+
+    def send_text2(self, text):  # 增加一个更加定制化的便捷接口
+        from pyxllib.prog.pupil import utc_timestamp
+
+        if self.title:
+            self.send_text(f'{utc_timestamp()} {get_host_nickname()} [{self.title}] {text}')
+        else:
+            self.send_text(f'{utc_timestamp()} {get_host_nickname()} {text}')
+
+    def __enter__(self):
+        self.send_text2('启动')
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        from pyxllib.prog.pupil import format_exception
+
+        if exc_tb is None:
+            self.send_text2('完成')
+        else:
+            self.send_text2(f'报错\n{format_exception(exc_val, 3)}')
+
+if __name__ == '__main__':
+    pass
+
+    from pyxllib.prog.newbie import typename

pyxllib/ext/win32lib.py

@@ -1,40 +1,40 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-# @Author : 陈坤泽
-# @Email  : 877362867@qq.com
-# @Date   : 2021/09/07 10:21
-
-
-import win32com.client as win32
-import pythoncom
-
-
-def get_win32_app(name, visible=False):
-    """ 启动可支持pywin32自动化处理的应用
-
-    Args:
-        str name: 应用名称，不区分大小写，比如word, excel, powerpoint, onenote
-            不带'.'的情况下，会自动添加'.Application'的后缀
-        visible: 应用是否可见
-
-    Returns: app
-
-    """
-    # 1 name
-    name = name.lower()
-    if '.' not in name:
-        name += '.application'
-
-    # 2 app
-    # 这里可能还有些问题，不同的应用，机制不太一样，后面再细化完善吧
-    try:
-        app = win32.GetActiveObject(f'{name}')  # 不能关联到普通方式打开的应用。但代码打开的应用都能找得到。
-    except pythoncom.com_error:
-        app = win32.gencache.EnsureDispatch(f'{name}')
-        # 还有种常见的初始化方法，是 win32com.client.Dispatch和win32com.client.dynamic.Dispatch
-        # from win32com.client.dynamic import Disypatch
-
-    if visible is not None:
-        app.Visible = visible
-
-    return app
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2021/09/07 10:21
+
+
+import win32com.client as win32
+import pythoncom
+
+
+def get_win32_app(name, visible=False):
+    """ 启动可支持pywin32自动化处理的应用
+
+    Args:
+        str name: 应用名称，不区分大小写，比如word, excel, powerpoint, onenote
+            不带'.'的情况下，会自动添加'.Application'的后缀
+        visible: 应用是否可见
+
+    Returns: app
+
+    """
+    # 1 name
+    name = name.lower()
+    if '.' not in name:
+        name += '.application'
+
+    # 2 app
+    # 这里可能还有些问题，不同的应用，机制不太一样，后面再细化完善吧
+    try:
+        app = win32.GetActiveObject(f'{name}')  # 不能关联到普通方式打开的应用。但代码打开的应用都能找得到。
+    except pythoncom.com_error:
+        app = win32.gencache.EnsureDispatch(f'{name}')
+        # 还有种常见的初始化方法，是 win32com.client.Dispatch和win32com.client.dynamic.Dispatch
+        # from win32com.client.dynamic import Disypatch
+
+    if visible is not None:
+        app.Visible = visible
+
+    return app

pyxllib/ext/wjxlib.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2024/11/19
+
+""" 问卷星 相关工具 """
+
+import os
+import io
+import time
+
+from loguru import logger
+from DrissionPage import Chromium
+import pandas as pd
+
+from pyxllib.ext.drissionlib import DpWebBase
+
+
+class WjxWeb(DpWebBase):
+    """ 问卷星网页的爬虫 """
+
+    def __init__(self, url=None):
+        super().__init__(url or 'https://www.wjx.cn')
+        self.login()
+
+    def login(self):
+        tab = self.tab
+
+        if tab.url.startswith('https://www.wjx.cn/wjx/activitystat/resultlimit.aspx'):
+            tab('tag:a@@text():登录').click()
+
+        if tab.url.lower().startswith('https://www.wjx.cn/login.aspx'):
+            tab('tag:input@@name=UserName').input(os.getenv('WJX_USERNAME'), clear=True)
+            tab('tag:input@@name=Password').input(os.getenv('WJX_PASSWORD'), clear=True)
+            tab('tag:input@@type=submit').click()
+
+    def get_page_num(self):
+        """
+        返回当前页编号和总页数 (idx, num)。
+        """
+        idx, num = map(int, self.tab('tag:span@@class=paging-num').text.split('/'))
+        return idx, num
+
+    def prev_page(self):
+        self.tab('tag:a@@class=go-pre').click()
+
+    def next_page(self):
+        self.tab('tag:a@@class=go-next').click()
+
+    def _parse_table(self):
+        """处理并解析网页中的表格数据"""
+        table_html = self.tab('tag:table').html
+        df = pd.read_html(io.StringIO(table_html))[0]  # 读取表格
+        df.columns = [col.replace('\ue645', '') for col in df.columns]
+        # "星标"的内容特殊字符
+        df.replace('\ue66b', '', regex=True, inplace=True)
+        # "操作"的内容特殊字符
+        df.replace('\ue6a3\ue6d4', '', regex=True, inplace=True)
+        return df
+
+    def set_num_of_page(self, num_of_page):
+        """ 查看数据页面，设置每页显示多少条记录 """
+        select = self.tab('tag:span@@text():每页显示').next('tag:select')
+        select.click()
+        opt = select(f'tag:option@@text()={num_of_page}')
+        if opt.attr('selected') != 'selected':
+            opt.click()
+        else:
+            select.click()
+
+    def get_df(self, all_pages=False):
+        """获得当前页面的表格数据，如果 all_pages 为 True，则下载所有页面的数据"""
+        # 初始化DataFrame列表，用于存储每页的数据
+        dfs = [self._parse_table()]  # 获取当前页面的数据
+
+        # 如果需要下载所有页面数据
+        if all_pages:
+            current_idx, total_pages = self.get_page_num()
+            while current_idx < total_pages:
+                self.next_page()  # 翻到下一页
+                time.sleep(2)
+                dfs.append(self._parse_table())  # 获取并处理新一页的数据
+                current_idx, total_pages = self.get_page_num()  # 更新页码信息
+
+        # 将所有数据合并为一个DataFrame
+        final_df = pd.concat(dfs, ignore_index=True) if all_pages else dfs[0]
+        return final_df

pyxllib/ext/wpsapi.py

@@ -0,0 +1,124 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Date   : 2024/07/31
+
+import os
+import re
+
+import requests
+import pandas as pd
+
+
+class WpsOnlineBook:
+    """ wps的"脚本令牌"调用模式
+
+    官方文档：https://airsheet.wps.cn/docs/apitoken/api.html
+    """
+
+    def __1_基础功能(self):
+        pass
+
+    def __init__(self, file_id, script_id=None, *, token=None):
+        self.headers = {
+            'Content-Type': "application/json",
+            'AirScript-Token': token or os.getenv('WPS_SCRIPT_TOKEN', ''),
+        }
+        self.file_id = file_id
+        self.default_script_id = script_id
+
+    def post_request(self, url, payload):
+        """
+        发送 POST 请求到指定的 URL 并返回响应结果
+        """
+        try:
+            resp = requests.post(url, json=payload, headers=self.headers)
+            resp.raise_for_status()  # 如果请求失败会抛出异常
+            return resp.json()
+        except requests.exceptions.RequestException as e:
+            print(f"请求失败: {e}")
+            return None
+
+    def run_script(self, script_id=None, context_argv=None, sync=True):
+        """ 原本执行 WPS 脚本并返回执行结果
+
+        :param script_id: 脚本 ID
+        :param context_argv: 脚本参数 (可选)
+            context本来能支持这些参数的：
+                dict argv: 传入的上下文参数对象，比如传入{name: 'xiaomeng', age: 18}， 在 AS 代码中可通过Context.argv.name获取到传入的值
+                str sheet_name: et,ksheet 运行时所在表名
+                str range: et,ksheet 运行时所在区域，例如$B$156
+                str link_from: et,ksheet 点击超链接所在单元格
+                str db_active_view: db 运行时所在 view 名
+                str db_selection: db 运行时所在选区
+            但是，sheet_name, range等，并看不到对运行代码有什么实质影响，不影响active，而且as里也引用不了sheet_name等值
+                所以退化，简化为只要传入content_argv参数就行
+        :param sync:
+            True, 同步运行
+            False, 异步运行
+
+        这个接口跟普通as一样，运行有30秒时限
+        """
+        url = f"https://www.kdocs.cn/api/v3/ide/file/{self.file_id}/script/{script_id}/{'sync_task' if sync else 'task'}"
+        payload = {
+            "Context": {'argv': context_argv or {}}
+        }
+        res = self.post_request(url, payload)
+        if sync:
+            return res['data']['result']
+        else:
+            return res
+
+    def __2_封装的更高级的接口(self):
+        """ 这系列的功能需要配套这个框架范式使用：
+        https://github.com/XLPRUtils/pyxllib/blob/master/pyxllib/text/airscript.js
+        """
+        pass
+
+    def run_func(self, func_name, *args):
+        """ 我自己常用的jsa框架，jsa那边已经简化了对接模式，所以一般都只用这个高级的接口即可
+        （旧函数名run_script2不再使用）
+        """
+        return self.run_script(self.default_script_id, context_argv={'funcName': func_name, 'args': args})
+
+    def get_sheet_data(self, sheet_name, fields, data_row=0, filter_empty_rows=True, *,
+                       return_mode='pd') -> pd.DataFrame:
+        """ 获取某张sheet表格数据
+
+        :param sheet_name: sheet表名
+        :param list[str] fields: 字段名列表
+        :param int data_row: 数据起始行，详细用法见packTableDataFields
+        :param return_mode: 'pd' or 'json'
+        """
+        data = self.run_func('packTableDataFields', sheet_name, fields, data_row, filter_empty_rows)
+        if return_mode == 'json':
+            return data
+        elif return_mode == 'pd':
+            return pd.DataFrame(data)
+
+    def write_arr(self, rows, start_cell, batch_size=None):
+        """ 把一个二维数组数据写入表格
+
+        :param rows: 一个n*m的数据
+        :param start_cell: 写入的起始位置，例如'A1'，也可以使用Sheet1!A1的格式表示具体的表格位置
+        :param batch_size: 为了避免一次写入内容过多，超时写入失败，可以分成多批运行
+            这里写每批的数据行数
+            默认表示一次性全部提交
+        :return:
+        """
+        if batch_size is None:
+            batch_size = len(rows)  # 如果未指定批次大小，一次性写入所有行
+
+        def func(m):
+            return str(int(m.group()) + batch_size)
+
+        current_cell = start_cell
+        for start in range(0, len(rows), batch_size):
+            end = start + batch_size
+            batch_rows = rows[start:end]
+            self.run_func('writeArrToSheet', batch_rows, current_cell)
+            current_cell = re.sub(r'\d+$', func, current_cell)
+
+
+if __name__ == '__main__':
+    pass

pyxllib/ext/xlwork.py

@@ -0,0 +1,9 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# @Author : 陈坤泽
+# @Email  : 877362867@qq.com
+# @Date   : 2025/04/17
+
+""" code4101的众多账号密码管理接口，工作目录等控制位置 """
+
+