mangroveorg/datawinners

# vim: ai ts=4 sts=4 et sw=4 encoding=utf-8
from selenium.webdriver.firefox import webdriver
from selenium.webdriver.firefox.webdriver import WebDriver

from selenium.webdriver.remote.webelement import WebElement


class DropDown(WebElement):
    def __init__(self, dropDownWebElement):
        super(DropDown, self).__init__(dropDownWebElement.parent, dropDownWebElement.id)
        self.webElement = dropDownWebElement
        self.selectOptions = self.webElement.find_elements_by_tag_name("option")

    def get_options(self):
        """Gets the list of options in the drop down
        """
        return [option.text for option in self.selectOptions]

    def get_selected(self):
        """ Gets the currently selected item in the drop down
        """
        for option in self.selectOptions:
            if option.is_selected():
                return option.get_attribute("value")
        return None

    def get_selected_option_text(self):
        """ Gets the currently selected item text in the drop down
        """
        for option in self.selectOptions:
            if option.is_selected():
                return option.get_attribute("text")
        return None

    def is_selected(self, itemText):
        """ Gets the currently selected item from the drop down
        """
        for option in self.selectOptions:
            if option.text == itemText:
                return option.is_selected()
        return False

    def set_selected(self, itemText):
        """ Selects the provided itemText in the drop down
        """
        for option in self.selectOptions:
            value = option.get_attribute("value")
            if value == itemText:
                option.click()
                break

    def set_selected_by_text(self, itemText):
        """ Selects the provided itemText in the drop down
        """
        for option in self.selectOptions:
            value = option.text
            if value == itemText:
                option.click()
                break