# -*- mode: python -*-
# -*- coding: utf-8 -*-

##############################################################################
#
# Gestion scolarite IUT
#
# Copyright (c) 1999 - 2024 Emmanuel Viennet.  All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
#   Emmanuel Viennet      emmanuel.viennet@viennet.net
#
##############################################################################

"""Rapports suivi:
  - statistiques decisions
  - suivi cohortes
"""
import collections
import os
import tempfile
import re
import time
import datetime
from operator import itemgetter

from flask import url_for, g, request
import pydot

from app import log
from app.but import jury_but
from app.comp import res_sem
from app.comp.res_compat import NotesTableCompat
from app.models import FormSemestre, ScolarAutorisationInscription
from app.models import FormationModalite
from app.models.etudiants import Identite

from app.scodoc import (
    codes_cursus,
    html_sco_header,
    sco_etud,
    sco_formsemestre,
    sco_formsemestre_inscriptions,
    sco_groups_view,
    sco_preferences,
)
from app.scodoc.gen_tables import GenTable
from app.scodoc.codes_cursus import code_semestre_validant
from app.scodoc.sco_exceptions import ScoValueError
from app.scodoc import notesdb as ndb
import app.scodoc.sco_utils as scu
import sco_version

MAX_ETUD_IN_DESCR = 20

LEGENDES_CODES_BUT = {
    "Nb_rcue_valides": "nb RCUE validés",
    "decision_annee": "code jury annuel BUT",
}


def formsemestre_etuds_stats(
    formsemestre: FormSemestre,
    only_primo=False,
    groups_infos: sco_groups_view.DisplayedGroupsInfos | None = None,
):
    """Récupère liste d'etudiants avec etat et decision."""
    nt: NotesTableCompat = res_sem.load_formsemestre_results(formsemestre)
    etudids = groups_infos.get_etudids() if groups_infos else set()
    rows = nt.get_table_moyennes_triees()
    # Décisions de jury BUT pour les semestres pairs seulement
    jury_but_mode = (
        formsemestre.formation.is_apc() and formsemestre.semestre_id % 2 == 0
    )
    # Construit liste d'étudiants du semestre avec leur decision
    etuds = []
    for t in rows:
        etudid = t[-1]
        if etudids and etudid not in etudids:
            continue
        etudiant = Identite.get_etud(etudid)
        etud = sco_etud.get_etud_info(etudid=etudid, filled=True)[0]
        etud["annee_admission"] = etud["annee"]  # plus explicite
        decision = nt.get_etud_decision_sem(etudid)
        if decision:
            etud["codedecision"] = decision["code"]
        etud["etat"] = nt.get_etud_etat(etudid)
        if etud["etat"] == "D":
            etud["codedecision"] = "DEM"
        if "codedecision" not in etud:
            etud["codedecision"] = "(nd)"  # pas de decision jury
        # Ajout devenir (autorisations inscriptions), utile pour stats passage
        aut_list = ScolarAutorisationInscription.query.filter_by(
            etudid=etudid, origin_formsemestre_id=formsemestre.id
        ).all()
        autorisations = [f"S{a.semestre_id}" for a in aut_list]
        autorisations.sort()
        autorisations_str = ", ".join(autorisations)
        etud["devenir"] = autorisations_str
        # Décisions de jury BUT (APC)
        if jury_but_mode:
            deca = jury_but.DecisionsProposeesAnnee(etudiant, formsemestre)
            etud["nb_rcue_valides"] = deca.nb_rcue_valides
            etud["decision_annee"] = deca.code_valide
        # Ajout clé 'bac-specialite'
        bs = []
        if etud["bac"]:
            bs.append(etud["bac"])
        if etud["specialite"]:
            bs.append(etud["specialite"])
        etud["bac-specialite"] = " ".join(bs)
        #
        if (not only_primo) or is_primo_etud(etud, formsemestre):
            etuds.append(etud)
    return etuds


def is_primo_etud(etud: dict, formsemestre: FormSemestre):
    """Determine si un (filled) etud a été inscrit avant ce semestre.
    Regarde la liste des semestres dans lesquels l'étudiant est inscrit.
    Si semestre pair, considère comme primo-entrants ceux qui étaient
    primo dans le précédent (S_{2n-1}).
    """
    debut_cur_iso = formsemestre.date_debut.isoformat()
    # si semestre impair et sem. précédent contigu, recule date debut
    if (
        (len(etud["sems"]) > 1)
        and (formsemestre.semestre_id % 2 == 0)
        and (etud["sems"][1]["semestre_id"] == (formsemestre.semestre_id - 1))
    ):
        debut_cur_iso = etud["sems"][1]["date_debut_iso"]
    for s in etud["sems"]:  # le + recent d'abord
        if s["date_debut_iso"] < debut_cur_iso:
            return False
    return True


def _categories_and_results(etuds, category, result):
    categories = {}
    results = {}
    for etud in etuds:
        categories[etud[category]] = True
        results[etud[result]] = True
    categories = list(categories.keys())
    categories.sort(key=scu.heterogeneous_sorting_key)
    results = list(results.keys())
    results.sort(key=scu.heterogeneous_sorting_key)
    return categories, results


def _results_by_category(
    etuds,
    category="",
    result="",
    category_name=None,
    formsemestre_id=None,
):
    """Construit table: categories (eg types de bacs) en ligne, décisions jury en colonnes

    etuds est une liste d'etuds (dicts).
    category et result sont des clés de etud (category définie les lignes, result les colonnes).

    Retourne une table.
    """
    if category_name is None:
        category_name = category
    # types de bacs differents:
    categories, results = _categories_and_results(etuds, category, result)
    #
    Count = {}  # { bac : { decision : nb_avec_ce_bac_et_ce_code } }
    results = {}  # { result_value : True }
    for etud in etuds:
        results[etud[result]] = True
        if etud[category] in Count:
            Count[etud[category]][etud[result]] += 1
        else:
            Count[etud[category]] = collections.defaultdict(int, {etud[result]: 1})

    # conversion en liste de dict
    C = [Count[cat] for cat in categories]
    # Totaux par lignes et colonnes
    tot = 0
    for l in [Count[cat] for cat in categories]:
        l["sum"] = sum(l.values())
        tot += l["sum"]
    # pourcentages sur chaque total de ligne
    for l in C:
        l["sumpercent"] = "%2.1f%%" % ((100.0 * l["sum"]) / tot)
    #
    codes = list(results.keys())
    codes.sort(key=scu.heterogeneous_sorting_key)

    bottom_titles = []
    if C:  # ligne du bas avec totaux:
        bottom_titles = {}
        for code in codes:
            bottom_titles[code] = sum([l[code] for l in C])
        bottom_titles["sum"] = tot
        bottom_titles["sumpercent"] = "100%"
        bottom_titles["row_title"] = "Total"

    # ajout titre ligne:
    for cat, l in zip(categories, C):
        l["row_title"] = cat if cat is not None else "?"

    #
    codes.append("sum")
    codes.append("sumpercent")

    # on veut { ADM : ADM, ... }
    titles = {x: x for x in codes}
    # sauf pour
    titles.update(LEGENDES_CODES_BUT)
    titles["sum"] = "Total"
    titles["sumpercent"] = "%"
    titles["DEM"] = "Dém."  # démissions
    titles["row_title"] = titles.get(category_name, category_name)
    return GenTable(
        titles=titles,
        columns_ids=codes,
        rows=C,
        bottom_titles=bottom_titles,
        html_col_width="4em",
        html_sortable=True,
        preferences=sco_preferences.SemPreferences(formsemestre_id),
        table_id=f"results_by_category-{category_name}",
    )


