自制简单按键助手

kylin121380

因实验需要延时点击，写了个简单的小助手 。

1. from PySide2.QtWidgets import QApplication, QMessageBox, QTableWidget, QFileDialog, QTableWidgetItem, QColorDialog, \
   
2.     QProgressBar, QStatusBar
   
3. from PySide2.QtUiTools import QUiLoader
   
4. from PySide2.QtCore import QFile, QUrl
   
5. from PySide2.QtGui import QIcon
   
6. from PySide2.QtWebEngineWidgets import QWebEngineView
   
7. 
   
8. import pyautogui as pa
   
9. import pyperclip as pp
   
10. import time
    
11. from threading import Thread
    
12. import os
    
13. 
    
14. class MainWin:
    
15.     def __init__(self):
    
16.         qfile = QFile('main.ui')
    
17.         qfile.open(qfile.ReadOnly)
    
18.         qfile.close()
    
19. 
    
20.         self.ui = QUiLoader().load(qfile)
    
21.         self.ui.setWindowTitle('autodesk')
    
22. 
    
23.         for each in pa.KEYBOARD_KEYS:
    
24.             if each in ['\t', '\n', '\r']:
    
25.                 self.ui.key_cbox.addItem(str(list(each)))
    
26.             else:
    
27.                 self.ui.key_cbox.addItem(each)
    
28. 
    
29.         self.command_list = []
    
30.         self.command_content = []
    
31.         self.show_list = []
    
32.         self.text_expend = ''
    
33.         self.command = ['鼠标左键', '鼠标右键', '鼠标左键双击', '按键', '按键按下', '按键松开',
    
34.                         '鼠标移动', '文本', '操作间隔时间', '操作移动时间', '鼠标左键拖拽', '鼠标左键拖拽']
    
35.         flag = 1
    
36.         # 读取保存的组合指令
    
37.         if 'command.txt' in os.listdir():
    
38.             with open('command.txt', 'r') as f:
    
39.                 for eachline in f:
    
40.                     if ':' in eachline and eachline[:-1].split(':')[0] in self.command:
    
41.                         if not flag:
    
42.                             self.command_content.append(self.text_expend[:-1])
    
43.                             self.show_list.append('文本:' + self.text_expend[:-1])
    
44.                             flag = 1
    
45. 
    
46.                         if eachline[:-1].split(':')[0] == '文本':
    
47.                             flag = 0
    
48.                             self.text_expend = ''
    
49.                             self.text_expend += eachline.split(':')[1]
    
50.                             self.command_list.append(eachline.split(':')[0])
    
51. 
    
52.                         else:
    
53.                             self.command_list.append(eachline[:-1].split(':')[0])
    
54.                             self.command_content.append(eachline[:-1].split(':')[1])
    
55.                             self.show_list.append(eachline[:-1])
    
56.                     else:
    
57.                         self.text_expend += eachline
    
58.                 if not flag:
    
59.                     self.command_content.append(self.text_expend[:-1])
    
60.                     self.show_list.append('文本:' + self.text_expend[:-1])
    
61. 
    
62.             for each in self.show_list:
    
63.                 self.ui.command_browser.append(each)  #
    
64. 
    
65.         print(self.command_list, self.command_content, self.show_list)
    
66. 
    
67.         self.dur = 0.1
    
68.         self.interval = 0.1
    
69. 
    
70.         self.ui.pos_button.clicked.connect(self.pos_button)
    
71.         self.ui.clear_button.clicked.connect(self.clear_button)
    
72.         self.ui.click_button.clicked.connect(self.click_button)
    
73.         self.ui.drag_button.clicked.connect(self.drag_button)
    
74.         self.ui.key_button.clicked.connect(self.key_button)
    
75.         self.ui.text_button.clicked.connect(self.text_button)
    
76.         self.ui.insert_button.clicked.connect(self.insert_button)
    
77.         self.ui.do_button.clicked.connect(self.do_button)
    
78.         self.ui.del_button.clicked.connect(self.del_button)
    
79.         self.ui.clear_button_1.clicked.connect(self.clear_button_1)
    
80. 
    
81.     def pos_button(self):
    
82.         time.sleep(3)
    
83.         x, y = pa.position()
    
84.         s = 'X: ' + str(x).rjust(4) + ', Y: ' + str(y).rjust(4)
    
85.         self.ui.pos_browser.append(s)
    
86. 
    
87.     def clear_button(self):
    
88.         self.ui.pos_browser.clear()
    
89. 
    
90.     def click_button(self):
    
91.         try:
    
92.             delay = float(self.ui.delay_line.text()) if self.ui.delay_line.text() else 0
    
93.             mouse = self.ui.mouse_cbox.currentText()
    
94.             click_num = int(self.ui.click_num_line.text())
    
95.             click_time = float(self.ui.click_time_line.text()) if self.ui.click_time_line.text() else 0
    
96.             x, y = self.ui.click_pos_line.text().split()
    
