blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

1、set_paths增加用户数据路径和缓存路径

Browse Source 
2、增加set_proxy函数
3、开始写set_comm函数，未完成
This commit is contained in:
g1879 2020-06-04 19:27:32 +08:00
parent e89a787efd
commit 3d04ab1405
1 changed files with 55 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
DrissionPage/easy_set.py
View File

@ -12,15 +12,19 @@ from DrissionPage.config import OptionsManager, DriverOptions
def set_paths(driver_path: str = None,
chrome_path: str = None,
debugger_address: str = None,
global_tmp_path: str = None,
tmp_path: str = None,
download_path: str = None,
uesr_data_path: str = None,
cache_path: str = None,
check_version: bool = True) -> None:
"""简易设置路径函数
:param driver_path: chromedriver.exe路径
:param chrome_path: chrome.exe路径
:param debugger_address: 调试浏览器地址，例：127.0.0.1:9222
:param download_path: 下载文件路径
:param global_tmp_path: 临时文件夹路径
:param tmp_path: 临时文件夹路径
:param uesr_data_path: 用户数据路径
:param cache_path: 缓存路径
:param check_version: 是否检查chromedriver和chrome是否匹配
:return: None
"""
@ -31,12 +35,36 @@ def set_paths(driver_path: str = None,
om.set_item('chrome_options', 'binary_location', chrome_path)
if debugger_address is not None:
om.set_item('chrome_options', 'debugger_address', debugger_address)
if global_tmp_path is not None:
om.set_item('paths', 'global_tmp_path', global_tmp_path)
if tmp_path is not None:
om.set_item('paths', 'global_tmp_path', tmp_path)
if download_path is not None:
experimental_options = om.get_value('chrome_options', 'experimental_options')
experimental_options['prefs']['download.default_directory'] = download_path
om.set_item('chrome_options', 'experimental_options', experimental_options)
if uesr_data_path is not None or cache_path is not None:
arguments = list(om.get_value('chrome_options', 'arguments'))
up_ok = cp_ok = False
to_remove = [] # 待删设置，检查完再一次删，免得影响列表后面的元素
for key, arg in enumerate(arguments):
if uesr_data_path is not None and '--user-data-dir' in arg:
if uesr_data_path == '':
to_remove.append(arg)
else:
arguments[key] = f'--user-data-dir={uesr_data_path}'
up_ok = True
if cache_path is not None and '--disk-cache-dir' in arg:
if cache_path == '':
to_remove.append(arg)
else:
arguments[key] = f'--disk-cache-dir={cache_path}'
cp_ok = True
for arg in to_remove:
arguments.remove(arg)
if uesr_data_path and not up_ok:
arguments.append(f'--user-data-dir={uesr_data_path}')
if cache_path and not cp_ok:
arguments.append(f'--disk-cache-dir={cache_path}')
om.set_item('chrome_options', 'arguments', arguments)
om.save()
if check_version:
check_driver_version(driver_path, chrome_path)
@ -53,6 +81,29 @@ def set_headless(on_off: bool) -> None:
do.save()
def set_proxy(proxy: str) -> None:
"""设置代理"""
do = DriverOptions()
pr_ok = False
for key, arg in enumerate(do.arguments):
if '--proxy-server' in arg:
do.remove_argument(do.arguments[key])
if proxy:
do.add_argument(f'--proxy-server={proxy}')
pr_ok = True
break
if not pr_ok:
do.add_argument(f'--proxy-server={proxy}')
do.save()
def set_comm(value:str) -> None:
if value=='no_img':
pass
elif value=='mute':
pass
elif value=='':
pass
def check_driver_version(driver_path: str = None, chrome_path: str = None) -> bool:
"""检查传入的chrome和chromedriver是否匹配"""
print('正在检测可用性...')