|
|
import win32con
import ctypes
import ctypes.wintypes
from threading import Thread, activeCount, enumerate
from time import sleep, time
from win32com.client import Dispatch
from bs4 import BeautifulSoup
ShellWindowsCLSID = '{9BA05972-F6A8-11CF-A442-00A0C90A8F39}'
ShellWindows = Dispatch(ShellWindowsCLSID)
class Hotkey(Thread):
user32 = ctypes.windll.user32
hkey_list = {}
hkey_flags = {} # 按下
hkey_running = {} # 启停
_reg_list = {} # 待注册热键信息
def regiskey(self, hwnd=None, flagid=0, fnkey=win32con.MOD_ALT, vkey=win32con.VK_F9): # 注册热键，默认一个alt+F9
return self.user32.RegisterHotKey(hwnd, flagid, fnkey, vkey)
def get_reginfo(self):
return self._reg_list
def get_id(self, func):
self_id = None
for id in self.get_reginfo():
if self.get_reginfo()[id]["func"] == func:
self_id = id
break
if self_id:
self.hkey_running[self_id] = True
return self_id
def get_running_state(self, self_id):
if self.hkey_running.get(self_id):
return self.hkey_running[self_id]
else:
return False
def reg(self, key, func, args=None):
id = int(str(round(time() * 10))[-6:])
fnkey = key[0]
vkey = key[1]
info = {
"fnkey": fnkey,
"vkey": vkey,
"func": func,
"args": args
}
self._reg_list[id] = info
# print(info) #这里待注册的信息
sleep(0.1)
return id
def fast_reg(self, id, key=(0, win32con.VK_HOME), func=lambda: print('热键注册开始')):
if not self.regiskey(None, id, key[0], key[1]):
print("热键注册失败")
return None
self.hkey_list[id] = func
self.hkey_flags[id] = False
return id
def callback(self):
def inner(self=self):
for flag in self.hkey_flags:
self.hkey_flags[flag] = False
while True:
for id, func in self.hkey_list.items():
if self.hkey_flags[id]:
args = self._reg_list[id]["args"]
if args:
# print(args) #这里打印传入给注册函数的参数
thread_it(func, *args)
else:
thread_it(func)
self.hkey_flags[id] = False
return inner
def run(self):
for id in self._reg_list:
reg_info = self._reg_list[id]
fnkey = reg_info["fnkey"]
vkey = reg_info["vkey"]
func = reg_info["func"]
self.fast_reg(id, (fnkey, vkey), func)
fn = self.callback()
thread_it(fn) # 启动监听热键按下线程
try:
msg = ctypes.wintypes.MSG()
while True:
if self.user32.GetMessageA(ctypes.byref(msg), None, 0, 0) != 0:
if msg.message == win32con.WM_HOTKEY:
if msg.wParam in self.hkey_list:
self.hkey_flags[msg.wParam] = True
self.user32.TranslateMessage(ctypes.byref(msg))
self.user32.DispatchMessageA(ctypes.byref(msg))
finally:
for id in self.hkey_list:
self.user32.UnregisterHotKey(None, id)
def thread_it(func, *args):
t = Thread(target=func, args=args)
t.setDaemon(True)
t.start()
def jump(func, hotkey):
self_id = hotkey.get_id(func)
#ShellWindows = Dispatch(ShellWindowsCLSID)
while hotkey.get_running_state(self_id):
getsn = StringProcess()
#print(f"{self_id : } 你正在1秒1次的跳动")
print(getsn)
sleep(1)
def stop_jump(start_id, hotkey):
hotkey.hkey_running[start_id] = False
print(f"{start_id} 即将停止")
sleep(1)
print(f'当前线程列表:{activeCount()}', enumerate())
def StringProcess():
#ShellWindowsCLSID = '{9BA05972-F6A8-11CF-A442-00A0C90A8F39}'
print("Find01")
#ShellWindows = Dispatch(ShellWindowsCLSID)
for shellwindow in ShellWindows:
if shellwindow.LocationURL == "www.baidu.com":
#print("Find")
ie = shellwindow
doc = ie.Document
content = doc.documentElement.innerHTML
soup = BeautifulSoup(content, "html.parser")
for fruits in soup.find_all('span', attrs={'class': ['nums_text']}):
#print(fruits.text)
return fruits.text
break
break
return -1
def main():
#ShellWindowsCLSID = '{9BA05972-F6A8-11CF-A442-00A0C90A8F39}'
#ShellWindows = Dispatch(ShellWindowsCLSID)
hotkey = Hotkey()
start_id = hotkey.reg(key=(win32con.MOD_ALT, win32con.VK_HOME), func=jump, args=(jump, hotkey)) # alt home键 开始
hotkey.reg(key=(0, win32con.VK_END), func=stop_jump, args=(start_id, hotkey)) # alt end键 结束
hotkey.start() # 启动热键主线程
print(f"当前总线程数量:{activeCount()}")
print('当前线程列表:', enumerate())
print('热键注册初始化完毕，尝试按组合键alt+Home 或者单键END看效果')
if __name__ == '__main__':
main()
--------------pyCharm运行后的错误提示------------
当前总线程数量:2
当前线程列表: [<_MainThread(MainThread, started 9432)>, <Hotkey(Thread-1, started 11092)>]
热键注册初始化完毕，尝试按组合键alt+Home 或者单键END看效果
Find01
Exception in thread Thread-3:
Traceback (most recent call last):
File "D:\Program Files\Python37\lib\threading.py", line 926, in _bootstrap_inner
self.run()
File "D:\Program Files\Python37\lib\threading.py", line 870, in run
self._target(*self._args, **self._kwargs)
File "D:/PycharmProjects/GetSn_20210317/GetSn_03.py", line 117, in jump
getsn = StringProcess()
File "D:/PycharmProjects/GetSn_20210317/GetSn_03.py", line 133, in StringProcess
for shellwindow in ShellWindows:
File "D:\PycharmProjects\GetSn_20210317\venv\lib\site-packages\win32com\client\dynamic.py", line 257, in __getitem__
raise TypeError("This object does not support enumeration")
TypeError: This object does not support enumeration
|
|
