import logging import time import typing import allure import configs import driver from gui.elements.base_object import BaseObject from scripts.tools.image import Image _logger = logging.getLogger(__name__) class QObject(BaseObject): def __init__(self, name, real_name: [str, dict] = None): super().__init__(name, real_name) self._image = Image(self.real_name) def __str__(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 self.object.enabled @property @allure.step('Get selected {0}') def is_selected(self) -> bool: return self.object.selected @property @allure.step('Get checked {0}') def is_checked(self) -> bool: return self.object.checked @property @allure.step('Get visible {0}') def is_visible(self) -> bool: try: return driver.waitForObjectExists(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 ) @allure.step('Hover {0}') def hover(self, timeout_msec: int = configs.timeouts.UI_LOAD_TIMEOUT_MSEC): def _hover(): try: driver.mouseMove(self.object) 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 )