mirror of
https://gitee.com/g1879/DrissionPage.git
synced 2024-12-10 04:00:23 +08:00
628 lines
22 KiB
Python
628 lines
22 KiB
Python
# -*- coding:utf-8 -*-
|
||
"""
|
||
@Author : g1879
|
||
@Contact : g1879@qq.com
|
||
"""
|
||
from pathlib import Path
|
||
from platform import system
|
||
from threading import Thread
|
||
from time import perf_counter, sleep
|
||
|
||
from DownloadKit import DownloadKit
|
||
from requests import Session
|
||
|
||
from .chromium_base import ChromiumBase, Timeout
|
||
from .chromium_driver import ChromiumDriver
|
||
from .chromium_tab import ChromiumTab
|
||
from .configs.chromium_options import ChromiumOptions
|
||
from .configs.driver_options import DriverOptions
|
||
from .functions.browser import connect_browser
|
||
from .functions.web import set_session_cookies
|
||
from .session_page import DownloadSetter
|
||
|
||
|
||
class ChromiumPage(ChromiumBase):
|
||
"""用于管理浏览器的类"""
|
||
|
||
def __init__(self, addr_driver_opts=None, tab_id=None, timeout=None):
|
||
"""
|
||
:param addr_driver_opts: 浏览器地址:端口、ChromiumDriver对象或ChromiumOptions对象
|
||
:param tab_id: 要控制的标签页id，不指定默认为激活的
|
||
:param timeout: 超时时间
|
||
"""
|
||
super().__init__(addr_driver_opts, tab_id, timeout)
|
||
self._session = None
|
||
self._download_set = None
|
||
self._download_kit = None
|
||
|
||
def _connect_browser(self, addr_driver_opts=None, tab_id=None):
|
||
"""连接浏览器，在第一次时运行
|
||
:param addr_driver_opts: 浏览器地址、ChromiumDriver对象或DriverOptions对象
|
||
:param tab_id: 要控制的标签页id，不指定默认为激活的
|
||
:return: None
|
||
"""
|
||
# 接管或启动浏览器
|
||
self._chromium_init()
|
||
if addr_driver_opts is None or isinstance(addr_driver_opts, (ChromiumOptions, DriverOptions)):
|
||
self._driver_options = addr_driver_opts or ChromiumOptions() # 从ini文件读取
|
||
self.address = self._driver_options.debugger_address
|
||
self.process = connect_browser(self._driver_options)[1]
|
||
json = self._control_session.get(f'http://{self.address}/json').json()
|
||
tab_id = [i['id'] for i in json if i['type'] == 'page'][0]
|
||
|
||
# 接收浏览器地址和端口
|
||
elif isinstance(addr_driver_opts, str):
|
||
self.address = addr_driver_opts
|
||
self._driver_options = ChromiumOptions(read_file=False)
|
||
self._driver_options.debugger_address = addr_driver_opts
|
||
self.process = connect_browser(self._driver_options)[1]
|
||
if not tab_id:
|
||
json = self._control_session.get(f'http://{self.address}/json').json()
|
||
tab_id = [i['id'] for i in json if i['type'] == 'page'][0]
|
||
|
||
# 接收传递过来的ChromiumDriver，浏览器
|
||
elif isinstance(addr_driver_opts, ChromiumDriver):
|
||
self._tab_obj = addr_driver_opts
|
||
self.address = addr_driver_opts.address
|
||
self.process = None
|
||
self._driver_options = ChromiumOptions(read_file=False)
|
||
self._driver_options.debugger_address = addr_driver_opts.address
|
||
|
||
else:
|
||
raise TypeError('只能接收ChromiumDriver或ChromiumOptions类型参数。')
|
||
|
||
self._set_options()
|
||
self._set_chromium_options()
|
||
self._init_page(tab_id)
|
||
self._get_document()
|
||
self._first_run = False
|
||
|
||
def _set_options(self):
|
||
"""设置WebPage中与s模式共用的配置，便于WebPage覆盖掉"""
|
||
self._timeouts = Timeout(self,
|
||
page_load=self._driver_options.timeouts['pageLoad'],
|
||
script=self._driver_options.timeouts['script'],
|
||
implicit=self._driver_options.timeouts['implicit'])
|
||
self._page_load_strategy = self._driver_options.page_load_strategy
|
||
|
||
def _set_chromium_options(self):
|
||
"""设置浏览器专有的配置"""
|
||
self._alert = Alert()
|
||
self._window_setter = None
|
||
|
||
def _init_page(self, tab_id=None):
|
||
"""新建页面、页面刷新、切换标签页后要进行的cdp参数初始化
|
||
:param tab_id: 要跳转到的标签页id
|
||
:return: None
|
||
"""
|
||
super()._init_page(tab_id)
|
||
self._tab_obj.Page.javascriptDialogOpening = self._on_alert_open
|
||
self._tab_obj.Page.javascriptDialogClosed = self._on_alert_close
|
||
self._main_tab = self.tab_id
|
||
self.download_set.use_DownloadKit()
|
||
# try:
|
||
# self.run_cdp('Browser.setDownloadBehavior', behavior='allow',
|
||
# downloadPath=self._driver_options.download_path, not_change=True)
|
||
# except:
|
||
# self.run_cdp('Page.setDownloadBehavior', behavior='allow',
|
||
# downloadPath=self._driver_options.download_path, not_change=True)
|
||
|
||
@property
|
||
def tabs_count(self):
|
||
"""返回标签页数量"""
|
||
return len(self.tabs)
|
||
|
||
@property
|
||
def tabs(self):
|
||
"""返回所有标签页id组成的列表"""
|
||
j = self._control_session.get(f'http://{self.address}/json').json()
|
||
return [i['id'] for i in j if i['type'] == 'page']
|
||
|
||
@property
|
||
def main_tab(self):
|
||
return self._main_tab
|
||
|
||
@property
|
||
def process_id(self):
|
||
"""返回浏览器进程id"""
|
||
try:
|
||
return self._driver.SystemInfo.getProcessInfo()['id']
|
||
except Exception:
|
||
return None
|
||
|
||
@property
|
||
def set_window(self):
|
||
"""返回用于设置窗口大小的对象"""
|
||
if self._window_setter is None:
|
||
self._window_setter = WindowSetter(self)
|
||
return self._window_setter
|
||
|
||
@property
|
||
def download_path(self):
|
||
"""返回默认下载路径"""
|
||
p = self._download_path or ''
|
||
return str(Path(p).absolute())
|
||
|
||
@property
|
||
def download_set(self):
|
||
"""返回用于设置下载参数的对象"""
|
||
if self._download_set is None:
|
||
self._download_set = ChromiumDownloadSetter(self)
|
||
return self._download_set
|
||
|
||
@property
|
||
def download(self):
|
||
"""返回下载器对象"""
|
||
self.cookies_to_session()
|
||
if self._download_kit is None:
|
||
self._download_kit = DownloadKit(session=self._session, goal_path=self.download_path)
|
||
return self._download_kit
|
||
|
||
def get_tab(self, tab_id=None):
|
||
"""获取一个标签页对象
|
||
:param tab_id: 要获取的标签页id，为None时获取当前tab
|
||
:return: 标签页对象
|
||
"""
|
||
tab_id = tab_id or self.tab_id
|
||
return ChromiumTab(self, tab_id)
|
||
|
||
def get_screenshot(self, path=None, as_bytes=None, full_page=False, left_top=None, right_bottom=None):
|
||
"""对页面进行截图，可对整个网页、可见网页、指定范围截图。对可视范围外截图需要90以上版本浏览器支持
|
||
:param path: 完整路径，后缀可选 'jpg','jpeg','png','webp'
|
||
:param as_bytes: 是否已字节形式返回图片，可选 'jpg','jpeg','png','webp'，生效时path参数无效
|
||
:param full_page: 是否整页截图，为True截取整个网页，为False截取可视窗口
|
||
:param left_top: 截取范围左上角坐标
|
||
:param right_bottom: 截取范围右下角角坐标
|
||
:return: 图片完整路径或字节文本
|
||
"""
|
||
if as_bytes:
|
||
if as_bytes is True:
|
||
pic_type = 'png'
|
||
else:
|
||
if as_bytes not in ('jpg', 'jpeg', 'png', 'webp'):
|
||
raise ValueError("只能接收'jpg', 'jpeg', 'png', 'webp'四种格式。")
|
||
pic_type = 'jpeg' if as_bytes == 'jpg' else as_bytes
|
||
|
||
else:
|
||
if not path:
|
||
raise ValueError('保存为文件时必须传入路径。')
|
||
path = Path(path)
|
||
pic_type = path.suffix.lower()
|
||
if pic_type not in ('.jpg', '.jpeg', '.png', '.webp'):
|
||
raise TypeError(f'不支持的文件格式：{pic_type}。')
|
||
pic_type = 'jpeg' if pic_type == '.jpg' else pic_type[1:]
|
||
|
||
width, height = self.size
|
||
if full_page:
|
||
vp = {'x': 0, 'y': 0, 'width': width, 'height': height, 'scale': 1}
|
||
png = self._wait_driver.Page.captureScreenshot(format=pic_type, captureBeyondViewport=True, clip=vp)['data']
|
||
else:
|
||
if left_top and right_bottom:
|
||
x, y = left_top
|
||
w = right_bottom[0] - x
|
||
h = right_bottom[1] - y
|
||
vp = {'x': x, 'y': y, 'width': w, 'height': h, 'scale': 1}
|
||
png = self._wait_driver.Page.captureScreenshot(format=pic_type, captureBeyondViewport=True, clip=vp)[
|
||
'data']
|
||
else:
|
||
png = self._wait_driver.Page.captureScreenshot(format=pic_type)['data']
|
||
|
||
from base64 import b64decode
|
||
png = b64decode(png)
|
||
|
||
if as_bytes:
|
||
return png
|
||
|
||
path.parent.mkdir(parents=True, exist_ok=True)
|
||
with open(path, 'wb') as f:
|
||
f.write(png)
|
||
return str(path.absolute())
|
||
|
||
def to_front(self):
|
||
"""激活当前标签页使其处于最前面"""
|
||
self._control_session.get(f'http://{self.address}/json/activate/{self.tab_id}')
|
||
|
||
def new_tab(self, url=None, switch_to=True):
|
||
"""新建一个标签页,该标签页在最后面
|
||
:param url: 新标签页跳转到的网址
|
||
:param switch_to: 新建标签页后是否把焦点移过去
|
||
:return: None
|
||
"""
|
||
if switch_to:
|
||
begin_tabs = set(self.tabs)
|
||
len_tabs = len(begin_tabs)
|
||
self.run_cdp('Target.createTarget', url='')
|
||
|
||
tabs = self.tabs
|
||
while len(tabs) == len_tabs:
|
||
tabs = self.tabs
|
||
sleep(.005)
|
||
|
||
new_tab = set(tabs) - begin_tabs
|
||
self._to_tab(new_tab.pop(), read_doc=False)
|
||
if url:
|
||
self.get(url)
|
||
|
||
elif url:
|
||
self.run_cdp('Target.createTarget', url=url)
|
||
|
||
else:
|
||
self.run_cdp('Target.createTarget', url='')
|
||
|
||
def set_main_tab(self, tab_id=None):
|
||
"""设置主tab
|
||
:param tab_id: 标签页id，不传入则设置当前tab
|
||
:return: None
|
||
"""
|
||
self._main_tab = tab_id or self.tab_id
|
||
|
||
def to_main_tab(self):
|
||
"""跳转到主标签页"""
|
||
self.to_tab(self._main_tab)
|
||
|
||
def to_tab(self, tab_id=None, activate=True):
|
||
"""跳转到标签页
|
||
:param tab_id: 标签页id字符串，默认跳转到main_tab
|
||
:param activate: 切换后是否变为活动状态
|
||
:return: None
|
||
"""
|
||
self._to_tab(tab_id, activate)
|
||
|
||
def _to_tab(self, tab_id=None, activate=True, read_doc=True):
|
||
"""跳转到标签页
|
||
:param tab_id: 标签页id字符串，默认跳转到main_tab
|
||
:param activate: 切换后是否变为活动状态
|
||
:param read_doc: 切换后是否读取文档
|
||
:return: None
|
||
"""
|
||
tabs = self.tabs
|
||
if not tab_id:
|
||
tab_id = self._main_tab
|
||
if tab_id not in tabs:
|
||
tab_id = tabs[0]
|
||
|
||
if activate:
|
||
self._control_session.get(f'http://{self.address}/json/activate/{tab_id}')
|
||
|
||
if tab_id == self.tab_id:
|
||
return
|
||
|
||
self._driver.stop()
|
||
self._init_page(tab_id)
|
||
if read_doc and self.ready_state == 'complete':
|
||
self._get_document()
|
||
|
||
def close_tabs(self, tab_ids=None, others=False):
|
||
"""关闭传入的标签页，默认关闭当前页。可传入多个
|
||
:param tab_ids: 要关闭的标签页id，可传入id组成的列表或元组，为None时关闭当前页
|
||
:param others: 是否关闭指定标签页之外的
|
||
:return: None
|
||
"""
|
||
all_tabs = set(self.tabs)
|
||
if isinstance(tab_ids, str):
|
||
tabs = {tab_ids}
|
||
elif tab_ids is None:
|
||
tabs = {self.tab_id}
|
||
else:
|
||
tabs = set(tab_ids)
|
||
|
||
if others:
|
||
tabs = all_tabs - tabs
|
||
|
||
end_len = len(all_tabs) - len(tabs)
|
||
if end_len <= 0:
|
||
self.quit()
|
||
return
|
||
|
||
if self.tab_id in tabs:
|
||
self._driver.stop()
|
||
|
||
for tab in tabs:
|
||
self._control_session.get(f'http://{self.address}/json/close/{tab}')
|
||
while len(self.tabs) != end_len:
|
||
sleep(.1)
|
||
|
||
if self._main_tab in tabs:
|
||
self._main_tab = self.tabs[0]
|
||
|
||
self.to_tab()
|
||
|
||
def close_other_tabs(self, tab_ids=None):
|
||
"""关闭传入的标签页以外标签页，默认保留当前页。可传入多个
|
||
:param tab_ids: 要保留的标签页id，可传入id组成的列表或元组，为None时保存当前页
|
||
:return: None
|
||
"""
|
||
self.close_tabs(tab_ids, True)
|
||
|
||
def handle_alert(self, accept=True, send=None, timeout=None):
|
||
"""处理提示框，可以自动等待提示框出现
|
||
:param accept: True表示确认，False表示取消，其它值不会按按钮但依然返回文本值
|
||
:param send: 处理prompt提示框时可输入文本
|
||
:param timeout: 等待提示框出现的超时时间，为None则使用self.timeout属性的值
|
||
:return: 提示框内容文本，未等到提示框则返回None
|
||
"""
|
||
timeout = timeout or self.timeout
|
||
end_time = perf_counter() + timeout
|
||
while not self._alert.activated and perf_counter() < end_time:
|
||
sleep(.1)
|
||
if not self._alert.activated:
|
||
return None
|
||
|
||
res_text = self._alert.text
|
||
if self._alert.type == 'prompt':
|
||
self._driver.Page.handleJavaScriptDialog(accept=accept, promptText=send)
|
||
else:
|
||
self._driver.Page.handleJavaScriptDialog(accept=accept)
|
||
return res_text
|
||
|
||
def hide_browser(self):
|
||
"""隐藏浏览器窗口，只在Windows系统可用"""
|
||
show_or_hide_browser(self, hide=True)
|
||
|
||
def show_browser(self):
|
||
"""显示浏览器窗口，只在Windows系统可用"""
|
||
show_or_hide_browser(self, hide=False)
|
||
|
||
def cookies_to_session(self):
|
||
"""把driver对象的cookies复制到session对象"""
|
||
if self._session is None:
|
||
self._session = Session()
|
||
selenium_user_agent = self._tab_obj.Runtime.evaluate(expression='navigator.userAgent;')['result']['value']
|
||
self._session.headers.update({"User-Agent": selenium_user_agent})
|
||
|
||
set_session_cookies(self._session, self.get_cookies(as_dict=True))
|
||
|
||
def quit(self):
|
||
"""关闭浏览器"""
|
||
self._tab_obj.Browser.close()
|
||
self._tab_obj.stop()
|
||
|
||
def _on_alert_close(self, **kwargs):
|
||
"""alert关闭时触发的方法"""
|
||
self._alert.activated = False
|
||
self._alert.text = None
|
||
self._alert.type = None
|
||
self._alert.defaultPrompt = None
|
||
self._alert.response_accept = kwargs.get('result')
|
||
self._alert.response_text = kwargs['userInput']
|
||
self._tab_obj.has_alert = False
|
||
|
||
def _on_alert_open(self, **kwargs):
|
||
"""alert出现时触发的方法"""
|
||
self._alert.activated = True
|
||
self._alert.text = kwargs['message']
|
||
self._alert.type = kwargs['message']
|
||
self._alert.defaultPrompt = kwargs.get('defaultPrompt', None)
|
||
self._alert.response_accept = None
|
||
self._alert.response_text = None
|
||
self._tab_obj.has_alert = True
|
||
|
||
|
||
class ChromiumDownloadSetter(DownloadSetter):
|
||
"""用于设置下载参数的类"""
|
||
|
||
def __init__(self, page):
|
||
super().__init__(page)
|
||
self._behavior = 'allow'
|
||
self._download_th = None
|
||
self._split = False
|
||
self._file_exists = 'rename'
|
||
|
||
def save_path(self, path):
|
||
"""设置下载路径
|
||
:param path: 下载路径
|
||
:return: None
|
||
"""
|
||
path = path or ''
|
||
path = Path(path).absolute()
|
||
path.mkdir(parents=True, exist_ok=True)
|
||
path = str(path)
|
||
self._page._download_path = path
|
||
try:
|
||
self._page.run_cdp('Browser.setDownloadBehavior', behavior='allow', downloadPath=path, not_change=True)
|
||
except:
|
||
self._page.run_cdp('Page.setDownloadBehavior', behavior='allow', downloadPath=path, not_change=True)
|
||
|
||
if self._page._download_kit is not None:
|
||
self._page.download.goal_path = path
|
||
|
||
def use_browser(self):
|
||
"""设置使用浏览器下载文件"""
|
||
self._page.driver.Page.downloadWillBegin = None
|
||
self._page.driver.Browser.downloadWillBegin = None
|
||
self._page.driver.Browser.setDownloadBehavior(behavior='allow', downloadPath=self._page.download_path)
|
||
self._behavior = 'allow'
|
||
|
||
def use_DownloadKit(self):
|
||
"""设置使用DownloadKit下载文件"""
|
||
self._page.driver.Page.downloadWillBegin = self._download_by_DownloadKit
|
||
self._page.driver.Browser.downloadWillBegin = self._download_by_DownloadKit
|
||
self._page.driver.Browser.setDownloadBehavior(behavior='deny')
|
||
self._behavior = 'deny'
|
||
|
||
def split(self, on_off):
|
||
"""设置是否允许拆分大文件用多线程下载
|
||
:param on_off: 是否启用多线程下载大文件
|
||
:return: None
|
||
"""
|
||
self._split = on_off
|
||
|
||
@property
|
||
def if_file_exists(self):
|
||
"""返回用于设置存在同名文件时处理方法的对象"""
|
||
return FileExists(self)
|
||
|
||
def _download_by_DownloadKit(self, **kwargs):
|
||
"""拦截浏览器下载并用downloadKit下载"""
|
||
self._page.run_cdp('Browser.cancelDownload', guid=kwargs['guid'], not_change=True)
|
||
self._page.download.add(file_url=kwargs['url'], goal_path=self._page.download_path,
|
||
rename=kwargs['suggestedFilename'], split=self._split, file_exists=self._file_exists)
|
||
if self._download_th is None or not self._download_th.is_alive():
|
||
self._download_th = Thread(target=self._wait_download_complete, daemon=False)
|
||
self._download_th.start()
|
||
|
||
def _wait_download_complete(self):
|
||
"""等待下载完成"""
|
||
self._page.download.wait()
|
||
|
||
|
||
class FileExists(object):
|
||
def __init__(self, setter):
|
||
self._setter = setter
|
||
|
||
def skip(self):
|
||
self._setter._file_exists = 'skip'
|
||
|
||
def rename(self):
|
||
self._setter._file_exists = 'rename'
|
||
|
||
def overwrite(self):
|
||
self._setter._file_exists = 'overwrite'
|
||
|
||
|
||
class Alert(object):
|
||
"""用于保存alert信息的类"""
|
||
|
||
def __init__(self):
|
||
self.activated = False
|
||
self.text = None
|
||
self.type = None
|
||
self.defaultPrompt = None
|
||
self.response_accept = None
|
||
self.response_text = None
|
||
|
||
|
||
class WindowSetter(object):
|
||
"""用于设置窗口大小的类"""
|
||
|
||
def __init__(self, page):
|
||
self.driver = page.driver
|
||
self.window_id = self._get_info()['windowId']
|
||
|
||
def maximized(self):
|
||
"""窗口最大化"""
|
||
self._perform({'windowState': 'maximized'})
|
||
|
||
def minimized(self):
|
||
"""窗口最小化"""
|
||
self._perform({'windowState': 'minimized'})
|
||
|
||
def fullscreen(self):
|
||
"""设置窗口为全屏"""
|
||
self._perform({'windowState': 'fullscreen'})
|
||
|
||
def normal(self):
|
||
"""设置窗口为常规模式"""
|
||
self._perform({'windowState': 'normal'})
|
||
self._perform({'windowState': 'normal'})
|
||
|
||
def size(self, width=None, height=None):
|
||
"""设置窗口大小
|
||
:param width: 窗口宽度
|
||
:param height: 窗口高度
|
||
:return: None
|
||
"""
|
||
if width or height:
|
||
info = self._get_info()['bounds']
|
||
width = width or info['width']
|
||
height = height or info['height']
|
||
self._perform({'width': width, 'height': height})
|
||
|
||
def location(self, x=None, y=None):
|
||
"""设置窗口在屏幕中的位置，相对左上角坐标
|
||
:param x: 距离顶部距离
|
||
:param y: 距离左边距离
|
||
:return: None
|
||
"""
|
||
if x or y:
|
||
self.normal()
|
||
info = self._get_info()['bounds']
|
||
x = x or info['left']
|
||
y = y or info['top']
|
||
self._perform({'left': x, 'top': y})
|
||
|
||
def _get_info(self):
|
||
"""获取窗口位置及大小信息"""
|
||
return self.driver.Browser.getWindowForTarget()
|
||
|
||
def _perform(self, bounds):
|
||
"""执行改变窗口大小操作
|
||
:param bounds: 控制数据
|
||
:return: None
|
||
"""
|
||
self.driver.Browser.setWindowBounds(windowId=self.window_id, bounds=bounds)
|
||
|
||
|
||
def show_or_hide_browser(page, hide=True):
|
||
"""执行显示或隐藏浏览器窗口
|
||
:param page: ChromePage对象
|
||
:param hide: 是否隐藏
|
||
:return: None
|
||
"""
|
||
if not page.address.startswith(('localhost', '127.0.0.1')):
|
||
return
|
||
|
||
if system().lower() != 'windows':
|
||
raise OSError('该方法只能在Windows系统使用。')
|
||
|
||
try:
|
||
from win32gui import ShowWindow
|
||
from win32con import SW_HIDE, SW_SHOW
|
||
except ImportError:
|
||
raise ImportError('请先安装：pip install pypiwin32')
|
||
|
||
pid = page.process_id or get_browser_progress_id(page.process, page.address)
|
||
if not pid:
|
||
return None
|
||
hds = get_chrome_hwnds_from_pid(pid, page.title)
|
||
sw = SW_HIDE if hide else SW_SHOW
|
||
for hd in hds:
|
||
ShowWindow(hd, sw)
|
||
|
||
|
||
def get_browser_progress_id(progress, address):
|
||
"""获取浏览器进程id
|
||
:param progress: 已知的进程对象，没有时传入None
|
||
:param address: 浏览器管理地址，含端口
|
||
:return: 进程id或None
|
||
"""
|
||
if progress:
|
||
return progress.pid
|
||
|
||
from os import popen
|
||
port = address.split(':')[-1]
|
||
txt = ''
|
||
progresses = popen(f'netstat -nao | findstr :{port}').read().split('\n')
|
||
for progress in progresses:
|
||
if 'LISTENING' in progress:
|
||
txt = progress
|
||
break
|
||
if not txt:
|
||
return None
|
||
|
||
return txt.split(' ')[-1]
|
||
|
||
|
||
def get_chrome_hwnds_from_pid(pid, title):
|
||
"""通过PID查询句柄ID
|
||
:param pid: 进程id
|
||
:param title: 窗口标题
|
||
:return: 进程句柄组成的列表
|
||
"""
|
||
try:
|
||
from win32gui import IsWindow, GetWindowText, EnumWindows
|
||
from win32process import GetWindowThreadProcessId
|
||
except ImportError:
|
||
raise ImportError('请先安装win32gui，pip install pypiwin32')
|
||
|
||
def callback(hwnd, hds):
|
||
if IsWindow(hwnd) and title in GetWindowText(hwnd):
|
||
_, found_pid = GetWindowThreadProcessId(hwnd)
|
||
if str(found_pid) == str(pid):
|
||
hds.append(hwnd)
|
||
return True
|
||
|
||
hwnds = []
|
||
EnumWindows(callback, hwnds)
|
||
return hwnds
|