set_window_size()加入最小化功能

This commit is contained in:
g1879 2021-01-29 10:43:41 +08:00
parent 9c96d16485
commit 0b7d55ff53

View File

@ -529,16 +529,19 @@ class DriverPage(object):
self.driver.back() self.driver.back()
def set_window_size(self, x: int = None, y: int = None) -> None: def set_window_size(self, x: int = None, y: int = None) -> None:
"""设置浏览器窗口大小,默认最大化 \n """设置浏览器窗口大小,默认最大化任一参数为0最小化 \n
:param x: 浏览器窗口高 :param x: 浏览器窗口高
:param y: 浏览器窗口宽 :param y: 浏览器窗口宽
:return: None :return: None
""" """
if not x and not y: if x is None and y is None:
self.driver.maximize_window() self.driver.maximize_window()
elif x == 0 or y == 0:
self.driver.minimize_window()
else: else:
if x <= 0 or y <= 0: if x < 0 or y < 0:
raise ValueError('Arguments x and y must greater than 0.') raise ValueError('Arguments x and y must greater than 0.')
new_x = x or self.driver.get_window_size()['width'] new_x = x or self.driver.get_window_size()['width']