plateforme-tfjm2/participation/management/commands/generate_seconds_sheet.py

# Copyright (C) 2024 by Animath
# SPDX-License-Identifier: GPL-3.0-or-later

from django.conf import settings
from django.core.management import BaseCommand
from django.utils.translation import activate
import gspread
from gspread.utils import a1_range_to_grid_range, MergeType

from ...models import Passage, Tournament


class Command(BaseCommand):
    def handle(self, *args, **options):
        activate('fr')
        gc = gspread.service_account_from_dict(settings.GOOGLE_SERVICE_CLIENT)
        try:
            spreadsheet = gc.open("Tableau des deuxièmes", folder_id=settings.NOTES_DRIVE_FOLDER_ID)
        except gspread.SpreadsheetNotFound:
            spreadsheet = gc.create("Tableau des deuxièmes", folder_id=settings.NOTES_DRIVE_FOLDER_ID)
            spreadsheet.update_locale("fr_FR")
            spreadsheet.share(None, "anyone", "writer", with_link=True)

        sheet = spreadsheet.sheet1

        header1 = ["Tournoi", "Équipe 2", "Tour 1", "", "", "", "", "", "", "", "Tour 2", "", "", "", "", "", "", "",
                   "Score total", "Score équipe 1", "Score équipe 3"]
        header2 = ["", ""] + 2 * ["PJ", "Problème", "Défenseur⋅se", "", "Opposant⋅e", "", "Rapporteur⋅rice", ""]
        header2 += ["", "", ""]
        header3 = ["", ""] + 2 * (["", ""] + 3 * ["Écrit", "Oral"]) + ["", "", ""]
        lines = [header1, header2, header3]
        nb_tournaments = Tournament.objects.filter(final=False).count()
        for tournament in Tournament.objects.filter(final=False).all():
            line = [tournament.name]
            lines.append(line)

            notes = dict()
            for participation in tournament.participations.filter(valid=True).all():
                note = sum(pool.average(participation)
                           for pool in tournament.pools.filter(participations=participation).all())
                if note:
                    notes[participation] = note

            if not notes:
                continue

            sorted_notes = sorted(notes.items(), key=lambda x: x[1], reverse=True)

            team1, score1 = sorted_notes[0]
            team2, score2 = sorted_notes[1]
            team3, score3 = sorted_notes[2]

            pool1 = tournament.pools.filter(round=1, participations=team2).first()
            defender_passage_1 = Passage.objects.get(pool__tournament=tournament, pool__round=1, defender=team2)
            opponent_passage_1 = Passage.objects.get(pool__tournament=tournament, pool__round=1, opponent=team2)
            reporter_passage_1 = Passage.objects.get(pool__tournament=tournament, pool__round=1, reporter=team2)
            pool2 = tournament.pools.filter(round=2, participations=team2).first()
            defender_passage_2 = Passage.objects.get(pool__tournament=tournament, pool__round=2, defender=team2)
            opponent_passage_2 = Passage.objects.get(pool__tournament=tournament, pool__round=2, opponent=team2)
            reporter_passage_2 = Passage.objects.get(pool__tournament=tournament, pool__round=2, reporter=team2)

            line.append(team2.team.trigram)
            line.append(str(pool1.jury_president or ""))
            line.append(f"Pb. {defender_passage_1.solution_number}")
            line.extend([defender_passage_1.average_defender_writing, defender_passage_1.average_defender_oral,
                         opponent_passage_1.average_opponent_writing, opponent_passage_1.average_opponent_oral,
                         reporter_passage_1.average_reporter_writing, reporter_passage_1.average_reporter_oral])
            line.append(str(pool2.jury_president or ""))
            line.append(f"Pb. {defender_passage_2.solution_number}")
            line.extend([defender_passage_2.average_defender_writing, defender_passage_2.average_defender_oral,
                         opponent_passage_2.average_opponent_writing, opponent_passage_2.average_opponent_oral,
                         reporter_passage_2.average_reporter_writing, reporter_passage_2.average_reporter_oral])
            line.extend([score2, f"{score1:.1f} ({team1.team.trigram})",
                         f"{score3:.1f} ({team3.team.trigram})"])

        sheet.update(lines)

        format_requests = []
        merge_cells = ["A1:A3", "B1:B3", "C1:J1", "K1:R1", "E2:F2", "G2:H2", "I2:J2", "M2:N2", "O2:P2", "Q2:R2",
                       "C2:C3", "D2:D3", "K2:K3", "L2:L3", "S1:S3", "T1:T3", "U1:U3"]
        format_requests.append({"unmergeCells": {"range": a1_range_to_grid_range("A1:AF", sheet.id)}})
        for name in merge_cells:
            grid_range = a1_range_to_grid_range(name, sheet.id)
            format_requests.append({"mergeCells": {"mergeType": MergeType.merge_all, "range": grid_range}})

        bold_ranges = [("A1:AF", False), ("A1:U3", True), (f"A4:A{3 + nb_tournaments}", True)]
        for bold_range, bold in bold_ranges:
            format_requests.append({
                "repeatCell": {
                    "range": a1_range_to_grid_range(bold_range, sheet.id),
                    "cell": {"userEnteredFormat": {"textFormat": {"bold": bold}}},
                    "fields": "userEnteredFormat(textFormat)",
                }
            })

        border_ranges = [("A1:AF", "0000"),
                         (f"A1:U{3 + nb_tournaments}", "1111")]
        sides_names = ['top', 'bottom', 'left', 'right']
        styles = ["NONE", "SOLID", "SOLID_MEDIUM", "SOLID_THICK", "DOUBLE"]
        for border_range, sides in border_ranges:
            borders = {}
            for side_name, side in zip(sides_names, sides):
                borders[side_name] = {"style": styles[int(side)]}
            format_requests.append({
                "repeatCell": {
                    "range": a1_range_to_grid_range(border_range, sheet.id),
                    "cell": {
                        "userEnteredFormat": {
                            "borders": borders,
                            "horizontalAlignment": "CENTER",
                        },
                    },
                    "fields": "userEnteredFormat(borders,horizontalAlignment)",
                }
            })

        column_widths = [("A", 120), ("B", 80), ("C", 180), ("D", 80)] + [(chr(ord("E") + i), 60) for i in range(6)]
        column_widths += [("K", 180), ("L", 80)] + [(chr(ord("M") + i), 60) for i in range(6)]
        column_widths += [("S", 100), ("T", 120), ("U", 120)]
        for column, width in column_widths:
            grid_range = a1_range_to_grid_range(column, sheet.id)
            format_requests.append({
                "updateDimensionProperties": {
                    "range": {
                        "sheetId": sheet.id,
                        "dimension": "COLUMNS",
                        "startIndex": grid_range['startColumnIndex'],
                        "endIndex": grid_range['endColumnIndex'],
                    },
                    "properties": {
                        "pixelSize": width,
                    },
                    "fields": "pixelSize",
                }
            })

        # Set number format, display only one decimal
        number_format_ranges = [f"E4:J{3 + nb_tournaments}", f"M4:S{3 + nb_tournaments}"]
        for number_format_range in number_format_ranges:
            format_requests.append({
                "repeatCell": {
                    "range": a1_range_to_grid_range(number_format_range, sheet.id),
                    "cell": {"userEnteredFormat": {"numberFormat": {"type": "NUMBER", "pattern": "0.0"}}},
                    "fields": "userEnteredFormat.numberFormat",
                }
            })

        body = {"requests": format_requests}
        sheet.client.batch_update(spreadsheet.id, body)