97.             dur = float(self.ui.click_duration_line.text()) if self.ui.click_duration_line.text() else 1
    
98.             self.ui.label_show.setText('程序进行中，请勿关闭窗口！')
    
99.             thread_click = Thread(target=self.thread_click,
    
100.                                   args=(int(x), int(y), click_num, click_time, mouse, dur, delay))
     
101.             # pa.click(int(x), int(y), int(click_num), float(click_time), button=mouse, duration=int(dur))
     
102.             thread_click.start()
     
103.         except ValueError:
     
104.             QMessageBox.critical(self.ui, '错误', '输入有误，请重新输入！')
     
105. 
     
106.     def thread_click(self, x, y, click_num, click_time, mouse, dur, delay):
     
107.         time.sleep(delay)
     
108.         pa.click(x, y, click_num, click_time, button=mouse, duration=dur)
     
109.         self.ui.label_show.setText('')
     
110.         # QMessageBox.about(self.ui, '完成', '请进行下一步操作')
     
111. 
     
112.     def drag_button(self):
     
113.         try:
     
114.             delay = float(self.ui.delay_line.text()) if self.ui.delay_line.text() else 0
     
115.             mouse = self.ui.mouse_drag_cbox.currentText()
     
116.             a, b = self.ui.drag_pos_line.text().split()
     
117.             x, y = self.ui.drag_pos_line_1.text().split()
     
118.             dur = float(self.ui.drag_duration_line.text()) if self.ui.drag_duration_line.text() else 6
     
119.             self.ui.label_show.setText('程序进行中，请勿关闭窗口！')
     
120.             thread_drag = Thread(target=self.thread_drag,
     
121.                                  args=(int(a), int(b), int(x), int(y), mouse, dur, delay))
     
122.             thread_drag.start()
     
123.         except ValueError:
     
124.             QMessageBox.critical(self.ui, '错误', '输入有误，请重新输入！')
     
125. 
     
126.     def thread_drag(self, a, b, x, y, mouse, dur, delay):
     
127.         time.sleep(delay)
     
128.         pa.moveTo(a, b)
     
129.         pa.dragTo(x, y, button=mouse, duration=dur)
     
130.         self.ui.label_show.setText('')
     
131.         # QMessageBox.about(self.ui, '完成', '请进行下一步操作')
     
132. 
     
133.     def key_button(self):
     
134.         try:
     
135.             delay = float(self.ui.delay_line.text()) if self.ui.delay_line.text() else 0
     
136.             x, y = self.ui.key_pos_line.text().split()
     
137.             dur = float(self.ui.key_duration_line.text()) if self.ui.key_duration_line.text() else 1
     
138.             k = self.ui.key_cbox.currentText()
     
139.             choice = self.ui.key_choice_cbox.currentText()
     
140.             self.ui.label_show.setText('程序进行中，请勿关闭窗口！')
     
141.             thread_key = Thread(target=self.thread_key,
     
142.                                 args=(int(x), int(y), dur, delay, k, choice))
     
143.             thread_key.start()
     
144.         except ValueError:
     
145.             QMessageBox.critical(self.ui, '错误', '输入有误，请重新输入！')
     
146. 
     
147.     def thread_key(self, x, y, dur, delay, k, choice):
     
148.         time.sleep(delay)
     
149.         pa.click(x, y, duration=dur)
     
150.         if choice == 'press':
     
151.             pa.press(k)
     
152.         elif choice == 'KeyDown':
     
153.             pa.keyDown(k)
     
154.         else:
     
155.             pa.keyUp(k)
     
156.         self.ui.label_show.setText('')
     
157.         # QMessageBox.about(self.ui, '完成', '请进行下一步操作')
     
158. 
     
159.     def text_button(self):
     
160.         try:
     
161.             delay = float(self.ui.delay_line.text()) if self.ui.delay_line.text() else 0
     
162.             x, y = self.ui.text_pos_line.text().split()
     
163.             dur = float(self.ui.text_duration_line.text()) if self.ui.text_duration_line.text() else 1
     
164.             content = self.ui.text_text.toPlainText()
     
165.             self.ui.label_show.setText('程序进行中，请勿关闭窗口！')
     
166.             thread_text = Thread(target=self.thread_text,
     
167.                                  args=(int(x), int(y), dur, delay, content))
     
168.             thread_text.start()
     
169.         except ValueError:
     
170.             QMessageBox.critical(self.ui, '错误', '输入有误，请重新输入！')
     
171. 
     
172.     def thread_text(self, x, y, dur, delay, content):
     
173.         time.sleep(delay)
     
174.         pa.click(x, y, duration=dur)
     
175.         pp.copy(content)
     
176.         pa.hotkey('ctrl', 'v')
     
177.         self.ui.label_show.setText('')
     
178.         # QMessageBox.about(self.ui, '完成', '请进行下一步操作')
     
179. 
     
180.     def insert_button(self):
     
181.         try:
     
