from __future__ import absolute_import


class BasePage(object):
    """Base class for PageObjects"""

    def __init__(self, browser):
        self.browser = browser

    @property
    def driver(self):
        return self.browser.driver

    def wait_until_loaded(self):
        self.browser.wait_until_not(".loading-indicator")


class BaseElement(object):
    def __init__(self, element):
        self.element = element


class ButtonElement(BaseElement):
    label_attr = "aria-label"
    disabled_attr = "aria-disabled"

    @property
    def disabled(self):
        return self.element.get_attribute(self.disabled_attr)

    @property
    def label(self):
        return self.element.get_attribute(self.label_attr)

    def click(self):
        self.element.click()


class ButtonWithIconElement(ButtonElement):
    @property
    def icon_href(self):
        return self.element.find_element_by_tag_name("use").get_attribute("href")


class TextBoxElement(BaseElement):
    pass


class ModalElement(BaseElement):
    pass