# pages
def formsemestre_report(
    formsemestre_id,
    etuds,
    category="bac",
    result="codedecision",
    category_name="",
    result_name="",
):
    """
    Tableau sur résultats (result) par type de category bac
    """
    sem = sco_formsemestre.get_formsemestre(formsemestre_id)
    if not category_name:
        category_name = category
    if not result_name:
        result_name = result
    if result_name == "codedecision":
        result_name = "résultats"
    #
    tab = _results_by_category(
        etuds,
        category=category,
        category_name=category_name,
        result=result,
        formsemestre_id=formsemestre_id,
    )
    #
    tab.filename = scu.make_filename("stats " + sem["titreannee"])

    tab.origin = f"Généré par {sco_version.SCONAME} le {scu.timedate_human_repr()}"
    tab.caption = (
        f"Répartition des résultats par {category_name}, semestre {sem['titreannee']}"
    )
    tab.html_caption = f"Répartition des résultats par {category_name}."
    return tab


def formsemestre_report_counts(
    formsemestre_id: int,
    fmt="html",
    category: str = "bac",
    result: str = None,
    allkeys: bool = False,
    only_primo: bool = False,
    group_ids: list[int] = None,  # si indiqué, ne prend que ces groupes
):
    """
    Tableau comptage avec choix des categories
    category: attribut en lignes
    result: attribut en colonnes
    only_primo: restreint aux primo-entrants (= non redoublants)
    allkeys: pour le menu du choix de l'attribut en colonnes:
        si vrai, toutes les valeurs présentes dans les données
        sinon liste prédéfinie (voir ci-dessous)
    """
    formsemestre = FormSemestre.get_formsemestre(formsemestre_id)
    groups_infos = sco_groups_view.DisplayedGroupsInfos(
        group_ids,
        formsemestre_id=formsemestre.id,
        select_all_when_unspecified=True,
    )
    # Décisions de jury BUT pour les semestres pairs seulement
    jury_but_mode = (
        formsemestre.formation.is_apc() and formsemestre.semestre_id % 2 == 0
    )

    if result is None:
        result = "statut" if formsemestre.formation.is_apc() else "codedecision"

    category_name = category.capitalize()
    title = "Comptages " + category_name
    etuds = formsemestre_etuds_stats(
        formsemestre, groups_infos=groups_infos, only_primo=only_primo
    )
    tab = formsemestre_report(
        formsemestre_id,
        etuds,
        category=category,
        result=result,
        category_name=category_name,
    )
    tab.base_url = url_for(
        "notes.formsemestre_report_counts",
        scodoc_dept=g.scodoc_dept,
        formsemestre_id=formsemestre_id,
        category=category,
        only_primo=int(bool(only_primo)),
        result=result,
        group_ids=group_ids,
    )
    if len(formsemestre.inscriptions) == 0:
        F = ["""<p><em>Aucun étudiant</em></p>"""]
    else:
        if allkeys:
            keys = list(etuds[0].keys())
        else:
            # clés présentées à l'utilisateur:
            keys = [
                "annee_bac",
                "annee_naissance",
                "bac",
                "specialite",
                "bac-specialite",
                "codedecision",
                "devenir",
                "etat",
                "civilite",
                "qualite",
                "villelycee",
                "statut",
                "annee_admission",
                "type_admission",
                "boursier",
                "boursier_prec",
            ]
            if jury_but_mode:
                keys += ["nb_rcue_valides", "decision_annee"]
        keys.sort(key=scu.heterogeneous_sorting_key)
        F = [
            f"""<form id="group_selector" name="f" method="get" action="{request.base_url}">
              Colonnes:
              <select name="result" onchange="document.f.submit()">
            """
        ]
        for k in keys:
            if k == result:
                selected = "selected"
            else:
                selected = ""
            F.append(
                '<option value="%s" %s>%s</option>'
                % (k, selected, LEGENDES_CODES_BUT.get(k, k))
            )
        F.append("</select>")
        F.append(' Lignes: <select name="category" onchange="document.f.submit()">')
        for k in keys:
            if k == category:
                selected = "selected"
            else:
                selected = ""
            F.append(
                '<option value="%s" %s>%s</option>'
                % (k, selected, LEGENDES_CODES_BUT.get(k, k))
            )
        F.append(
            f"""
            </select>
            <div style="margin-top:12px;">
            <input type="checkbox" name="only_primo" onchange="document.f.submit()"
                {'checked' if only_primo else ''}>
                Restreindre aux primo-entrants</input>
            <span style="margin: 12px;">
            Restreindre au(x) groupe(s)&nbsp;:
            {sco_groups_view.menu_groups_choice(groups_infos, submit_on_change=True)
                if groups_infos else ''}
            </span>
            <input type="hidden" name="formsemestre_id" value="{formsemestre_id}"/>
            </div>
            </form>
            """
        )

    tableau = tab.make_page(
        fmt=fmt,
        title="""<h2 class="formsemestre">Comptes croisés</h2>""",
        with_html_headers=False,
    )
    if fmt != "html":
        return tableau
    H = [
        html_sco_header.sco_header(
            cssstyles=sco_groups_view.CSSSTYLES,
            javascripts=sco_groups_view.JAVASCRIPTS,
            page_title=title,
        ),
        tableau,
        "\n".join(F),
        """<p class="help">Le tableau affiche le nombre d'étudiants de ce semestre dans chacun
          des cas choisis: à l'aide des deux menus, vous pouvez choisir les catégories utilisées
          pour les lignes et les colonnes. Le <tt>codedecision</tt> est le code de la décision
          du jury.
          </p>""",
        html_sco_header.sco_footer(),
    ]
    return "\n".join(H)


