import allure from gui.elements.qt.button import Button from gui.elements.qt.text_edit import TextEdit from gui.elements.qt.object import QObject class AuthenticatePopup(QObject): def __init__(self): super(AuthenticatePopup, self).__init__('contextMenu_PopupItem') self._password_text_edit = TextEdit('sharedPopup_Password_Input') self._primary_button = Button('sharedPopup_Primary_Button') @allure.step('Authenticate action with password') def authenticate(self, password: str, attempt: int = 2): self._password_text_edit.text = password self._primary_button.click() try: self._primary_button.wait_until_hidden() except AssertionError as err: if attempt: self.authenticate(password, attempt - 1) else: raise err