import allure from gui.elements.button import Button from gui.elements.object import QObject class ChangesDetectedToastMessage(QObject): def __init__(self): super(ChangesDetectedToastMessage, self).__init__( 'mainWindow_settingsDirtyToastMessage_SettingsDirtyToastMessage') self._save_button = Button('settingsSave_StatusButton') @allure.step('Save changes') def save(self): self._save_button.click() self.wait_until_hidden()