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


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