# --------------------------------------------------------------------------
def table_suivi_cohorte(
    formsemestre: FormSemestre,
    groups_infos,
    percent=False,
    bac="",  # selection sur type de bac
    bacspecialite="",
    annee_bac="",
    annee_admission="",
    civilite=None,
    statut="",
    only_primo=False,
):
    """
    Tableau indiquant le nombre d'etudiants de la cohorte dans chaque état:
    Etat     date_debut_Sn   date1  date2 ...
    S_n       #inscrits en Sn
    S_n+1
    ...
    S_last
    Diplome
    Sorties

    Determination des dates: on regroupe les semestres commençant à des dates proches

    """
    sem = sco_formsemestre.get_formsemestre(formsemestre.id)
    # sem est le semestre origine
    t0 = time.time()

    def logt(op):
        if 0:  # debug, set to 0 in production
            log("%s: %s" % (op, time.time() - t0))

    logt("table_suivi_cohorte: start")
    # 1-- Liste des semestres posterieurs dans lesquels ont été les etudiants de sem
    nt: NotesTableCompat = res_sem.load_formsemestre_results(formsemestre)
    etudids_inscrits = {ins.etudid for ins in formsemestre.inscriptions}
    etudids_groups = groups_infos.get_etudids()
    etudids = etudids_inscrits.intersection(etudids_groups)
    logt("A: orig etuds set")
    S = {formsemestre.id: sem}  # ensemble de formsemestre_id
    orig_set = set()  # ensemble d'etudid du semestre d'origine
    bacs = set()
    bacspecialites = set()
    annee_bacs = set()
    annee_admissions = set()
    civilites = set()
    statuts = set()
    for etudid in etudids:
        etud = sco_etud.get_etud_info(etudid=etudid, filled=True)[0]
        etud["annee_admission"] = etud["annee"]
        bacspe = (etud["bac"] or "?") + " / " + (etud["specialite"] or "")
        # sélection sur bac:
        if (
            (not bac or (bac == etud["bac"]))
            and (not bacspecialite or (bacspecialite == bacspe))
            and (not annee_bac or (annee_bac == str(etud["annee_bac"])))
            and (
                not annee_admission or (annee_admission == str(etud["annee_admission"]))
            )
            and (not civilite or (civilite == etud["civilite"]))
            and (not statut or (statut == etud["statut"]))
            and (not only_primo or is_primo_etud(etud, formsemestre))
        ):
            orig_set.add(etudid)
            # semestres suivants:
            for s in etud["sems"]:
                if ndb.DateDMYtoISO(s["date_debut"]) > ndb.DateDMYtoISO(
                    sem["date_debut"]
                ):
                    S[s["formsemestre_id"]] = s
        if etud.get("bac", False):
            bacs.add(etud["bac"])
            bacspecialites.add(bacspe)
            annee_bacs.add(str(etud["annee_bac"]))
        if etud["annee_admission"] is not None:
            annee_admissions.add(str(etud["annee_admission"]))
        civilites.add(etud["civilite"])
        if etud["statut"]:  # ne montre pas les statuts non renseignés
            statuts.add(etud["statut"])
    sems = list(S.values())
    # tri les semestres par date de debut
    for s in sems:
        d, m, y = [int(x) for x in s["date_debut"].split("/")]
        s["date_debut_dt"] = datetime.datetime(y, m, d)
    sems.sort(key=itemgetter("date_debut_dt"))

    # 2-- Pour chaque semestre, trouve l'ensemble des etudiants venant de sem
    logt("B: etuds sets")
    sem["members"] = orig_set
    for s in sems:
        ins = sco_formsemestre_inscriptions.do_formsemestre_inscription_list(
            args={"formsemestre_id": s["formsemestre_id"]}
        )  # sans dems
        inset = set([i["etudid"] for i in ins])
        s["members"] = orig_set.intersection(inset)
        nb_dipl = 0  # combien de diplomes dans ce semestre ?
        if s["semestre_id"] == nt.parcours.NB_SEM:
            s_formsemestre = FormSemestre.query.get_or_404(s["formsemestre_id"])
            nt: NotesTableCompat = res_sem.load_formsemestre_results(s_formsemestre)
            for etudid in s["members"]:
                dec = nt.get_etud_decision_sem(etudid)
                if dec and code_semestre_validant(dec["code"]):
                    nb_dipl += 1
        s["nb_dipl"] = nb_dipl

    # 3-- Regroupe les semestres par date de debut
    class PeriodSem:
        def __init__(self, datedebut: datetime.datetime, sems: list[dict]):
            self.datedebut = datedebut
            self.sems = sems

    # semestre de depart:
    d, m, y = [int(x) for x in sem["date_debut"].split("/")]
    porigin = PeriodSem(datetime.datetime(y, m, d), [sem])
    P = []  #  liste de periodsem
    #
    tolerance = datetime.timedelta(days=45)
    for s in sems:
        merged = False
        for p in P:
            if abs(s["date_debut_dt"] - p.datedebut) < tolerance:
                p.sems.append(s)
                merged = True
                break
        if not merged:
            p = PeriodSem(s["date_debut_dt"], [s])
            P.append(p)

    # 4-- regroupe par indice de semestre S_i
    indices_sems = list({s["semestre_id"] for s in sems})
    indices_sems.sort()
    for p in P:
        p.nb_etuds = 0  # nombre total d'etudiants dans la periode
        p.sems_by_id = collections.defaultdict(list)
        for s in p.sems:
            p.sems_by_id[s["semestre_id"]].append(s)
            p.nb_etuds += len(s["members"])

    # 5-- Contruit table
    logt("C: build table")
    nb_initial = len(sem["members"])

    def fmtval(x):
        if not x:
            return ""  # ne montre pas les 0
        if percent:
            return "%2.1f%%" % (100.0 * x / nb_initial)
        else:
            return x

    L = [
        {
            "row_title": "Origine: S%s" % sem["semestre_id"],
            porigin.datedebut: nb_initial,
            "_css_row_class": "sorttop",
        }
    ]
    if nb_initial <= MAX_ETUD_IN_DESCR:
        etud_descr = _descr_etud_set(sem["members"])
        L[0]["_%s_help" % porigin.datedebut] = etud_descr
    for idx_sem in indices_sems:
        if idx_sem >= 0:
            d = {"row_title": "S%s" % idx_sem}
        else:
            d = {"row_title": "Autre semestre"}

        for p in P:
            etuds_period = set()
            for s in p.sems:
                if s["semestre_id"] == idx_sem:
                    etuds_period = etuds_period.union(s["members"])
            nbetuds = len(etuds_period)
            if nbetuds:
                d[p.datedebut] = fmtval(nbetuds)
                if nbetuds <= MAX_ETUD_IN_DESCR:  # si peu d'etudiants, indique la liste
                    etud_descr = _descr_etud_set(etuds_period)
                    d["_%s_help" % p.datedebut] = etud_descr
        L.append(d)
    # Compte nb de démissions et de ré-orientation par période
    logt("D: cout dems reos")
    sem["dems"], sem["reos"] = _count_dem_reo(formsemestre.id, sem["members"])
    for p in P:
        p.dems = set()
        p.reos = set()
        for s in p.sems:
            d, r = _count_dem_reo(s["formsemestre_id"], s["members"])
            p.dems.update(d)
            p.reos.update(r)
    # Nombre total d'etudiants par periode
    l = {
        "row_title": "Inscrits",
        "row_title_help": "Nombre d'étudiants inscrits",
        "_table_part": "foot",
        porigin.datedebut: fmtval(nb_initial),
    }
    for p in P:
        l[p.datedebut] = fmtval(p.nb_etuds)
    L.append(l)
    # Nombre de démissions par période
    l = {
        "row_title": "Démissions",
        "row_title_help": "Nombre de démissions pendant la période",
        "_table_part": "foot",
        porigin.datedebut: fmtval(len(sem["dems"])),
    }
    if len(sem["dems"]) <= MAX_ETUD_IN_DESCR:
        etud_descr = _descr_etud_set(sem["dems"])
        l["_%s_help" % porigin.datedebut] = etud_descr
    for p in P:
        l[p.datedebut] = fmtval(len(p.dems))
        if len(p.dems) <= MAX_ETUD_IN_DESCR:
            etud_descr = _descr_etud_set(p.dems)
            l["_%s_help" % p.datedebut] = etud_descr
    L.append(l)
    # Nombre de réorientations par période
    l = {
        "row_title": "Echecs",
        "row_title_help": "Ré-orientations (décisions NAR)",
        "_table_part": "foot",
        porigin.datedebut: fmtval(len(sem["reos"])),
    }
    if len(sem["reos"]) < 10:
        etud_descr = _descr_etud_set(sem["reos"])
        l["_%s_help" % porigin.datedebut] = etud_descr
    for p in P:
        l[p.datedebut] = fmtval(len(p.reos))
        if len(p.reos) <= MAX_ETUD_IN_DESCR:
            etud_descr = _descr_etud_set(p.reos)
            l["_%s_help" % p.datedebut] = etud_descr
    L.append(l)
    # derniere ligne: nombre et pourcentage de diplomes
    l = {
        "row_title": "Diplômes",
        "row_title_help": "Nombre de diplômés à la fin de la période",
        "_table_part": "foot",
    }
    for p in P:
        nb_dipl = 0
        for s in p.sems:
            nb_dipl += s["nb_dipl"]
        l[p.datedebut] = fmtval(nb_dipl)
    L.append(l)

    columns_ids = [p.datedebut for p in P]
    titles = dict([(p.datedebut, p.datedebut.strftime(scu.DATE_FMT)) for p in P])
    titles[porigin.datedebut] = porigin.datedebut.strftime(scu.DATE_FMT)
    if percent:
        pp = "(en % de la population initiale) "
        titles["row_title"] = "%"
    else:
        pp = ""
        titles["row_title"] = ""
    if only_primo:
        pp += "(restreint aux primo-entrants) "
    if bac:
        dbac = " (bacs %s)" % bac
    else:
        dbac = ""
    if bacspecialite:
        dbac += " (spécialité %s)" % bacspecialite
    if annee_bac:
        dbac += " (année bac %s)" % annee_bac
    if annee_admission:
        dbac += " (année admission %s)" % annee_admission
    if civilite:
        dbac += " civilité: %s" % civilite
    if statut:
        dbac += " statut: %s" % statut
    tab = GenTable(
        caption="Suivi cohorte " + pp + sem["titreannee"] + dbac,
        columns_ids=columns_ids,
        filename=scu.make_filename("cohorte " + sem["titreannee"]),
        html_class="table_cohorte",
        html_col_width="4em",
        html_sortable=True,
        origin=f"Généré par {sco_version.SCONAME} le {scu.timedate_human_repr()}",
        page_title="Suivi cohorte " + sem["titreannee"],
        preferences=sco_preferences.SemPreferences(formsemestre.id),
        rows=L,
        table_id="table_suivi_cohorte",
        titles=titles,
    )
    # Explication: liste des semestres associés à chaque date
    if not P:
        expl = [
            '<p class="help">(aucun étudiant trouvé dans un semestre ultérieur)</p>'
        ]
    else:
        expl = ["<h3>Semestres associés à chaque date:</h3><ul>"]
        for p in P:
            expl.append(f"""<li><b>{p.datedebut.strftime(scu.DATE_FMT)}</b>:""")
            ls = []
            for s in p.sems:
                ls.append(formsemestre.html_link_status())
            expl.append(", ".join(ls) + "</li>")
        expl.append("</ul>")
    return (
        tab,
        "\n".join(expl),
        bacs,
        bacspecialites,
        annee_bacs,
        annee_admissions,
        civilites,
        statuts,
    )


