TabbycatDebate/tabbycat

from django.utils.html import escape
from django.utils.translation import gettext as _

from utils.misc import reverse_tournament
from utils.tables import TabbycatTableBuilder

from .utils import get_status_meta


class ResultsTableBuilder(TabbycatTableBuilder):

    def add_ballot_check_in_columns(self, debates, key):

        status_header = {
            'key': key,
            'tooltip': _("Whether this debate's ballot has been checked in"),
            'icon': "compass",
        }
        status_cells = []
        for debate in debates:
            cell = {
                'icon': 'check' if debate.checked_in else 'x',
                'class': 'text-secondary' if debate.checked_in else 'text-dark',
                'sort': 1 if debate.checked_in else 0,
                'tooltip': debate.checked_tooltip,
                'check': 'checked' if debate.checked_in else 'missing', # Hook for vue
                'id': debate.id,
                'identifier': debate.barcode if debate.barcode else None,
            }
            status_cells.append(cell)
        self.add_column(status_header, status_cells)

    def add_ballot_status_columns(self, debates, key):

        status_header = {
            'key': key,
            'tooltip': _("Status of this debate's ballot"),
            'icon': "crosshair",
        }
        status_cells = []
        for debate in debates:
            meta = get_status_meta(debate)
            cell = {
                'icon': meta[0],
                'class': meta[1],
                'sort': meta[2],
                'tooltip': meta[3],
                'status': debate.result_status, # Hook for vue
                'id': debate.id,
            }
            status_cells.append(cell)
        self.add_column(status_header, status_cells)

    def get_ballot_cells(self, debate, tournament, view_role, user):
        # These are prefetched, so sort using Python rather than generating an SQL query
        ballotsubmissions = sorted(debate.ballotsubmission_set.all(), key=lambda x: x.version)
        if view_role == 'admin':
            new_link = 'old-results-ballotset-new'
            merge_link = 'results-merge-latest'
        else:
            new_link = 'old-results-assistant-ballotset-new'
            merge_link = 'results-assistant-merge-latest'

        return {
            'component': 'ballots-cell',
            'ballots': [b.serialize(tournament) for b in ballotsubmissions],
            'current_user': escape(user.username),
            'acting_role': view_role,
            'new_ballot': reverse_tournament(new_link, self.tournament,
                                             kwargs={'debate_id': debate.id}),
            'merge_ballot': reverse_tournament(merge_link, self.tournament,
                                               kwargs={'debate_id': debate.id}),
        }

    def add_ballot_entry_columns(self, debates, view_role, user):

        entry_header = {'key': 'EB', 'icon': "plus-circle"}
        entry_cells = [self.get_ballot_cells(d, self.tournament, view_role, user) for d in debates]
        self.add_column(entry_header, entry_cells)