from views.base_view import BaseViewObject
from views.base_element import *
from tests import tests_data


class OkButtonAPK(BaseButton):

    def __init__(self, driver):
        super(OkButtonAPK, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[@text='OK']")


class ContinueButtonAPK(BaseButton):

    def __init__(self, driver):
        super(ContinueButtonAPK, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[@text='Continue']")


class TypeMessageEditBox(BaseEditBox):

    def __init__(self, driver):
        super(TypeMessageEditBox, self).__init__(driver)
        self.locator = \
            self.Locator.xpath_selector("//*[@content-desc!='chat-message-input']")


class RequestPasswordIcon(BaseButton):

    def __init__(self, driver):
        super(RequestPasswordIcon, self).__init__(driver)
        self.locator = self.Locator.xpath_selector("//*[@content-desc='request-password']")


class HomeView(BaseViewObject):

    def __init__(self, driver):
        super(HomeView, self).__init__(driver)
        self.continue_button_apk = ContinueButtonAPK(driver)
        self.ok_button_apk = OkButtonAPK(driver)

        for i in self.ok_button_apk, self.continue_button_apk:
            try:
                i.click()
            except (NoSuchElementException, TimeoutException):
                pass

        self.type_message_edit_box = TypeMessageEditBox(driver)
        self.request_password_icon = RequestPasswordIcon(driver)