def formsemestre_suivi_cohorte(
    formsemestre_id,
    fmt="html",
    group_ids: list[int] = None,  # si indiqué, ne prend que ces groupes
    percent=1,
    bac="",
    bacspecialite="",
    annee_bac="",
    annee_admission="",
    civilite=None,
    statut="",
    only_primo=False,
) -> str:
    """Affiche suivi cohortes par numero de semestre"""
    try:
        annee_bac = str(annee_bac or "")
        annee_admission = str(annee_admission or "")
        percent = int(percent)
    except ValueError as exc:
        raise ScoValueError("formsemestre_suivi_cohorte: argument invalide") from exc

    formsemestre = FormSemestre.get_formsemestre(formsemestre_id)
    groups_infos = sco_groups_view.DisplayedGroupsInfos(
        group_ids,
        formsemestre_id=formsemestre.id,
        select_all_when_unspecified=True,
    )
    (
        tab,
        expl,
        bacs,
        bacspecialites,
        annee_bacs,
        annee_admissions,
        civilites,
        statuts,
    ) = table_suivi_cohorte(
        formsemestre,
        groups_infos=groups_infos,
        percent=percent,
        bac=bac,
        bacspecialite=bacspecialite,
        annee_bac=annee_bac,
        annee_admission=annee_admission,
        civilite=civilite,
        statut=statut,
        only_primo=only_primo,
    )
    tab.base_url = (
        "%s?formsemestre_id=%s&percent=%s&bac=%s&bacspecialite=%s&civilite=%s"
        % (request.base_url, formsemestre.id, percent, bac, bacspecialite, civilite)
    )
    if only_primo:
        tab.base_url += "&only_primo=on"
    t = tab.make_page(fmt=fmt, with_html_headers=False)
    if fmt != "html":
        return t

    base_url = request.base_url
    burl = "%s?formsemestre_id=%s&bac=%s&bacspecialite=%s&civilite=%s&statut=%s" % (
        base_url,
        formsemestre.id,
        bac,
        bacspecialite,
        civilite,
        statut,
    )
    if percent:
        pplink = f"""<p><a class="stdlink"
            href="{burl}&percent=0">Afficher les résultats bruts</a></p>"""
    else:
        pplink = f"""<p><a class="stdlink"
            href="{burl}&percent=1">Afficher les résultats en pourcentages</a></p>"""

    H = [
        html_sco_header.sco_header(
            cssstyles=sco_groups_view.CSSSTYLES,
            javascripts=sco_groups_view.JAVASCRIPTS,
            page_title=tab.page_title,
        ),
        """<h2 class="formsemestre">Suivi cohorte: devenir des étudiants de ce semestre</h2>""",
        _gen_form_selectetuds(
            formsemestre.id,
            groups_infos=groups_infos,
            only_primo=only_primo,
            bac=bac,
            bacspecialite=bacspecialite,
            annee_bac=annee_bac,
            annee_admission=annee_admission,
            civilite=civilite,
            statut=statut,
            bacs=bacs,
            bacspecialites=bacspecialites,
            annee_bacs=annee_bacs,
            annee_admissions=annee_admissions,
            civilites=civilites,
            statuts=statuts,
            percent=percent,
        ),
        t,
        f"""{pplink}
    <p class="help">Nombre d'étudiants dans chaque semestre.
    Les dates indiquées sont les dates approximatives de <b>début</b> des semestres
    (les semestres commençant à des dates proches sont groupés). Le nombre de diplômés
    est celui à la <b>fin</b> du semestre correspondant.
    Lorsqu'il y a moins de {MAX_ETUD_IN_DESCR} étudiants dans une case, vous pouvez
    afficher leurs noms en passant le curseur sur le chiffre.
    </p>
    <p class="help">Les menus permettent de n'étudier que certaines catégories
    d'étudiants (titulaires d'un type de bac, garçons ou filles).
    La case "restreindre aux primo-entrants" permet de ne considérer que les étudiants
    qui n'ont jamais été inscrits dans ScoDoc avant le semestre considéré.
    </p>
    """,
        expl,
        html_sco_header.sco_footer(),
    ]
    return "\n".join(H)


def _gen_form_selectetuds(
    formsemestre_id,
    percent=None,
    only_primo=None,
    bac=None,
    bacspecialite=None,
    annee_bac=None,
    annee_admission=None,
    civilite=None,
    statut=None,
    bacs=None,
    bacspecialites=None,
    annee_bacs=None,
    annee_admissions=None,
    civilites=None,
    statuts=None,
    groups_infos: sco_groups_view.DisplayedGroupsInfos = None,
):
    """HTML form pour choix criteres selection etudiants"""
    annee_bacs = annee_bacs or []
    annee_admissions = annee_admissions or []
    bacs = list(bacs)
    bacs.sort(key=scu.heterogeneous_sorting_key)
    bacspecialites = list(bacspecialites)
    bacspecialites.sort(key=scu.heterogeneous_sorting_key)
    # on peut avoir un mix de chaines vides et d'entiers:
    annee_bacs = [int(x) if x else 0 for x in annee_bacs]
    annee_bacs.sort()
    annee_admissions = [int(x) if x else 0 for x in annee_admissions]
    annee_admissions.sort()
    civilites = list(civilites)
    civilites.sort()
    statuts = list(statuts)
    statuts.sort()
    #
    if bac:
        selected = ""
    else:
        selected = 'selected="selected"'
    F = [
        f"""<form id="group_selector" name="f" method="get" action="{request.base_url}">
    <div>Bac:
    <select name="bac" onchange="javascript: submit(this);">
        <option value="" {selected}>tous</option>
    """
    ]
    for b in bacs:
        if bac == b:
            selected = 'selected="selected"'
        else:
            selected = ""
        F.append(f'<option value="{b}" {selected}>{b}</option>')
    F.append("</select>")
    if bacspecialite:
        selected = ""
    else:
        selected = 'selected="selected"'
    F.append(
        f"""&nbsp; Bac/Specialité:
    <select name="bacspecialite" onchange="javascript: submit(this);">
    <option value="" {selected}>tous</option>
    """
    )
    for b in bacspecialites:
        if bacspecialite == b:
            selected = 'selected="selected"'
        else:
            selected = ""
        F.append(f'<option value="{b}" {selected}>{b}</option>')
    F.append("</select>")
    #
    F.append(
        "&nbsp; Année bac: " + _gen_select_annee("annee_bac", annee_bacs, annee_bac)
    )
    F.append(
        "&nbsp; Année admission: "
        + _gen_select_annee("annee_admission", annee_admissions, annee_admission)
    )
    #
    F.append(
        f"""&nbsp; Genre: <select name="civilite" onchange="javascript: submit(this);">
    <option value="" {selected}>tous</option>
    """
    )
    for b in civilites:
        if civilite == b:
            selected = 'selected="selected"'
        else:
            selected = ""
        F.append(f'<option value="{b}" {selected}>{b}</option>')
    F.append("</select>")

    F.append(
        f"""&nbsp; Statut: <select name="statut" onchange="javascript: submit(this);">
    <option value="" {selected}>tous</option>
    """
    )
    for b in statuts:
        if statut == b:
            selected = 'selected="selected"'
        else:
            selected = ""
        F.append(f'<option value="{b}" {selected}>{b}</option>')

    F.append(
        f"""
        </select>
        <div style="margin-top:12px;">
            <input type="checkbox" name="only_primo"
                onchange="javascript: submit(this);"
                {'checked="1"' if only_primo else ""}/>Restreindre aux primo-entrants

            <span style="margin: 12px;">
            Restreindre au(x) groupe(s)&nbsp;:
            {sco_groups_view.menu_groups_choice(groups_infos, submit_on_change=True)
                if groups_infos else ''}
            </span>
            <input type="hidden" name="formsemestre_id" value="{formsemestre_id}"/>
            <input type="hidden" name="percent" value="{percent}"/>
        </div>
        </div>
        </form>
        """
    )

    return "\n".join(F)


