161 lines
4.5 KiB
Python
161 lines
4.5 KiB
Python
import logging
|
|
import time
|
|
import typing
|
|
|
|
import allure
|
|
|
|
import configs
|
|
import driver
|
|
from gui import objects_map
|
|
from scripts.tools.image import Image
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class QObject:
|
|
|
|
def __init__(self, name, real_name: [str, dict] = None):
|
|
self.symbolic_name = name
|
|
if real_name:
|
|
self.real_name = real_name
|
|
else:
|
|
self.real_name = getattr(objects_map, name)
|
|
self._image = Image(self.real_name)
|
|
|
|
def __getattr__(self, attr: str):
|
|
value = getattr(self.object, attr)
|
|
return value
|
|
|
|
def __str__(self):
|
|
return f'{type(self).__qualname__}({self.symbolic_name})'
|
|
|
|
def __repr__(self):
|
|
return f'{type(self).__qualname__}({self.symbolic_name})'
|
|
|
|
@property
|
|
@allure.step('Get object {0}')
|
|
def object(self):
|
|
return driver.waitForObject(self.real_name, configs.timeouts.UI_LOAD_TIMEOUT_MSEC)
|
|
|
|
@property
|
|
@allure.step('Get object exists {0}')
|
|
def exists(self) -> bool:
|
|
return driver.object.exists(self.real_name)
|
|
|
|
@property
|
|
@allure.step('Get bounds {0}')
|
|
def bounds(self):
|
|
return driver.object.globalBounds(self.object)
|
|
|
|
@property
|
|
@allure.step('Get "x" coordinate {0}')
|
|
def x(self) -> int:
|
|
return self.bounds.x
|
|
|
|
@property
|
|
@allure.step('Get "y" coordinate {0}')
|
|
def y(self) -> int:
|
|
return self.bounds.y
|
|
|
|
@property
|
|
@allure.step('Get width {0}')
|
|
def width(self) -> int:
|
|
return int(self.bounds.width)
|
|
|
|
@property
|
|
@allure.step('Get height {0}')
|
|
def height(self) -> int:
|
|
return int(self.bounds.height)
|
|
|
|
@property
|
|
@allure.step('Get central coordinate {0}')
|
|
def center(self):
|
|
return self.bounds.center()
|
|
|
|
@property
|
|
@allure.step('Get enabled {0}')
|
|
def is_enabled(self) -> bool:
|
|
return getattr(self, 'enabled')
|
|
|
|
@property
|
|
@allure.step('Get selected {0}')
|
|
def is_selected(self) -> bool:
|
|
return getattr(self, 'selected')
|
|
|
|
@property
|
|
@allure.step('Get checked {0}')
|
|
def is_checked(self) -> bool:
|
|
return getattr(self, 'checked')
|
|
|
|
@property
|
|
@allure.step('Get visible {0}')
|
|
def is_visible(self) -> bool:
|
|
try:
|
|
return driver.waitForObject(self.real_name, 0).visible
|
|
except (AttributeError, LookupError, RuntimeError):
|
|
return False
|
|
|
|
@property
|
|
@allure.step('Get image {0}')
|
|
def image(self):
|
|
self._image.update_view()
|
|
return self._image
|
|
|
|
@allure.step('Click {0}')
|
|
def click(
|
|
self,
|
|
x: int = None,
|
|
y: int = None,
|
|
button=None
|
|
):
|
|
driver.mouseClick(
|
|
self.object,
|
|
x or self.width // 2,
|
|
y or self.height // 2,
|
|
button or driver.Qt.LeftButton
|
|
)
|
|
_logger.info(f'{self}: clicked')
|
|
|
|
@allure.step('Hover {0}')
|
|
def hover(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC):
|
|
def _hover():
|
|
try:
|
|
driver.mouseMove(self.object)
|
|
_logger.info(f'{self}: mouse hovered')
|
|
return getattr(self.object, 'hovered', True)
|
|
except RuntimeError as err:
|
|
_logger.debug(err)
|
|
time.sleep(1)
|
|
return False
|
|
|
|
assert driver.waitFor(lambda: _hover(), timeout_msec)
|
|
return self
|
|
|
|
@allure.step('Open context menu')
|
|
def open_context_menu(
|
|
self,
|
|
x: typing.Union[int, driver.UiTypes.ScreenPoint] = None,
|
|
y: typing.Union[int, driver.UiTypes.ScreenPoint] = None,
|
|
):
|
|
self.click(
|
|
x or self.width // 2,
|
|
y or self.height // 2,
|
|
driver.Qt.RightButton
|
|
)
|
|
_logger.info(f'{self}: clicked via Right Mouse Button')
|
|
|
|
@allure.step('Wait until appears {0}')
|
|
def wait_until_appears(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC):
|
|
assert driver.waitFor(lambda: self.is_visible, timeout_msec), f'Object {self} is not visible'
|
|
_logger.info(f'{self}: is visible')
|
|
return self
|
|
|
|
@allure.step('Wait until hidden {0}')
|
|
def wait_until_hidden(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC):
|
|
assert driver.waitFor(lambda: not self.is_visible, timeout_msec), f'Object {self} is not hidden'
|
|
_logger.info(f'{self}: is hidden')
|
|
|
|
@classmethod
|
|
def wait_for(cls, condition, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC) -> bool:
|
|
return driver.waitFor(lambda: condition, timeout_msec)
|