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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (127) hide show
  1. pyxllib/__init__.py +14 -21
  2. pyxllib/algo/__init__.py +8 -8
  3. pyxllib/algo/disjoint.py +54 -54
  4. pyxllib/algo/geo.py +537 -541
  5. pyxllib/algo/intervals.py +964 -964
  6. pyxllib/algo/matcher.py +389 -389
  7. pyxllib/algo/newbie.py +166 -166
  8. pyxllib/algo/pupil.py +629 -629
  9. pyxllib/algo/shapelylib.py +67 -67
  10. pyxllib/algo/specialist.py +241 -241
  11. pyxllib/algo/stat.py +494 -494
  12. pyxllib/algo/treelib.py +145 -149
  13. pyxllib/algo/unitlib.py +62 -66
  14. pyxllib/autogui/__init__.py +5 -5
  15. pyxllib/autogui/activewin.py +246 -246
  16. pyxllib/autogui/all.py +9 -9
  17. pyxllib/autogui/autogui.py +846 -852
  18. pyxllib/autogui/uiautolib.py +362 -362
  19. pyxllib/autogui/virtualkey.py +102 -102
  20. pyxllib/autogui/wechat.py +827 -827
  21. pyxllib/autogui/wechat_msg.py +421 -421
  22. pyxllib/autogui/wxautolib.py +84 -84
  23. pyxllib/cv/__init__.py +5 -5
  24. pyxllib/cv/expert.py +267 -267
  25. pyxllib/cv/imfile.py +159 -159
  26. pyxllib/cv/imhash.py +39 -39
  27. pyxllib/cv/pupil.py +9 -9
  28. pyxllib/cv/rgbfmt.py +1525 -1525
  29. pyxllib/cv/slidercaptcha.py +137 -137
  30. pyxllib/cv/trackbartools.py +251 -251
  31. pyxllib/cv/xlcvlib.py +1040 -1040
  32. pyxllib/cv/xlpillib.py +423 -423
  33. pyxllib/data/echarts.py +236 -240
  34. pyxllib/data/jsonlib.py +85 -89
  35. pyxllib/data/oss.py +72 -72
  36. pyxllib/data/pglib.py +1111 -1127
  37. pyxllib/data/sqlite.py +568 -568
  38. pyxllib/data/sqllib.py +297 -297
  39. pyxllib/ext/JLineViewer.py +505 -505
  40. pyxllib/ext/__init__.py +6 -6
  41. pyxllib/ext/demolib.py +251 -246
  42. pyxllib/ext/drissionlib.py +277 -277
  43. pyxllib/ext/kq5034lib.py +12 -12
  44. pyxllib/ext/qt.py +449 -449
  45. pyxllib/ext/robustprocfile.py +493 -497
  46. pyxllib/ext/seleniumlib.py +76 -76
  47. pyxllib/ext/tk.py +173 -173
  48. pyxllib/ext/unixlib.py +821 -827
  49. pyxllib/ext/utools.py +345 -351
  50. pyxllib/ext/webhook.py +124 -119
  51. pyxllib/ext/win32lib.py +40 -40
  52. pyxllib/ext/wjxlib.py +91 -88
  53. pyxllib/ext/wpsapi.py +124 -124
  54. pyxllib/ext/xlwork.py +9 -9
  55. pyxllib/ext/yuquelib.py +1110 -1105
  56. pyxllib/file/__init__.py +17 -17
  57. pyxllib/file/docxlib.py +757 -761
  58. pyxllib/file/gitlib.py +309 -309
  59. pyxllib/file/libreoffice.py +165 -165
  60. pyxllib/file/movielib.py +144 -148
  61. pyxllib/file/newbie.py +10 -10
  62. pyxllib/file/onenotelib.py +1469 -1469
  63. pyxllib/file/packlib/__init__.py +330 -330
  64. pyxllib/file/packlib/zipfile.py +2441 -2441
  65. pyxllib/file/pdflib.py +422 -426
  66. pyxllib/file/pupil.py +185 -185
  67. pyxllib/file/specialist/__init__.py +681 -685
  68. pyxllib/file/specialist/dirlib.py +799 -799
  69. pyxllib/file/specialist/download.py +193 -193
  70. pyxllib/file/specialist/filelib.py +2825 -2829
  71. pyxllib/file/xlsxlib.py +3122 -3131
  72. pyxllib/file/xlsyncfile.py +341 -341
  73. pyxllib/prog/__init__.py +5 -5
  74. pyxllib/prog/cachetools.py +58 -64
  75. pyxllib/prog/deprecatedlib.py +233 -233
  76. pyxllib/prog/filelock.py +42 -42
  77. pyxllib/prog/ipyexec.py +253 -253
  78. pyxllib/prog/multiprogs.py +940 -940
  79. pyxllib/prog/newbie.py +451 -451
  80. pyxllib/prog/pupil.py +1208 -1197
  81. pyxllib/prog/sitepackages.py +33 -33
  82. pyxllib/prog/specialist/__init__.py +348 -391
  83. pyxllib/prog/specialist/bc.py +203 -203
  84. pyxllib/prog/specialist/browser.py +497 -497
  85. pyxllib/prog/specialist/common.py +347 -347
  86. pyxllib/prog/specialist/datetime.py +198 -198
  87. pyxllib/prog/specialist/tictoc.py +240 -240
  88. pyxllib/prog/specialist/xllog.py +180 -180
  89. pyxllib/prog/xlosenv.py +110 -108
  90. pyxllib/stdlib/__init__.py +17 -17
  91. pyxllib/stdlib/tablepyxl/__init__.py +10 -10
  92. pyxllib/stdlib/tablepyxl/style.py +303 -303
  93. pyxllib/stdlib/tablepyxl/tablepyxl.py +130 -130
  94. pyxllib/text/__init__.py +8 -8
  95. pyxllib/text/ahocorasick.py +36 -39
  96. pyxllib/text/airscript.js +754 -744
  97. pyxllib/text/charclasslib.py +121 -121
  98. pyxllib/text/jiebalib.py +267 -267
  99. pyxllib/text/jinjalib.py +27 -32
  100. pyxllib/text/jsa_ai_prompt.md +271 -271
  101. pyxllib/text/jscode.py +922 -922
  102. pyxllib/text/latex/__init__.py +158 -158
  103. pyxllib/text/levenshtein.py +303 -303
  104. pyxllib/text/nestenv.py +1215 -1215
  105. pyxllib/text/newbie.py +300 -300
  106. pyxllib/text/pupil/__init__.py +8 -8
  107. pyxllib/text/pupil/common.py +1121 -1121
  108. pyxllib/text/pupil/xlalign.py +326 -326
  109. pyxllib/text/pycode.py +47 -47
  110. pyxllib/text/specialist/__init__.py +8 -8
  111. pyxllib/text/specialist/common.py +112 -112
  112. pyxllib/text/specialist/ptag.py +186 -186
  113. pyxllib/text/spellchecker.py +172 -172
  114. pyxllib/text/templates/echart_base.html +10 -10
  115. pyxllib/text/templates/highlight_code.html +16 -16
  116. pyxllib/text/templates/latex_editor.html +102 -102
  117. pyxllib/text/vbacode.py +17 -17
  118. pyxllib/text/xmllib.py +741 -747
  119. pyxllib/xl.py +42 -39
  120. pyxllib/xlcv.py +17 -17
  121. pyxllib-3.201.1.dist-info/METADATA +296 -0
  122. pyxllib-3.201.1.dist-info/RECORD +125 -0
  123. {pyxllib-0.3.197.dist-info → pyxllib-3.201.1.dist-info}/licenses/LICENSE +190 -190
  124. pyxllib/ext/old.py +0 -663
  125. pyxllib-0.3.197.dist-info/METADATA +0 -48
  126. pyxllib-0.3.197.dist-info/RECORD +0 -126
  127. {pyxllib-0.3.197.dist-info → pyxllib-3.201.1.dist-info}/WHEEL +0 -0
