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

DriverOptions和OptionsManager的save()方法默认保存到当前文件，传入'default'保存到默认ini文件

Browse Source
This commit is contained in:
g1879 2020-11-20 23:21:43 +08:00
parent 96340024c5
commit 1abbf8e9a6
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
DrissionPage/config.py
View File

@ -92,11 +92,15 @@ class OptionsManager(object):
def save(self, path: str = None): def save(self, path: str = None):
"""保存配置文件 \n """保存配置文件 \n
:param path: ini文件的路径，默认保存到模块文件夹下的 :param path: ini文件的路径，传入 'default' 保存到默认ini文件
:return: 当前对象 :return: 当前对象
""" """
path = path or self.path path = Path(__file__).parent / 'configs.ini' if path == 'default' else path
path = Path(path or self.path)
path = path / 'config.ini' if path.is_dir() else path
path = path.absolute()
self._conf.write(open(path, 'w', encoding='utf-8')) self._conf.write(open(path, 'w', encoding='utf-8'))
return self return self
@ -108,6 +112,7 @@ class DriverOptions(Options):
def __init__(self, read_file: bool = True, ini_path: str = None): def __init__(self, read_file: bool = True, ini_path: str = None):
"""初始化，默认从文件读取设置 \n """初始化，默认从文件读取设置 \n
:param read_file: 是否从默认ini文件中读取配置信息 :param read_file: 是否从默认ini文件中读取配置信息
:param ini_path: ini文件路径，为None则读取默认ini文件
""" """
super().__init__() super().__init__()
self._driver_path = None self._driver_path = None
@ -136,12 +141,15 @@ class DriverOptions(Options):
def save(self, path: str = None): def save(self, path: str = None):
"""保存设置到文件 \n """保存设置到文件 \n
:param path: ini文件的路径，默认保存到模块文件夹下的 :param path: ini文件的路径，传入 'default' 保存到默认ini文件
:return: 当前对象 :return: 当前对象
""" """
om = OptionsManager() om = OptionsManager()
options = _chrome_options_to_dict(self) options = _chrome_options_to_dict(self)
path = path or self.path path = Path(__file__).parent / 'configs.ini' if path == 'default' else path
path = Path(path or self.path)
path = path / 'config.ini' if path.is_dir() else path
path = path.absolute()
for i in options: for i in options:
if i == 'driver_path': if i == 'driver_path':
@ -206,6 +214,7 @@ class DriverOptions(Options):
:param on_off: 开或关 :param on_off: 开或关
:return: 当前对象 :return: 当前对象
""" """
on_off = True if on_off else False
return self.set_argument('--headless', on_off) return self.set_argument('--headless', on_off)
def set_no_imgs(self, on_off: bool = True): def set_no_imgs(self, on_off: bool = True):
@ -213,6 +222,7 @@ class DriverOptions(Options):
:param on_off: 开或关 :param on_off: 开或关
:return: 当前对象 :return: 当前对象
""" """
on_off = True if on_off else False
return self.set_argument('--blink-settings=imagesEnabled=false', on_off) return self.set_argument('--blink-settings=imagesEnabled=false', on_off)
def set_no_js(self, on_off: bool = True): def set_no_js(self, on_off: bool = True):
@ -220,6 +230,7 @@ class DriverOptions(Options):
:param on_off: 开或关 :param on_off: 开或关
:return: 当前对象 :return: 当前对象
""" """
on_off = True if on_off else False
return self.set_argument('--disable-javascript', on_off) return self.set_argument('--disable-javascript', on_off)
def set_mute(self, on_off: bool = True): def set_mute(self, on_off: bool = True):
@ -227,6 +238,7 @@ class DriverOptions(Options):
:param on_off: 开或关 :param on_off: 开或关
:return: 当前对象 :return: 当前对象
""" """
on_off = True if on_off else False
return self.set_argument('--mute-audio', on_off) return self.set_argument('--mute-audio', on_off)
def set_user_agent(self, user_agent: str): def set_user_agent(self, user_agent: str):