mangroveorg/datawinners

from pages.AutomaticReplySmsPage.automatic_reply_sms_page import AutomaticReplySmsPage
from pages.broadcastSMSpage.broadcast_sms_locator import *
from pages.page import Page
from tests.testsettings import UI_TEST_TIMEOUT


class BroadcastSmsPage(Page):
    def __init__(self, driver):
        Page.__init__(self, driver)

    def write_sms_content(self, sms_data):
        self.driver.find_text_box(SMS_CONTENT_TB).enter_text(sms_data)

    def click_send(self):
         self.driver.find(SEND_BROADCAST_SMS_BTN).click()

    def get_sms_content(self):
        return self.driver.find_text_box(SMS_CONTENT_TB).get_attribute("value")

    def choose_type_other_people(self, other_numbers=None):
        self.driver.find(SEND_TO_DDCL).click()
        self.driver.find(OTHER_PEOPLE_OPTION_DDCL).click()
        if other_numbers is not None:
            self.driver.find_text_box(SEND_TO_TB).enter_text(other_numbers)

    def is_other_people_help_text_visible(self):
        return self.driver.find(OTHER_PEOPLE_HELP_TEXT).is_displayed()

    def get_other_people_number_error(self):
        return self.driver.find(OTHER_PEOPLE_ERROR_TEXT_BY_CSS).text

    def get_number_of_recipients_text_for_unregistered_and_associated_data_senders(self):
        self.driver.find(DROP_DOWN).click()
        self.driver.wait_for_element(UI_TEST_TIMEOUT, NUMBER_OF_ANONYMOUS_AND_ASSOCIATED_RECIPIENTS_TEXT_BY_CSS, True)
        number_of_recipients_text = self.driver.find(NUMBER_OF_ANONYMOUS_AND_ASSOCIATED_RECIPIENTS_TEXT_BY_CSS).text
        self.driver.find(DROP_DOWN).click()
        return number_of_recipients_text

    def get_number_of_recipients_for_associated_data_senders(self):
        self.driver.find(DROP_DOWN).click()
        self.driver.wait_for_element(UI_TEST_TIMEOUT, NUMBER_OF_ASSOCIATED_RECIPIENTS_TEXT_BY_CSS, True)
        number_of_recipients = int(self.driver.find(NUMBER_OF_ASSOCIATED_RECIPIENTS_TEXT_BY_CSS).text.split(' ')[0])
        self.driver.find(DROP_DOWN).click()
        return number_of_recipients

    def is_warning_shown(self):
        return self.driver.find(by_id("more_people_warning")).is_displayed()

    def close_warning_dialog(self):
        return self.driver.find(by_id("ok_button")).click()

    def navigate_to_automatic_reply_sms_page(self):
        self.driver.find(REPLY_SMS_LINK).click()
        return AutomaticReplySmsPage(self.driver)

    def is_send_a_message_to_unregistered_present(self):
        self.driver.find(SEND_TO_DDCL).click()
        return self.driver.find(by_css("ul li a#AllSubmitted")).is_displayed()

    def choose_contacts_linked_to_questionnaire(self):
        self.driver.find(CONTACTS_LINKED_TO_QUESTIONNAIRE_LOCATOR).click()

    def get_all_available_questionnaires(self):
        self.choose_contacts_linked_to_questionnaire()
        elements = self.driver.find_elements_(by_css("#send-sms-section > div.tableRow:nth-of-type(7) div.tableCell:nth-of-type(2) > div > input"))
        return [project.get_attribute('value') for project in elements]