def _gen_select_annee(field, values, value) -> str:
    "une menu html select"
    menu_html = f"""<select name="{field}" onchange="javascript: submit(this);">
    <option value="" {'' if value else 'selected="selected"'}>tous</option>
    """
    for val in values:
        selected = 'selected="selected"' if str(value) == str(val) else ""
        menu_html += f"""<option value="{val}" {selected}>{val}</option>"""
    return menu_html + "</select>"


def _descr_etud_set(etudids) -> str:
    "textual html description of a set of etudids"
    etuds = []
    for etudid in etudids:
        etud = sco_etud.get_etud_info(etudid=etudid, filled=True)[0]
        etud["annee_admission"] = etud["annee"]  # plus explicite
        etuds.append(etud)
    # sort by name
    etuds.sort(key=itemgetter("nom"))
    return ", ".join([e["nomprenom"] for e in etuds])


def _count_dem_reo(formsemestre_id, etudids):
    "count nb of demissions and reorientation in this etud set"
    formsemestre = FormSemestre.get_formsemestre(formsemestre_id)
    nt: NotesTableCompat = res_sem.load_formsemestre_results(formsemestre)
    validations_annuelles = nt.get_validations_annee() if nt.is_apc else {}
    dems = set()
    reos = set()
    for etudid in etudids:
        if nt.get_etud_etat(etudid) == "D":
            dems.add(etudid)
        if nt.is_apc:
            # BUT: utilise les validations annuelles
            validation = validations_annuelles.get(etudid)
            if validation and validation.code in codes_cursus.CODES_SEM_REO:
                reos.add(etudid)
        else:
            # Autres formations: validations de semestres
            dec = nt.get_etud_decision_sem(etudid)
            if dec and dec["code"] in codes_cursus.CODES_SEM_REO:
                reos.add(etudid)
    return dems, reos


EXP_LIC = re.compile(r"licence", re.I)
EXP_LPRO = re.compile(r"professionnelle", re.I)


def _code_sem(
    semestre_id: int, titre: str, mois_debut: int, short=True, prefix=""
) -> str:
    "code semestre: S1 ou S1d"
    idx = semestre_id
    # semestre décalé ?
    # les semestres pairs normaux commencent entre janvier et mars
    # les impairs normaux entre aout et decembre
    d = ""
    if idx > 0:
        if (idx % 2 and mois_debut < 3) or (idx % 2 == 0 and mois_debut >= 8):
            d = "d"
    if idx == -1:
        if short:
            idx = "Autre "
        else:
            idx = titre + " "
            idx = EXP_LPRO.sub("pro.", idx)
            idx = EXP_LIC.sub("Lic.", idx)
            prefix = ""  # indique titre au lieu de Sn
    return prefix + str(idx) + d


def _code_sem_formsemestre(formsemestre: FormSemestre, short=True, prefix="") -> str:
    "code semestre: S1 ou S1d"
    titre = formsemestre.titre
    mois_debut = formsemestre.date_debut.month
    semestre_id = formsemestre.semestre_id
    return _code_sem(semestre_id, titre, mois_debut, short=short, prefix=prefix)


def _code_sem_dict(sem, short=True, prefix="") -> str:
    "code semestre: S1 ou S1d, à parit d'un dict (sem ScoDoc 7)"
    titre = sem["titre"]
    mois_debut = int(sem["date_debut"].split("/")[1]) if sem["date_debut"] else 0
    semestre_id = sem["semestre_id"]
    return _code_sem(semestre_id, titre, mois_debut, short=short, prefix=prefix)


def get_code_cursus_etud(
    etudid: int,
    sems: list[dict] = None,
    formsemestres: list[FormSemestre] | None = None,
    prefix="",
    separator="",
) -> tuple[str, dict]:
    """calcule un code de cursus (parcours) pour un etudiant
    exemples:
       1234A pour un etudiant ayant effectué S1, S2, S3, S4 puis diplome
       12D   pour un étudiant en S1, S2 puis démission en S2
       12R   pour un etudiant en S1, S2 réorienté en fin de S2

    On peut passer soir la liste des semestres dict (anciennes fonctions ScoDoc7)
    soit la liste des FormSemestre.
    Construit aussi un dict: { semestre_id : decision_jury | None }
    """
    # Nota: approche plus moderne:
    # ' '.join([ f"S{ins.formsemestre.semestre_id}"
    #           for ins in reversed(etud.inscriptions())
    #           if ins.formsemestre.formation.formation_code == XXX ])
    #
    p = []
    decisions_jury = {}

    if formsemestres is None:
        formsemestres = [
            FormSemestre.query.get_or_404(s["formsemestre_id"]) for s in (sems or [])
        ]

    # élimine les semestres spéciaux hors cursus (LP en 1 sem., ...)
    formsemestres = [s for s in formsemestres if s.semestre_id >= 0]
    i = len(formsemestres) - 1
    while i >= 0:
        # 'sems' est a l'envers, du plus recent au plus ancien
        formsemestre = formsemestres[i]
        nt: NotesTableCompat = res_sem.load_formsemestre_results(formsemestre)

        p.append(_code_sem_formsemestre(formsemestre, prefix=prefix))
        # code decisions jury de chaque semestre:
        if nt.get_etud_etat(etudid) == "D":
            decisions_jury[formsemestre.semestre_id] = "DEM"
        else:
            dec = nt.get_etud_decision_sem(etudid)
            if not dec:
                decisions_jury[formsemestre.semestre_id] = ""
            else:
                decisions_jury[formsemestre.semestre_id] = dec["code"]
        # code etat dans le code_cursus sur dernier semestre seulement
        if i == 0:
            # Démission
            if nt.get_etud_etat(etudid) == "D":
                p.append(":D")
            else:
                dec = nt.get_etud_decision_sem(etudid)
                if dec and dec["code"] in codes_cursus.CODES_SEM_REO:
                    p.append(":R")
                if (
                    dec
                    and formsemestre.semestre_id == nt.parcours.NB_SEM
                    and code_semestre_validant(dec["code"])
                ):
                    p.append(":A")
        i -= 1
    return separator.join(p), decisions_jury