@@ -1,362 +1,362 @@
1
- #!/usr/bin/env python3
2
- # -*- coding: utf-8 -*-
3
- # @Author : 陈坤泽
4
- # @Email : 877362867@qq.com
5
- # @Date : 2024/11/05
6
-
7
- """
8
- 以uiautomation为核心的相关工具库
9
- """
10
-
11
- import os
12
- import sys
13
- import textwrap
14
- import time
15
- from typing import Iterable, Callable, List
16
- import subprocess
17
- import tempfile
18
-
19
- import psutil
20
- import pandas as pd
21
- from fastcore.basics import GetAttr
22
-
23
- from loguru import logger
24
- # ui组件大多是树形组织结构,auto库自带树形操作太弱。没有专业的树形结构库,能搞个毛线。
25
- from anytree import NodeMixin
26
-
27
- import ctypes
28
- from ctypes import wintypes
29
-
30
- if sys.platform == 'win32':
31
- import win32con
32
- import win32gui
33
- import win32process
34
- import win32clipboard
35
-
36
- import uiautomation as uia
37
- from uiautomation import WindowControl
38
-
39
-
40
- def __1_clipboard_utils():
41
- pass
42
-
43
-
44
- def retry_on_failure(max_retries: int = 5):
45
- """
46
- 一个装饰器,用于在失败时重试执行被装饰的函数。
47
-
48
- Args:
49
- max_retries (int): 最大重试次数。
50
-
51
- Returns:
52
- Callable: 包装后的函数。
53
- """
54
-
55
- def decorator(func: Callable):
56
- def wrapper(*args, **kwargs):
57
- for attempt in range(max_retries):
58
- try:
59
- if func(*args, **kwargs):
60
- return True
61
- except Exception as e:
62
- time.sleep(.05)
63
- print(f"Attempt {attempt + 1} failed: {e}")
64
- return False
65
-
66
- return wrapper
67
-
68
- return decorator
69
-
70
-
71
- def set_clipboard_data(fmt: int, buf: ctypes.Array) -> bool:
72
- """
73
- 将数据设置到Windows剪切板中。
74
-
75
- Args:
76
- fmt (int): 数据格式,例如 win32clipboard.CF_HDROP。
77
- buf (ctypes.Array): 要设置到剪切板的数据。
78
-
79
- Returns:
80
- bool: 操作成功返回 True,否则返回 False。
81
- """
82
- try:
83
- win32clipboard.OpenClipboard()
84
- win32clipboard.EmptyClipboard()
85
- win32clipboard.SetClipboardData(fmt, buf)
86
- return True
87
- except Exception as e:
88
- print(f"Error setting clipboard data: {e}")
89
- return False
90
- finally:
91
- win32clipboard.CloseClipboard()
92
-
93
-
94
- def get_clipboard_files() -> List[str]:
95
- """
96
- 获取剪切板中的文件路径列表。
97
-
98
- Returns:
99
- List[str]: 包含剪切板中文件路径的列表,如果没有文件路径或操作失败,返回空列表。
100
- """
101
- try:
102
- win32clipboard.OpenClipboard()
103
- if win32clipboard.IsClipboardFormatAvailable(win32clipboard.CF_HDROP):
104
- return list(win32clipboard.GetClipboardData(win32clipboard.CF_HDROP))
105
- else:
106
- return list()
107
- finally:
108
- win32clipboard.CloseClipboard()
109
-
110
-
111
- @retry_on_failure(max_retries=5)
112
- def validate_clipboard_files(file_paths: Iterable[str], fmt: int, buf: ctypes.Array) -> bool:
113
- """
114
- 验证剪切板中的文件路径是否与给定的文件路径一致。
115
-
116
- Args:
117
- file_paths (Iterable): 一个包含文件路径的可迭代对象,每个路径都是一个字符串。
118
- fmt (int): 数据格式,例如 win32clipboard.CF_HDROP。
119
- buf (ctypes.Array): 要验证的剪切板数据。
120
-
121
- Returns:
122
- bool: 如果剪切板中的文件路径与给定的文件路径一致,则返回 True
123
-
124
- Raises:
125
- ValueError: 如果剪切板文件路径与给定文件路径不一致。
126
- """
127
- # 设置文件到剪切板
128
- set_clipboard_data(fmt, buf)
129
- # 验证剪切板中的文件路径是否与给定的文件路径一致
130
- if set(get_clipboard_files()) == set(file_paths):
131
- return True
132
- raise ValueError("剪切板文件路径不对哇!")
133
-
134
-
135
- def copy_files_to_clipboard(file_paths: Iterable[str]) -> bool:
136
- """
137
- 将一系列文件路径复制到Windows剪切板。这允许用户在其他应用程序中,如文件资源管理器中粘贴这些文件。
138
-
139
- Args:
140
- file_paths (Iterable): 一个包含文件路径的可迭代对象,每个路径都是一个字符串。
141
-
142
- Returns:
143
- bool: 如果成功将文件路径复制到剪切板,则返回 True,否则返回 False
144
- """
145
- # 定义所需的 Windows 结构和函数
146
- CF_HDROP = 15
147
-
148
- class DROPFILES(ctypes.Structure):
149
- _fields_ = [("pFiles", wintypes.DWORD),
150
- ("pt", wintypes.POINT),
151
- ("fNC", wintypes.BOOL),
152
- ("fWide", wintypes.BOOL)]
153
-
154
- offset = ctypes.sizeof(DROPFILES)
155
- length = sum(len(p) + 1 for p in file_paths) + 1
156
- size = offset + length * ctypes.sizeof(wintypes.WCHAR)
157
- buf = (ctypes.c_char * size)()
158
- df = DROPFILES.from_buffer(buf)
159
- df.pFiles, df.fWide = offset, True
160
- for path in file_paths:
161
- path = os.path.normpath(path)
162
- array_t = ctypes.c_wchar * (len(path) + 1)
163
- path_buf = array_t.from_buffer(buf, offset)
164
- path_buf.value = path
165
- offset += ctypes.sizeof(path_buf)
166
- buf[offset:offset + ctypes.sizeof(wintypes.WCHAR)] = b'\0\0'
167
-
168
- # 验证文件是否成功复制到剪切板
169
- return validate_clipboard_files([os.path.normpath(file) for file in file_paths], CF_HDROP, buf=buf)
170
-
171
-
172
- def __2_窗口功能():
173
- pass
174
-
175
-
176
- def get_windows_info():
177
- """ 得到当前机器的全部窗口信息清单 """
178
- window_list = []
179
-
180
- def get_all_hwnd(hwnd, mouse):
181
- thread_id, process_id = win32process.GetWindowThreadProcessId(hwnd)
182
- proc = psutil.Process(process_id)
183
-
184
- is_window = win32gui.IsWindow(hwnd)
185
- is_enabled = win32gui.IsWindowEnabled(hwnd)
186
- is_visible = win32gui.IsWindowVisible(hwnd)
187
- text = win32gui.GetWindowText(hwnd)
188
-
189
- data = {
190
- 'proc_name': proc.name(),
191
- 'process_id': process_id,
192
- 'thread_id': thread_id,
193
- 'hwnd': hwnd,
194
- 'ClassName': win32gui.GetClassName(hwnd),
195
- 'ControlTypeName': '',
196
- 'WindowText': text,
197
- 'IsWindow': is_window,
198
- 'IsWindowEnabled': is_enabled,
199
- 'IsWindowVisible': is_visible
200
- }
201
-
202
- if not data['proc_name'].endswith('.tmp') and is_visible:
203
- ctrl = uia.ControlFromHandle(hwnd)
204
- data['ControlTypeName'] = ctrl.ControlTypeName
205
-
206
- window_list.append(data)
207
-
208
- win32gui.EnumWindows(get_all_hwnd, 0)
209
- return pd.DataFrame(window_list)
210
-
211
-
212
- def find_ctrl(class_name=None, name=None, **kwargs):
213
- if class_name is not None:
214
- kwargs['ClassName'] = class_name
215
- if name is not None:
216
- kwargs['Name'] = name
217
- ctrl = uia.WindowControl(**kwargs)
218
- return ctrl
219
-
220
-
221
- class UiCtrlNode(NodeMixin, GetAttr, WindowControl):
222
- _default = 'ctrl'
223
-
224
- def __0_构建(self):
225
- pass
226
-
227
- def __init__(self, ctrl, parent=None, *, build_depth=-1):
228
- """
229
- :param ctrl: 当前节点
230
- :param parent: 父结点
231
- :param build_depth: 自动构建多少层树节点,默认-1表示构建全部节点
232
- """
233
- # 初始化节点信息
234
- self.ctrl = ctrl
235
- # 试过了,没用,因为新找出来的都是新构建的类,找不到proxy的
236
- # self.ctrl.proxy: 'UiCtrlNode' = self # 再给其扩展一个proxy属性,指向其升级过的对象
237
- self.ctrl_type = ctrl.ControlTypeName
238
- self.text = ctrl.Name
239
- self.parent = parent # 指定父节点,用于形成树结构
240
-
241
- # 自动递归创建子节点
242
- self.build_children(build_depth)
243
-
244
- @classmethod
245
- def init_from_name(cls, class_name=None, name=None, *, build_depth=-1, **kwargs):
246
- ctrl = find_ctrl(class_name=class_name, name=name, **kwargs)
247
- return cls(ctrl, build_depth=build_depth)
248
-
249
- def activate(self, check_seconds=2):
250
- """ 激活当前窗口
251
- """
252
- while True:
253
- # todo 这种限定情况并不严谨,有概率出现重复的~
254
- hwnd = win32gui.FindWindow(self.ctrl.ClassName, self.text)
255
- # logger.info(hwnd)
256
- if not hwnd:
257
- return
258
-
259
- if win32gui.GetForegroundWindow() != hwnd:
260
- win32gui.ShowWindow(hwnd, win32con.SW_RESTORE)
261
-
262
- try:
263
- # 在这里执行SetForegroundWindow,只有程序的第1次运行有效,之后就会被很多全屏类的应用占用最前置,覆盖不了了
264
- # 为了解决这问题,就只能暴力每次都新开一个程序来执行这个SetForegroundWindow操作
265
- subprocess.run(
266
- [sys.executable, "-c", f"import win32gui; win32gui.SetForegroundWindow({hwnd})"],
267
- stdout=subprocess.PIPE,
268
- )
269
- except Exception as e:
270
- pass
271
- # 理论上并不需要等待,但加个等待,有助于稳定性检测,如果当前窗口在check_seconds秒内频繁切换,
272
- # 使用activate虽然激活了,但并不安全,只有check_seconds秒内维持稳定在这个窗口,再进行下游任务会更好
273
- time.sleep(check_seconds)
274
- else:
275
- break
276
-
277
- def build_children(self, build_depth=-1, child_node_class=None):
278
- """ 创建并添加子节点到树中 """
279
- if build_depth == 0:
280
- return
281
- self.children = [] # 删除现有的所有子结点
282
- child_node_class = child_node_class or self.__class__
283
- for child_ctrl in self.ctrl.GetChildren():
284
- child_node_class(child_ctrl, parent=self, build_depth=build_depth - 1)
285
-
286
- def __1_调试(self):
287
- pass
288
-
289
- def trace_rect(self, duration_per_circle=2, num_circles=1):
290
- """ 用鼠标勾画出当前组件的矩形位置区域 """
291
- from pyxllib.autogui.autogui import UiTracePath
292
-
293
- rect = self.ctrl.BoundingRectangle
294
- ltrb = [rect.left, rect.top, rect.right, rect.bottom]
295
- UiTracePath.from_ltrb(ltrb,
296
- duration_per_circle=duration_per_circle,
297
- num_circles=num_circles)
298
-
299
- def __2_功能(self):
300
- pass
301
-
302
- def __getattr__(self, item):
303
- # 尝试从self.ctrl中获取属性
304
- return getattr(self.ctrl, item)
305
-
306
- def __getitem__(self, index):
307
- """ 通过索引直接访问子节点
308
-
309
- ui操作经常要各种结构化的访问,加个这个简化引用方式
310
- """
311
- try:
312
- return self.children[index]
313
- except IndexError: # 如果出现下标错误,需要自动重新刷新所有控件
314
- # self.parent重建是不够的,但我也不知道为什么self.root重建后就可以了
315
- # 我的理解是重建后self自己不是都不存在了?
316
- self.root.build_children()
317
- # 应该在有些情况下self.root重建还能继续使用,但有些特殊情况应该会炸
318
- return self.children[index]
319
-
320
- def get_ctrl_hash_tag(self, level=1):
321
- """ 生成节点的哈希字符串,以反映子树结构,一般用来对节点做分类及映射到对应处理函数 """
322
- # 当前节点的类型标识符
323
- hash_strs = [f"{level}{self.ctrl_type[0].lower()}"]
324
- # 遍历所有子节点,递归生成子节点的哈希值
325
- for child in self.children:
326
- hash_strs.append(f"{child.get_ctrl_hash_tag(level + 1)}")
327
- return ''.join(hash_strs)
328
-
329
- def __3_展示(self):
330
- pass
331
-
332
- def _format_text(self, text):
333
- """ 将换行替换为空格的小工具方法 """
334
- return text.replace('\n', ' ')
335
-
336
- def __repr__(self):
337
- """ 用于在打印节点时显示关键信息 """
338
- return f"UiNode(ctrl_type={self.ctrl_type}, text={self._format_text(self.text)})"
339
-
340
- def render_tree(self):
341
- """ 展示以self为根节点的整体内容结构 """
342
- # 1 渲染自身
343
- line = [self.ctrl_type]
344
- line.append(self._format_text(self.text))
345
-
346
- # 加上控件的坐标信息
347
- rect = self.ctrl.BoundingRectangle
348
- line.append(f"[{rect.left}, {rect.top}, {rect.right}, {rect.bottom}]")
349
-
350
- # 我的hash值
351
- tag = self.get_ctrl_hash_tag()
352
- if len(tag) <= 64:
353
- line.append(tag)
354
-
355
- lines = [' '.join(line)]
356
-
357
- # 2 子结点情况
358
- for child in self.children:
359
- line = child.render_tree()
360
- line = textwrap.indent(line, ' ')
361
- lines.append(line)
362
- return '\n'.join(lines)
1
+ #!/usr/bin/env python3
2
+ # -*- coding: utf-8 -*-
3
+ # @Author : 陈坤泽
4
+ # @Email : 877362867@qq.com
5
+ # @Date : 2024/11/05
6
+
7
+ """
8
+ 以uiautomation为核心的相关工具库
9
+ """
10
+
11
+ import os
12
+ import sys
13
+ import textwrap
14
+ import time
15
+ from typing import Iterable, Callable, List
16
+ import subprocess
17
+ import tempfile
18
+
19
+ import psutil
20
+ import pandas as pd
21
+ from fastcore.basics import GetAttr
22
+
23
+ from loguru import logger
24
+ # ui组件大多是树形组织结构,auto库自带树形操作太弱。没有专业的树形结构库,能搞个毛线。
25
+ from anytree import NodeMixin
26
+
27
+ import ctypes
28
+ from ctypes import wintypes
29
+
30
+ if sys.platform == 'win32':
31
+ import win32con
32
+ import win32gui
33
+ import win32process
34
+ import win32clipboard
35
+
36
+ import uiautomation as uia
37
+ from uiautomation import WindowControl
38
+
39
+
40
+ def __1_clipboard_utils():
41
+ pass
42
+
43
+
44
+ def retry_on_failure(max_retries: int = 5):
45
+ """
46
+ 一个装饰器,用于在失败时重试执行被装饰的函数。
47
+
48
+ Args:
49
+ max_retries (int): 最大重试次数。
50
+
51
+ Returns:
52
+ Callable: 包装后的函数。
53
+ """
54
+
55
+ def decorator(func: Callable):
56
+ def wrapper(*args, **kwargs):
57
+ for attempt in range(max_retries):
58
+ try:
59
+ if func(*args, **kwargs):
60
+ return True
61
+ except Exception as e:
62
+ time.sleep(.05)
63
+ print(f"Attempt {attempt + 1} failed: {e}")
64
+ return False
65
+
66
+ return wrapper
67
+
68
+ return decorator
69
+
70
+
71
+ def set_clipboard_data(fmt: int, buf: ctypes.Array) -> bool:
72
+ """
73
+ 将数据设置到Windows剪切板中。
74
+
75
+ Args:
76
+ fmt (int): 数据格式,例如 win32clipboard.CF_HDROP。
77
+ buf (ctypes.Array): 要设置到剪切板的数据。
78
+
79
+ Returns:
80
+ bool: 操作成功返回 True,否则返回 False。
81
+ """
82
+ try:
83
+ win32clipboard.OpenClipboard()
84
+ win32clipboard.EmptyClipboard()
85
+ win32clipboard.SetClipboardData(fmt, buf)
86
+ return True
87
+ except Exception as e:
88
+ print(f"Error setting clipboard data: {e}")
89
+ return False
90
+ finally:
91
+ win32clipboard.CloseClipboard()
92
+
93
+
94
+ def get_clipboard_files() -> List[str]:
95
+ """
96
+ 获取剪切板中的文件路径列表。
97
+
98
+ Returns:
99
+ List[str]: 包含剪切板中文件路径的列表,如果没有文件路径或操作失败,返回空列表。
100
+ """
101
+ try:
102
+ win32clipboard.OpenClipboard()
103
+ if win32clipboard.IsClipboardFormatAvailable(win32clipboard.CF_HDROP):
104
+ return list(win32clipboard.GetClipboardData(win32clipboard.CF_HDROP))
105
+ else:
106
+ return list()
107
+ finally:
108
+ win32clipboard.CloseClipboard()
109
+
110
+
111
+ @retry_on_failure(max_retries=5)
112
+ def validate_clipboard_files(file_paths: Iterable[str], fmt: int, buf: ctypes.Array) -> bool:
113
+ """
114
+ 验证剪切板中的文件路径是否与给定的文件路径一致。
115
+
116
+ Args:
117
+ file_paths (Iterable): 一个包含文件路径的可迭代对象,每个路径都是一个字符串。
118
+ fmt (int): 数据格式,例如 win32clipboard.CF_HDROP。
119
+ buf (ctypes.Array): 要验证的剪切板数据。
120
+
121
+ Returns:
122
+ bool: 如果剪切板中的文件路径与给定的文件路径一致,则返回 True
123
+
124
+ Raises:
125
+ ValueError: 如果剪切板文件路径与给定文件路径不一致。
126
+ """
127
+ # 设置文件到剪切板
128
+ set_clipboard_data(fmt, buf)
129
+ # 验证剪切板中的文件路径是否与给定的文件路径一致
130
+ if set(get_clipboard_files()) == set(file_paths):
131
+ return True
132
+ raise ValueError("剪切板文件路径不对哇!")
133
+
134
+
135
+ def copy_files_to_clipboard(file_paths: Iterable[str]) -> bool:
136
+ """
137
+ 将一系列文件路径复制到Windows剪切板。这允许用户在其他应用程序中,如文件资源管理器中粘贴这些文件。
138
+
139
+ Args:
140
+ file_paths (Iterable): 一个包含文件路径的可迭代对象,每个路径都是一个字符串。
141
+
142
+ Returns:
143
+ bool: 如果成功将文件路径复制到剪切板,则返回 True,否则返回 False
144
+ """
145
+ # 定义所需的 Windows 结构和函数
146
+ CF_HDROP = 15
147
+
148
+ class DROPFILES(ctypes.Structure):
149
+ _fields_ = [("pFiles", wintypes.DWORD),
150
+ ("pt", wintypes.POINT),
151
+ ("fNC", wintypes.BOOL),
152
+ ("fWide", wintypes.BOOL)]
153
+
154
+ offset = ctypes.sizeof(DROPFILES)
155
+ length = sum(len(p) + 1 for p in file_paths) + 1
156
+ size = offset + length * ctypes.sizeof(wintypes.WCHAR)
157
+ buf = (ctypes.c_char * size)()
158
+ df = DROPFILES.from_buffer(buf)
159
+ df.pFiles, df.fWide = offset, True
160
+ for path in file_paths:
161
+ path = os.path.normpath(path)
162
+ array_t = ctypes.c_wchar * (len(path) + 1)
163
+ path_buf = array_t.from_buffer(buf, offset)
164
+ path_buf.value = path
165
+ offset += ctypes.sizeof(path_buf)
166
+ buf[offset:offset + ctypes.sizeof(wintypes.WCHAR)] = b'\0\0'
167
+
168
+ # 验证文件是否成功复制到剪切板
169
+ return validate_clipboard_files([os.path.normpath(file) for file in file_paths], CF_HDROP, buf=buf)
170
+
171
+
172
+ def __2_窗口功能():
173
+ pass
174
+
175
+
176
+ def get_windows_info():
177
+ """ 得到当前机器的全部窗口信息清单 """
178
+ window_list = []
179
+
180
+ def get_all_hwnd(hwnd, mouse):
181
+ thread_id, process_id = win32process.GetWindowThreadProcessId(hwnd)
182
+ proc = psutil.Process(process_id)
183
+
184
+ is_window = win32gui.IsWindow(hwnd)
185
+ is_enabled = win32gui.IsWindowEnabled(hwnd)
186
+ is_visible = win32gui.IsWindowVisible(hwnd)
187
+ text = win32gui.GetWindowText(hwnd)
188
+
189
+ data = {
190
+ 'proc_name': proc.name(),
191
+ 'process_id': process_id,
192
+ 'thread_id': thread_id,
193
+ 'hwnd': hwnd,
194
+ 'ClassName': win32gui.GetClassName(hwnd),
195
+ 'ControlTypeName': '',
196
+ 'WindowText': text,
197
+ 'IsWindow': is_window,
198
+ 'IsWindowEnabled': is_enabled,
199
+ 'IsWindowVisible': is_visible
200
+ }
201
+
202
+ if not data['proc_name'].endswith('.tmp') and is_visible:
203
+ ctrl = uia.ControlFromHandle(hwnd)
204
+ data['ControlTypeName'] = ctrl.ControlTypeName
205
+
206
+ window_list.append(data)
207
+
208
+ win32gui.EnumWindows(get_all_hwnd, 0)
209
+ return pd.DataFrame(window_list)
210
+
211
+
212
+ def find_ctrl(class_name=None, name=None, **kwargs):
213
+ if class_name is not None:
214
+ kwargs['ClassName'] = class_name
215
+ if name is not None:
216
+ kwargs['Name'] = name
217
+ ctrl = uia.WindowControl(**kwargs)
218
+ return ctrl
219
+
220
+
221
+ class UiCtrlNode(NodeMixin, GetAttr, WindowControl):
222
+ _default = 'ctrl'
223
+
224
+ def __0_构建(self):
225
+ pass
226
+
227
+ def __init__(self, ctrl, parent=None, *, build_depth=-1):
228
+ """
229
+ :param ctrl: 当前节点
230
+ :param parent: 父结点
231
+ :param build_depth: 自动构建多少层树节点,默认-1表示构建全部节点
232
+ """
233
+ # 初始化节点信息
234
+ self.ctrl = ctrl
235
+ # 试过了,没用,因为新找出来的都是新构建的类,找不到proxy的
236
+ # self.ctrl.proxy: 'UiCtrlNode' = self # 再给其扩展一个proxy属性,指向其升级过的对象
237
+ self.ctrl_type = ctrl.ControlTypeName
238
+ self.text = ctrl.Name
239
+ self.parent = parent # 指定父节点,用于形成树结构
240
+
241
+ # 自动递归创建子节点
242
+ self.build_children(build_depth)
243
+
244
+ @classmethod
245
+ def init_from_name(cls, class_name=None, name=None, *, build_depth=-1, **kwargs):
246
+ ctrl = find_ctrl(class_name=class_name, name=name, **kwargs)
247
+ return cls(ctrl, build_depth=build_depth)
248
+
249
+ def activate(self, check_seconds=2):
250
+ """ 激活当前窗口
251
+ """
252
+ while True:
253
+ # todo 这种限定情况并不严谨,有概率出现重复的~
254
+ hwnd = win32gui.FindWindow(self.ctrl.ClassName, self.text)
255
+ # logger.info(hwnd)
256
+ if not hwnd:
257
+ return
258
+
259
+ if win32gui.GetForegroundWindow() != hwnd:
260
+ win32gui.ShowWindow(hwnd, win32con.SW_RESTORE)
261
+
262
+ try:
263
+ # 在这里执行SetForegroundWindow,只有程序的第1次运行有效,之后就会被很多全屏类的应用占用最前置,覆盖不了了
264
+ # 为了解决这问题,就只能暴力每次都新开一个程序来执行这个SetForegroundWindow操作
265
+ subprocess.run(
266
+ [sys.executable, "-c", f"import win32gui; win32gui.SetForegroundWindow({hwnd})"],
267
+ stdout=subprocess.PIPE,
268
+ )
269
+ except Exception as e:
270
+ pass
271
+ # 理论上并不需要等待,但加个等待,有助于稳定性检测,如果当前窗口在check_seconds秒内频繁切换,
272
+ # 使用activate虽然激活了,但并不安全,只有check_seconds秒内维持稳定在这个窗口,再进行下游任务会更好
273
+ time.sleep(check_seconds)
274
+ else:
275
+ break
276
+
277
+ def build_children(self, build_depth=-1, child_node_class=None):
278
+ """ 创建并添加子节点到树中 """
279
+ if build_depth == 0:
280
+ return
281
+ self.children = [] # 删除现有的所有子结点
282
+ child_node_class = child_node_class or self.__class__
283
+ for child_ctrl in self.ctrl.GetChildren():
284
+ child_node_class(child_ctrl, parent=self, build_depth=build_depth - 1)
285
+
286
+ def __1_调试(self):
287
+ pass
288
+
289
+ def trace_rect(self, duration_per_circle=2, num_circles=1):
290
+ """ 用鼠标勾画出当前组件的矩形位置区域 """
291
+ from pyxllib.autogui.autogui import UiTracePath
292
+
293
+ rect = self.ctrl.BoundingRectangle
294
+ ltrb = [rect.left, rect.top, rect.right, rect.bottom]
295
+ UiTracePath.from_ltrb(ltrb,
296
+ duration_per_circle=duration_per_circle,
297
+ num_circles=num_circles)
298
+
299
+ def __2_功能(self):
300
+ pass
301
+
302
+ def __getattr__(self, item):
303
+ # 尝试从self.ctrl中获取属性
304
+ return getattr(self.ctrl, item)
305
+
306
+ def __getitem__(self, index):
307
+ """ 通过索引直接访问子节点
308
+
309
+ ui操作经常要各种结构化的访问,加个这个简化引用方式
310
+ """
311
+ try:
312
+ return self.children[index]
313
+ except IndexError: # 如果出现下标错误,需要自动重新刷新所有控件
314
+ # self.parent重建是不够的,但我也不知道为什么self.root重建后就可以了
315
+ # 我的理解是重建后self自己不是都不存在了?
316
+ self.root.build_children()
317
+ # 应该在有些情况下self.root重建还能继续使用,但有些特殊情况应该会炸
318
+ return self.children[index]
319
+
320
+ def get_ctrl_hash_tag(self, level=1):
321
+ """ 生成节点的哈希字符串,以反映子树结构,一般用来对节点做分类及映射到对应处理函数 """
322
+ # 当前节点的类型标识符
323
+ hash_strs = [f"{level}{self.ctrl_type[0].lower()}"]
324
+ # 遍历所有子节点,递归生成子节点的哈希值
325
+ for child in self.children:
326
+ hash_strs.append(f"{child.get_ctrl_hash_tag(level + 1)}")
327
+ return ''.join(hash_strs)
328
+
329
+ def __3_展示(self):
330
+ pass
331
+
332
+ def _format_text(self, text):
333
+ """ 将换行替换为空格的小工具方法 """
334
+ return text.replace('\n', ' ')
335
+
336
+ def __repr__(self):
337
+ """ 用于在打印节点时显示关键信息 """
338
+ return f"UiNode(ctrl_type={self.ctrl_type}, text={self._format_text(self.text)})"
339
+
340
+ def render_tree(self):
341
+ """ 展示以self为根节点的整体内容结构 """
342
+ # 1 渲染自身
343
+ line = [self.ctrl_type]
344
+ line.append(self._format_text(self.text))
345
+
346
+ # 加上控件的坐标信息
347
+ rect = self.ctrl.BoundingRectangle
348
+ line.append(f"[{rect.left}, {rect.top}, {rect.right}, {rect.bottom}]")
349
+
350
+ # 我的hash值
351
+ tag = self.get_ctrl_hash_tag()
352
+ if len(tag) <= 64:
353
+ line.append(tag)
354
+
355
+ lines = [' '.join(line)]
356
+
357
+ # 2 子结点情况
358
+ for child in self.children:
359
+ line = child.render_tree()
360
+ line = textwrap.indent(line, ' ')
361
+ lines.append(line)
362
+ return '\n'.join(lines)