mangroveorg/datawinners

# vim: ai ts=4 sts=4 et sw=4 encoding=utf-8
from pages.dataanalysispage.data_analysis_locator import *
import time
from pages.submissionlogpage.submission_log_page import SubmissionLogPage
from pages.websubmissionpage.web_submission_page import WebSubmissionPage


class DataAnalysisPage(SubmissionLogPage):
    def __init__(self, driver):
        super(DataAnalysisPage, self).__init__(driver)

    def get_all_data_on_nth_row(self, row):
        row_data = []
        time.sleep(2)
        column_count = self.get_number_of_columns(row)
        if column_count == 1:
            return self.get_empty_datatable_text()
        for col in range(1, column_count + 1):
            row_data.append(self.get_cell_value(row, col))
        return row_data

    def navigate_to_web_submission_tab(self):
        self.driver.find(by_css('.secondary_tab>li:nth-child(4) a')).click()
        return WebSubmissionPage(self.driver)

    def navigate_to_all_data_record_page(self):
        """
        Function to navigate all data record page

        Return data all data record
         """
        self.driver.find(ALL_DATA_RECORDS_LINK).click()
        self.driver.wait_for_element(20, by_css("table.submission_table"))
        return SubmissionLogPage(self.driver)

    def go_to_chart_view(self):
        """
        Function to navigate on next data records page
        """


        self.driver.find(CHART_VIEW_LINK).click()

    def get_no_charts_text(self):
        return self.driver.find(by_css('#no_charts_here')).text

    def is_chart_visible(self):
        return self.driver.find_visible_element(CHART_ELEMENT)

    def get_chart_question_title(self, chart_number):
        return self.driver.find(by_css('#chart_ol li:nth-of-type(' + chart_number + ') h6')).text

    def get_number_of_columns(self, row):
        columns = self.driver.find_elements_(by_xpath(".//*[@id='analysis_table']/tbody/tr[%s]/td" % row))
        return len(columns)

    def get_cell_value(self, row, column):
        # row+1 for ignoring extra row for select all msg
        return self.driver.find(
            by_xpath(".//*[@id='analysis_table']/tbody/tr[%s]/td[%s]" % (row, column))).text