def tsp_etud_list(
    formsemestre_id,
    groups_infos: sco_groups_view.DisplayedGroupsInfos | None = None,
    only_primo=False,
    bac="",  # selection sur type de bac
    bacspecialite="",
    annee_bac="",
    annee_admission="",
    civilite="",
    statut="",
):
    """Liste des etuds a considerer dans table suivi cursus
    ramene aussi ensembles des bacs, genres, statuts de (tous) les etudiants
    """
    formsemestre = FormSemestre.get_formsemestre(formsemestre_id)
    etudids_inscrits = {ins.etudid for ins in formsemestre.inscriptions}
    if groups_infos:
        etudids_groups = groups_infos.get_etudids()
        etudids = etudids_inscrits.intersection(etudids_groups)
    else:
        etudids = etudids_inscrits
    etuds = []
    bacs = set()
    bacspecialites = set()
    annee_bacs = set()
    annee_admissions = set()
    civilites = set()
    statuts = set()
    for etudid in etudids:
        etud = sco_etud.get_etud_info(etudid=etudid, filled=True)[0]
        etud["annee_admission"] = etud["annee"]  # plus explicite
        bacspe = (etud["bac"] or "?") + " / " + (etud["specialite"] or "")
        # sélection sur bac, primo, ...:
        if (
            (not bac or (bac == etud["bac"]))
            and (not bacspecialite or (bacspecialite == bacspe))
            and (not annee_bac or (annee_bac == str(etud["annee_bac"])))
            and (
                not annee_admission or (annee_admission == str(etud["annee_admission"]))
            )
            and (not civilite or (civilite == etud["civilite"]))
            and (not statut or (statut == etud["statut"]))
            and (not only_primo or is_primo_etud(etud, formsemestre))
        ):
            etuds.append(etud)

        bacs.add(etud["bac"])
        bacspecialites.add(bacspe)
        annee_bacs.add(etud["annee_bac"])
        annee_admissions.add(etud["annee_admission"])
        civilites.add(etud["civilite"])
        if etud["statut"]:  # ne montre pas les statuts non renseignés
            statuts.add(etud["statut"])
    return etuds, bacs, bacspecialites, annee_bacs, annee_admissions, civilites, statuts


def tsp_grouped_list(codes_etuds):
    """Liste pour table regroupant le nombre d'étudiants
    (+ bulle avec les noms) de chaque cursus (parcours)"""
    L = []
    parcours = list(codes_etuds.keys())
    parcours.sort()
    for p in parcours:
        nb = len(codes_etuds[p])
        l = {"parcours": p, "nb": nb}
        if nb <= MAX_ETUD_IN_DESCR:
            l["_nb_help"] = _descr_etud_set([e["etudid"] for e in codes_etuds[p]])
        L.append(l)
    # tri par effectifs décroissants
    L.sort(key=itemgetter("nb"))
    return L


def table_suivi_cursus(formsemestre_id, only_primo=False, grouped_parcours=True):
    """Tableau recapitulant tous les parcours"""
    sem = sco_formsemestre.get_formsemestre(formsemestre_id)
    (
        etuds,
        bacs,
        bacspecialites,
        annee_bacs,
        annee_admissions,
        civilites,
        statuts,
    ) = tsp_etud_list(formsemestre_id, only_primo=only_primo)
    codes_etuds = collections.defaultdict(list)
    for etud in etuds:
        etud["code_cursus"], etud["decisions_jury"] = get_code_cursus_etud(
            etud["etudid"], sems=etud["sems"]
        )
        codes_etuds[etud["code_cursus"]].append(etud)
        fiche_url = url_for(
            "scolar.fiche_etud", scodoc_dept=g.scodoc_dept, etudid=etud["etudid"]
        )
        etud["_nom_target"] = fiche_url
        etud["_prenom_target"] = fiche_url
        etud["_nom_td_attrs"] = f'''id="{etud['etudid']}" class="etudinfo"'''

    titles = {
        "parcours": "Code cursus",
        "nb": "Nombre d'étudiants",
        "civilite": "",
        "nom": "Nom",
        "prenom": "Prénom",
        "etudid": "etudid",
        "code_cursus": "Code cursus",
        "bac": "Bac",
        "specialite": "Spe.",
    }

    if grouped_parcours:
        L = tsp_grouped_list(codes_etuds)
        columns_ids = ("parcours", "nb")
    else:
        # Table avec le cursus de chaque étudiant:
        L = etuds
        columns_ids = (
            "etudid",
            "civilite",
            "nom",
            "prenom",
            "bac",
            "specialite",
            "code_cursus",
        )
        # Calcule intitulés de colonnes
        S = set()
        sems_ids = list(S.union(*[list(e["decisions_jury"].keys()) for e in etuds]))
        sems_ids.sort()
        sem_tits = ["S%s" % s for s in sems_ids]
        titles.update([(s, s) for s in sem_tits])
        columns_ids += tuple(sem_tits)
        for etud in etuds:
            for s in etud["decisions_jury"]:
                etud["S%s" % s] = etud["decisions_jury"][s]

    if only_primo:
        primostr = "primo-entrants du"
    else:
        primostr = "passés dans le"
    tab = GenTable(
        columns_ids=columns_ids,
        rows=L,
        titles=titles,
        origin="Généré par %s le " % sco_version.SCONAME
        + scu.timedate_human_repr()
        + "",
        caption="Parcours suivis, étudiants %s semestre " % primostr
        + sem["titreannee"],
        page_title="Parcours " + sem["titreannee"],
        html_sortable=True,
        html_class="table_leftalign table_listegroupe",
        html_next_section="""<table class="help">
                    <tr><td><tt>1, 2, ...</tt></td><td> numéros de semestres</td></tr>
                    <tr><td><tt>1d, 2d, ...</tt></td><td>semestres "décalés"</td></tr>
                    <tr><td><tt>:A</tt></td><td> étudiants diplômés</td></tr>
                    <tr><td><tt>:R</tt></td><td> étudiants réorientés</td></tr>
                    <tr><td><tt>:D</tt></td><td> étudiants démissionnaires</td></tr>
                    </table>""",
        bottom_titles={
            "parcours": "Total",
            "nb": len(etuds),
            "code_cursus": len(etuds),
        },
        preferences=sco_preferences.SemPreferences(formsemestre_id),
        table_id="table_suivi_cursus",
    )
    return tab


def tsp_form_primo_group(
    only_primo, no_grouping, formsemestre_id, fmt, groups_infos=None
) -> str:
    """Element de formulaire pour choisir si restriction aux primos entrants,
    groupement par lycees et groupes
    """
    primo_checked = 'checked="1"' if only_primo else ""
    no_grouping_checked = 'checked="1"' if no_grouping else ""
    return f"""
        <form id="group_selector" name="f" method="get" action="{request.base_url}">
        <input type="checkbox" name="only_primo"
            onchange="document.f.submit()" {primo_checked}>Restreindre aux primo-entrants</input>
        <input type="checkbox" name="no_grouping" onchange="document.f.submit()"
            {no_grouping_checked}>Lister chaque étudiant</input>

        <span style="margin: 12px;">
            Restreindre au(x) groupe(s)&nbsp;:
            {sco_groups_view.menu_groups_choice(groups_infos, submit_on_change=True)
                if groups_infos else ''}
        </span>
        <input type="hidden" name="formsemestre_id" value="{formsemestre_id}"/>
        <input type="hidden" name="fmt" value="{fmt}"/>
        </form>
        """


