"""
Students grade peer submissions.
"""

from bok_choy.page_object import PageObject
from bok_choy.promise import Promise
from .rubric import RubricPage


class PeerGradePage(PageObject):
    """
    Students grade peer submissions.
    """

    url = None

    def is_browser_on_page(self):
        def _is_correct_page():
            is_present = (
                self.q(css='div.peer-grading-tools').present or
                self.q(css='div.grading-panel.current-state').present
            )
            return is_present, is_present

        return Promise(_is_correct_page, 'On the peer grading page.').fulfill()

    @property
    def problem_list(self):
        """
        Return the list of available problems to peer grade.
        """
        return self.q(css='a.problem-button').text

    def select_problem(self, problem_name):
        """
        Choose the problem with `problem_name` to start grading or calibrating.
        """
        index = self.problem_list.index(problem_name) + 1
        self.q(css='a.problem-button:nth-of-type({})'.format(index)).first.click()

    @property
    def rubric(self):
        """
        Return a `RubricPage` to allow students to grade their peers.
        If no rubric is available, raises a `BrokenPromise` exception.
        """
        rubric = RubricPage(self.browser)
        rubric.wait_for_page()
        return rubric