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

优化_chrome_options_to_dict，若传入字典或None则直接返回

Browse Source
This commit is contained in:
g1879 2020-05-25 23:51:55 +08:00
parent 756e2e6e48
commit f82203ea45
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
DrissionPage/config.py
View File

@ -7,7 +7,7 @@
"""
from configparser import ConfigParser, NoSectionError, NoOptionError
from pathlib import Path
from typing import Any
from typing import Any, Union
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
@ -140,12 +140,18 @@ def _dict_to_chrome_options(options: dict) -> Options:
return chrome_options
def _chrome_options_to_dict(options: Options) -> dict:
def _chrome_options_to_dict(options: Union[dict, Options, None]) -> Union[dict, None]:
if options is None or isinstance(options, dict):
return options
re_dict = dict()
re_dict['binary_location'] = options.binary_location
re_dict['debugger_address'] = options.debugger_address
re_dict['arguments'] = options.arguments
re_dict['extensions'] = options.extensions
re_dict['experimental_options'] = options.experimental_options
# re_dict['capabilities'] = options.capabilities
if options.debugger_address:
re_dict['debugger_address'] = options.debugger_address
else:
re_dict['binary_location'] = options.binary_location
re_dict['debugger_address'] = options.debugger_address
re_dict['arguments'] = options.arguments
re_dict['extensions'] = options.extensions
re_dict['experimental_options'] = options.experimental_options
# re_dict['capabilities'] = options.capabilities
return re_dict