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

Drission支持接收现有的driver和session创建。可与selenium无缝连接

Browse Source
This commit is contained in:
g1879 2020-10-18 13:08:31 +08:00
parent ac90820f1e
commit 5cf0231892
1 changed files with 30 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
DrissionPage/drission.py
View File

@ -22,33 +22,46 @@ class Drission(object):
"""Drission类整合了WebDriver对象和HTLSession对象，可按要求创建、关闭及同步cookies""" """Drission类整合了WebDriver对象和HTLSession对象，可按要求创建、关闭及同步cookies"""
def __init__(self, def __init__(self,
driver_options: Union[dict, Options] = None, driver_or_options: Union[WebDriver, dict, Options] = None,
session_options: dict = None, session_or_options: Union[Session, HTMLSession, dict] = None,
ini_path: str = None, ini_path: str = None,
proxy: dict = None): proxy: dict = None):
"""初始化配置信息，但不生成session和driver实例 \n """初始化，可接收现成的WebDriver和Session对象，或接收它们的配置信息 \n
:param driver_options: chrome设置，Options类或设置字典 :param driver_or_options: driver对象或chrome设置，Options类或设置字典
:param session_options: session设置 :param session_or_options: session、HTMLSession对象或session设置
:param ini_path: ini文件路径' :param ini_path: ini文件路径
:param proxy: 代理设置 :param proxy: 代理设置
""" """
self._session = None self._session = None
self._driver = None self._driver = None
self._driver_path = 'chromedriver' self._driver_path = 'chromedriver'
self._proxy = proxy self._proxy = proxy
if session_options is None: if isinstance(session_or_options, HTMLSession):
self._session_options = OptionsManager(ini_path).get_option('session_options') self._session = session_or_options
elif isinstance(session_or_options, Session):
self._session = HTMLSession()
for key in session_or_options.__dict__: # session对象强制升级为子类HTMLSession对象
if key != 'hooks':
self._session.__dict__[key] = session_or_options.__dict__[key]
else:
self._session.hooks['response'].extend(session_or_options.hooks['response'])
else: else:
self._session_options = session_options if session_or_options is None:
if driver_options is None: self._session_options = OptionsManager(ini_path).get_option('session_options')
om = OptionsManager(ini_path) else:
self._driver_options = om.get_option('chrome_options') self._session_options = session_or_options
if 'chromedriver_path' in om.get_option('paths') and om.get_option('paths')['chromedriver_path']: if isinstance(driver_or_options, WebDriver):
self._driver_path = om.get_option('paths')['chromedriver_path'] self._driver = driver_or_options
else: else:
self._driver_options = _chrome_options_to_dict(driver_options) if driver_or_options is None:
if 'driver_path' in self._driver_options and self._driver_options['driver_path']: om = OptionsManager(ini_path)
self._driver_path = self._driver_options['driver_path'] self._driver_options = om.get_option('chrome_options')
if 'chromedriver_path' in om.get_option('paths') and om.get_option('paths')['chromedriver_path']:
self._driver_path = om.get_option('paths')['chromedriver_path']
else:
self._driver_options = _chrome_options_to_dict(driver_or_options)
if 'driver_path' in self._driver_options and self._driver_options['driver_path']:
self._driver_path = self._driver_options['driver_path']
@property @property
def session(self) -> HTMLSession: def session(self) -> HTMLSession: