mode可接收大写字母

This commit is contained in:
g1879 2020-08-13 14:16:43 +08:00
parent 7a5c3b6f76
commit 2ab3b7b5cb

View File

@ -38,8 +38,8 @@ class MixPage(Null, SessionPage, DriverPage):
:param mode: 默认使用selenium的d模式 :param mode: 默认使用selenium的d模式
""" """
super().__init__() super().__init__()
if drission in ['s', 'd']: if drission in ['s', 'd', 'S', 'D']:
mode = drission mode = drission.lower()
drission = None drission = None
self._drission = drission or Drission() self._drission = drission or Drission()
self._session = None self._session = None
@ -86,7 +86,7 @@ class MixPage(Null, SessionPage, DriverPage):
:param mode: 模式字符串 :param mode: 模式字符串
:param go: 是否跳转到原模式的url :param go: 是否跳转到原模式的url
""" """
if mode == self._mode: if mode is not None and mode.lower() == self._mode:
return return
self._mode = 's' if self._mode == 'd' else 'd' self._mode = 's' if self._mode == 'd' else 'd'
if self._mode == 'd': # s转d if self._mode == 'd': # s转d