diff --git a/DrissionPage/mix_page.py b/DrissionPage/mix_page.py index 955e6e0..f818440 100644 --- a/DrissionPage/mix_page.py +++ b/DrissionPage/mix_page.py @@ -32,12 +32,15 @@ class MixPage(Null, SessionPage, DriverPage): 这些功能由DriverPage和SessionPage类实现。 """ - def __init__(self, drission: Drission = None, mode: str = 'd', timeout: float = 10): + def __init__(self, drission: Union[Drission, str] = None, mode: str = 'd', timeout: float = 10): """初始化函数 - :param drission: 整合了driver和session的类 + :param drission: 整合了driver和session的类,传入's'或'd'时快速配置相应模式 :param mode: 默认使用selenium的d模式 """ super().__init__() + if drission in ['s', 'd']: + mode = drission + drission = None self._drission = drission or Drission() self._session = None self._driver = None