def formsemestre_suivi_cursus(
    formsemestre_id,
    fmt="html",
    only_primo=False,
    no_grouping=False,
):
    """Effectifs dans les differents cursus possibles."""
    tab = table_suivi_cursus(
        formsemestre_id,
        only_primo=only_primo,
        grouped_parcours=not no_grouping,
    )
    tab.base_url = "%s?formsemestre_id=%s" % (request.base_url, formsemestre_id)
    if only_primo:
        tab.base_url += "&only_primo=1"
    if no_grouping:
        tab.base_url += "&no_grouping=1"
    t = tab.make_page(fmt=fmt, with_html_headers=False)
    if fmt != "html":
        return t
    F = [
        tsp_form_primo_group(
            only_primo, no_grouping, formsemestre_id, fmt, groups_infos=None
        )
    ]

    H = [
        html_sco_header.sco_header(
            page_title=tab.page_title,
            javascripts=["js/etud_info.js"],
        ),
        """<h2 class="formsemestre">Cursus suivis par les étudiants de ce semestre</h2>""",
        "\n".join(F),
        t,
        html_sco_header.sco_footer(),
    ]
    return "\n".join(H)


# -------------
def graph_cursus(
    formsemestre_id,
    groups_infos: sco_groups_view.DisplayedGroupsInfos | None = None,
    fmt="svg",
    only_primo=False,
    bac="",  # selection sur type de bac
    bacspecialite="",
    annee_bac="",
    annee_admission="",
    civilite="",
    statut="",
):
    """"""
    (
        etuds,
        bacs,
        bacspecialites,
        annee_bacs,
        annee_admissions,
        civilites,
        statuts,
    ) = tsp_etud_list(
        formsemestre_id,
        groups_infos=groups_infos,
        only_primo=only_primo,
        bac=bac,
        bacspecialite=bacspecialite,
        annee_bac=annee_bac,
        annee_admission=annee_admission,
        civilite=civilite,
        statut=statut,
    )
    # log('graph_cursus: %s etuds (only_primo=%s)' % (len(etuds), only_primo))
    if not etuds:
        return (
            "",
            etuds,
            bacs,
            bacspecialites,
            annee_bacs,
            annee_admissions,
            civilites,
            statuts,
        )
    edges = collections.defaultdict(set)
    # {("SEM"formsemestre_id_origin, "SEM"formsemestre_id_dest) : etud_set}

    def sem_node_name(sem, prefix="SEM"):
        "pydot node name for this integer id"
        return prefix + str(sem["formsemestre_id"])

    sems = {}
    effectifs = collections.defaultdict(set)  # formsemestre_id : etud_set
    decisions = collections.defaultdict(dict)  # formsemestre_id : { code : nb_etud }
    isolated_nodes = []  # [ node_name_de_formsemestre_id, ... ]
    connected_nodes = set()  # { node_name_de_formsemestre_id }
    diploma_nodes = []
    dem_nodes = {}  # formsemestre_id : noeud (node name) pour demissionnaires
    nar_nodes = {}  # formsemestre_id : noeud pour NAR
    for etud in etuds:
        nxt = {}
        etudid = etud["etudid"]
        for s in etud["sems"]:  # du plus recent au plus ancien
            s_formsemestre = FormSemestre.query.get_or_404(s["formsemestre_id"])
            nt: NotesTableCompat = res_sem.load_formsemestre_results(s_formsemestre)
            dec = nt.get_etud_decision_sem(etudid)
            if nxt:
                if (
                    s["semestre_id"] == nt.parcours.NB_SEM
                    and dec
                    and code_semestre_validant(dec["code"])
                    and nt.get_etud_etat(etudid) == scu.INSCRIT
                ):
                    # cas particulier du diplome puis poursuite etude
                    edges[
                        (
                            sem_node_name(s, "_dipl_"),
                            sem_node_name(nxt),  # = "SEM{formsemestre_id}"
                        )
                    ].add(etudid)
                else:
                    edges[(sem_node_name(s), sem_node_name(nxt))].add(etudid)
                connected_nodes.add(sem_node_name(s))
                connected_nodes.add(sem_node_name(nxt))
            else:
                isolated_nodes.append(sem_node_name(s))
            sems[s["formsemestre_id"]] = s
            effectifs[s["formsemestre_id"]].add(etudid)
            nxt = s
            # Compte decisions jury de chaque semestres:
            dc = decisions[s["formsemestre_id"]]
            if dec:
                if dec["code"] in dc:
                    dc[dec["code"]] += 1
                else:
                    dc[dec["code"]] = 1
            # ajout noeud pour demissionnaires
            if nt.get_etud_etat(etudid) == "D":
                nid = sem_node_name(s, "_dem_")
                dem_nodes[s["formsemestre_id"]] = nid
                edges[(sem_node_name(s), nid)].add(etudid)
            # ajout noeud pour NAR (seulement pour noeud de depart)
            if (
                s["formsemestre_id"] == formsemestre_id
                and dec
                and dec["code"] == codes_cursus.NAR
            ):
                nid = sem_node_name(s, "_nar_")
                nar_nodes[s["formsemestre_id"]] = nid
                edges[(sem_node_name(s), nid)].add(etudid)

            # si "terminal", ajoute noeud pour diplomes
            if s["semestre_id"] == nt.parcours.NB_SEM:
                if (
                    dec
                    and code_semestre_validant(dec["code"])
                    and nt.get_etud_etat(etudid) == scu.INSCRIT
                ):
                    nid = sem_node_name(s, "_dipl_")
                    edges[(sem_node_name(s), nid)].add(etudid)
                    diploma_nodes.append(nid)
    #
    g = scu.graph_from_edges(list(edges.keys()))
    for fid in isolated_nodes:
        if not fid in connected_nodes:
            n = pydot.Node(name=fid)
            g.add_node(n)
    g.set("rankdir", "LR")  # left to right
    g.set_fontname("Helvetica")
    if fmt == "svg":
        g.set_bgcolor("#fffff0")  # ou 'transparent'
    # titres des semestres:
    for s in sems.values():
        n = g.get_node(sem_node_name(s))[0]
        log("s['formsemestre_id'] = %s" % s["formsemestre_id"])
        log("n=%s" % n)
        log("get=%s" % g.get_node(sem_node_name(s)))
        log("nodes names = %s" % [x.get_name() for x in g.get_node_list()])
        if s["modalite"] and s["modalite"] != FormationModalite.DEFAULT_MODALITE:
            modalite = " " + s["modalite"]
        else:
            modalite = ""
        label = "%s%s\\n%d/%s - %d/%s\\n%d" % (
            _code_sem_dict(s, short=False, prefix="S"),
            modalite,
            s["mois_debut_ord"],
            s["annee_debut"][2:],
            s["mois_fin_ord"],
            s["annee_fin"][2:],
            len(effectifs[s["formsemestre_id"]]),
        )
        n.set("label", scu.suppress_accents(label))
        n.set_fontname("Helvetica")
        n.set_fontsize(8.0)
        n.set_width(1.2)
        n.set_shape("box")
        n.set_URL(f"formsemestre_status?formsemestre_id={s['formsemestre_id']}")
    # semestre de depart en vert
    n = g.get_node("SEM" + str(formsemestre_id))[0]
    n.set_color("green")
    n.set_style("filled")
    n.set_fillcolor("lightgreen")
    n.set_penwidth(2.0)
    # demissions en rouge, octagonal
    for nid in dem_nodes.values():
        n = g.get_node(nid)[0]
        n.set_color("red")
        n.set_shape("octagon")
        n.set("label", "Dem.")

    # NAR en rouge, Mcircle
    for nid in nar_nodes.values():
        n = g.get_node(nid)[0]
        n.set_color("red")
        n.set_shape("Mcircle")
        n.set("label", codes_cursus.NAR)
    # diplomes:
    for nid in diploma_nodes:
        n = g.get_node(nid)[0]
        n.set_color("red")
        n.set_shape("ellipse")
        n.set("label", "Diplome")  # bug si accent (pas compris pourquoi)
    # Arètes:
    bubbles = {}  # substitue titres pour bulle aides: src_id:dst_id : etud_descr
    for src_id, dst_id in edges.keys():
        e = g.get_edge(src_id, dst_id)[0]
        e.set("arrowhead", "normal")
        e.set("arrowsize", 1)
        e.set_label(len(edges[(src_id, dst_id)]))
        e.set_fontname("Helvetica")
        e.set_fontsize(8.0)
        # bulle avec liste etudiants
        if len(edges[(src_id, dst_id)]) <= MAX_ETUD_IN_DESCR:
            etud_descr = _descr_etud_set(edges[(src_id, dst_id)])
            bubbles[src_id + ":" + dst_id] = etud_descr
            e.set_URL(f"__xxxetudlist__?{src_id}:{dst_id}")
    # Genere graphe
    _, path = tempfile.mkstemp(".gr")
    g.write(path=path, format=fmt)
    with open(path, "rb") as f:
        data = f.read()
    log("dot generated %d bytes in %s format" % (len(data), fmt))
    if not data:
        log("graph.to_string=%s" % g.to_string())
        raise ValueError("Erreur lors de la génération du document au format %s" % fmt)
    os.unlink(path)
    if fmt == "svg":
        # dot génère un document XML complet, il faut enlever l'en-tête
        data_str = data.decode("utf-8")
        data = "<svg" + "<svg".join(data_str.split("<svg")[1:])

        # Substitution des titres des URL des aretes pour bulles aide
        def repl(m):
            return '<a title="%s"' % bubbles[m.group("sd")]

        exp = re.compile(
            r'<a.*?href="__xxxetudlist__\?(?P<sd>\w*:\w*).*?".*?xlink:title=".*?"', re.M
        )
        data = exp.sub(repl, data)
        # Substitution des titres des boites (semestres)
        exp1 = re.compile(
            r'<a xlink:href="formsemestre_status\?formsemestre_id=(?P<fid>\w*).*?".*?xlink:title="(?P<title>.*?)"',
            re.M | re.DOTALL,
        )

        def repl_title(m):
            fid = int(m.group("fid"))
            title = sems[fid]["titreannee"]
            if decisions[fid]:
                title += (
                    (" (" + str(decisions[fid]) + ")").replace("{", "").replace("'", "")
                )
            return (
                '<a xlink:href="formsemestre_status?formsemestre_id=%s" xlink:title="%s"'
                % (fid, scu.suppress_accents(title))
            )  # evite accents car svg utf-8 vs page en latin1...

        data = exp1.sub(repl_title, data)
        # Substitution de Arial par Helvetica (new prblem in Debian 5) ???
        # bug turnaround: il doit bien y avoir un endroit ou regler cela ?
        # cf http://groups.google.com/group/pydot/browse_thread/thread/b3704c53e331e2ec
        data = data.replace("font-family:Arial", "font-family:Helvetica")

    return (
        data,
        etuds,
        bacs,
        bacspecialites,
        annee_bacs,
        annee_admissions,
        civilites,
        statuts,
    )


