From 50f2130f9c50144f4618d216ef1fdf1d3103f283 Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Sun, 7 Jun 2020 00:14:42 +0800
Subject: [PATCH] =?UTF-8?q?MixPage=E7=9A=84=5F=5Finit=5F=5F()=E7=9A=84dris?=
 =?UTF-8?q?sion=E5=8F=82=E6=95=B0=E5=8F=AF=E6=8E=A5=E6=94=B6's'=E6=88=96'd?=
 =?UTF-8?q?'=EF=BC=8C=E4=BB=A5=E7=AE=80=E6=98=93=E6=96=B9=E5=BC=8F?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BA=E5=AF=B9=E8=B1=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/mix_page.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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