182.             command = self.ui.com_cbox.currentText()
     
183.             content = self.ui.com_text.toPlainText()
     
184.             self.command_list.append(command)
     
185.             if command in ['鼠标左键', '鼠标右键', '鼠标左键双击']:
     
186.                 show = command + ':'
     
187.                 self.command_content.append(content)
     
188.             elif command in ['按键', '按键按下', '按键松开']:
     
189.                 show = command + ':' + self.ui.key_cbox.currentText()
     
190.                 self.command_content.append(self.ui.key_cbox.currentText())
     
191.             elif command in ['鼠标移动', '文本', '操作间隔时间', '操作移动时间', '鼠标左键拖拽', '鼠标左键拖拽']:
     
192.                 if command in ['鼠标移动', '鼠标左键拖拽', '鼠标左键拖拽']:
     
193.                     int(content.split()[0])
     
194.                     int(content.split()[1])
     
195.                     if len(content.split()) != 2:
     
196.                         raise IndexError
     
197.                 elif command in ['操作间隔时间', '操作移动时间']:
     
198.                     float(content)
     
199.                 show = command + ':' + content
     
200.                 self.command_content.append(content)
     
201.             self.ui.command_browser.append(show)
     
202.             self.show_list.append(show)
     
203.             self.ui.com_text.clear()
     
204.         except (ValueError, IndexError):
     
205.             QMessageBox.critical(self.ui, '错误', '输入有误，请重新输入！')
     
206. 
     
207.     def do_button(self):
     
208.         try:
     
209.             if not self.command_list:
     
210.                 raise ValueError
     
211.             self.ui.label_show.setText('程序进行中，请勿关闭窗口！')
     
212.             thread_do = Thread(target=self.thread_do,
     
213.                                args=())
     
214.             thread_do.start()
     
215.         except ValueError:
     
216.             QMessageBox.critical(self.ui, '错误', '请输入指令！')
     
217. 
     
218.     def thread_do(self):
     
219.         delay = float(self.ui.delay_line.text()) if self.ui.delay_line.text() else 0
     
220.         time.sleep(delay)
     
221.         for index in range(len(self.command_list)):
     
222.             if self.command_list[index] == '鼠标移动':
     
223.                 pa.moveTo(int(self.command_content[index].split()[0]), int(self.command_content[index].split()[1]))
     
224.             elif self.command_list[index] == '鼠标左键':
     
225.                 pa.click()
     
226.             elif self.command_list[index] == '鼠标右键':
     
227.                 pa.click(button='right')
     
228.             elif self.command_list[index] == '鼠标左键双击':
     
229.                 pa.doubleClick()
     
230.             elif self.command_list[index] == '鼠标左键拖拽':
     
231.                 pa.dragTo(int(self.command_content[index].split()[0]), int(self.command_content[index].split()[1]), duration=6)
     
232.             elif self.command_list[index] == '鼠标右键拖拽':
     
233.                 pa.dragTo(int(self.command_content[index].split()[0]), int(self.command_content[index].split()[1]), button='right', duration=6)
     
234.             elif self.command_list[index] == '按键':
     
235.                 pa.press(self.command_content[index])
     
236.             elif self.command_list[index] == '按键按下':
     
237.                 pa.keyDown(self.command_content[index])
     
238.             elif self.command_list[index] == '按键松开':
     
239.                 pa.keyUp(self.command_content[index])
     
240.             elif self.command_list[index] == '文本':
     
241.                 pp.copy(self.command_content[index])
     
242.                 pa.hotkey('ctrl', 'v')
     
243.             elif self.command_list[index] == '操作移动时间':
     
244.                 self.dur = int(self.command_content[index])
     
245.             elif self.command_list[index] == '操作间隔时间':
     
246.                 self.interval = int(self.command_content[index])
     
247.             time.sleep(self.interval)
     
248.         with open('command.txt', 'w') as f:
     
249.             for each in self.show_list:
     
250.                 f.write(each + '\n')
     
251.         self.ui.label_show.setText('')
     
252.         # QMessageBox.about(self.ui, '完成', '请进行下一步操作')
     
253. 
     
254.     def del_button(self):
     
255.         try:
     
256.             self.command_content.pop()
     
257.             self.command_list.pop()
     
258.             self.show_list.pop()
     
259.             self.ui.command_browser.clear()
     
260.             for each in self.show_list:
     
261.                 self.ui.command_browser.append(each)
     
262.         except IndexError:
     
263.             pass
     
264. 
     
265.     def clear_button_1(self):
     
266.         self.command_content.clear()
     
267.         self.command_list.clear()
     
268.         self.show_list.clear()
     
269.         self.ui.command_browser.clear()
     
270. 
     
271. 
     
272. app = QApplication([])
     
273. app.setWindowIcon(QIcon('ico.png'))
     
274. mainwin = MainWin()
     
275. mainwin.ui.show()
     
276. app.exec_()

复制代码

coolliqing

厉害厉害