def formsemestre_graph_cursus(
    formsemestre_id,
    group_ids: list[int] = None,  # si indiqué, ne prend que ces groupes
    fmt="html",
    only_primo=False,
    bac="",  # selection sur type de bac
    bacspecialite="",
    annee_bac="",
    annee_admission="",
    civilite="",
    statut="",
    allkeys=False,  # unused
):
    """Graphe suivi cohortes"""
    annee_bac = str(annee_bac or "")
    annee_admission = str(annee_admission or "")
    formsemestre = FormSemestre.get_formsemestre(formsemestre_id)
    groups_infos = sco_groups_view.DisplayedGroupsInfos(
        group_ids,
        formsemestre_id=formsemestre.id,
        select_all_when_unspecified=True,
    )
    sem = sco_formsemestre.get_formsemestre(formsemestre_id)
    if fmt == "pdf":
        (
            doc,
            etuds,
            bacs,
            bacspecialites,
            annee_bacs,
            annee_admissions,
            civilites,
            statuts,
        ) = graph_cursus(
            formsemestre_id,
            groups_infos=groups_infos,
            fmt="pdf",
            only_primo=only_primo,
            bac=bac,
            bacspecialite=bacspecialite,
            annee_bac=annee_bac,
            annee_admission=annee_admission,
            civilite=civilite,
            statut=statut,
        )
        filename = scu.make_filename("flux " + sem["titreannee"])
        return scu.sendPDFFile(doc, filename + ".pdf")
    elif fmt == "png":
        #
        (
            doc,
            etuds,
            bacs,
            bacspecialites,
            annee_bacs,
            annee_admissions,
            civilites,
            statuts,
        ) = graph_cursus(
            formsemestre_id,
            groups_infos=groups_infos,
            fmt="png",
            only_primo=only_primo,
            bac=bac,
            bacspecialite=bacspecialite,
            annee_bac=annee_bac,
            annee_admission=annee_admission,
            civilite=civilite,
            statut=statut,
        )
        return scu.send_file(
            doc,
            filename="flux " + sem["titreannee"],
            suffix=".png",
            attached=True,
            mime="image/png",
        )
    elif fmt == "html":
        url_kw = {
            "scodoc_dept": g.scodoc_dept,
            "formsemestre_id": formsemestre_id,
            "bac": bac,
            "specialite": bacspecialite,
            "civilite": civilite,
            "statut": statut,
        }
        if only_primo:
            url_kw["only_primo"] = "on"
        (
            doc,
            etuds,
            bacs,
            bacspecialites,
            annee_bacs,
            annee_admissions,
            civilites,
            statuts,
        ) = graph_cursus(
            formsemestre_id,
            groups_infos=groups_infos,
            only_primo=only_primo,
            bac=bac,
            bacspecialite=bacspecialite,
            annee_bac=annee_bac,
            annee_admission=annee_admission,
            civilite=civilite,
            statut=statut,
        )

        H = [
            html_sco_header.sco_header(
                cssstyles=sco_groups_view.CSSSTYLES,
                javascripts=sco_groups_view.JAVASCRIPTS,
                page_title="Graphe cursus de %(titreannee)s" % sem,
                no_side_bar=True,
            ),
            """<h2 class="formsemestre">Cursus des étudiants de ce semestre</h2>""",
            doc,
            f"<p>{len(etuds)} étudiants sélectionnés</p>",
            _gen_form_selectetuds(
                formsemestre_id,
                groups_infos=groups_infos,
                only_primo=only_primo,
                bac=bac,
                bacspecialite=bacspecialite,
                annee_bac=annee_bac,
                annee_admission=annee_admission,
                civilite=civilite,
                statut=statut,
                bacs=bacs,
                bacspecialites=bacspecialites,
                annee_bacs=annee_bacs,
                annee_admissions=annee_admissions,
                civilites=civilites,
                statuts=statuts,
                percent=0,
            ),
            """<p>Origine et devenir des étudiants inscrits dans %(titreannee)s"""
            % sem,
            """(<a href="%s">version pdf</a>"""
            % url_for("notes.formsemestre_graph_cursus", fmt="pdf", **url_kw),
            """, <a href="%s">image PNG</a>)"""
            % url_for("notes.formsemestre_graph_cursus", fmt="png", **url_kw),
            f"""
            </p>
            <p class="help">Le graphe permet de suivre les étudiants inscrits dans le semestre
              sélectionné (dessiné en vert). Chaque rectangle représente un semestre (cliquez dedans
              pour afficher son tableau de bord). Les flèches indiquent le nombre d'étudiants
              passant d'un semestre à l'autre (s'il y en a moins de {MAX_ETUD_IN_DESCR}, vous
              pouvez visualiser leurs noms en passant le curseur sur le chiffre).
            </p>
            <p class="help">
            Le menu <em>Restreindre au(x) groupe(s)</em> permet de restreindre l'étude aux
            étudiants appartenant aux groupes indiqués <em>dans le semestre d'origine</em>.
            </p>
            """,
            html_sco_header.sco_footer(),
        ]
        return "\n".join(H)
    else:
        raise ValueError(f"invalid format: {fmt}")