Code/app/plan_d_action/utils/interface/parser.py

import networkx as nx


def preparer_graphe(G: nx.DiGraph) -> tuple[nx.DiGraph, dict[str | int, int]]:
    """Nettoie et prépare le graphe pour l'analyse.

    Args:
        G (nx.DiGraph): Le graphe d'origine à nettoyer.

    Returns:
        tuple: Un tuple contenant le graphe nettoyé et les niveaux temporels associés
            aux nœuds du graphe. Le graphe nettoyé a eu ses nœuds qui ne respectaient
            pas les critères d'admissibilité enlevés.
    """
    niveaux_temp = {
        node: int(str(attrs.get("niveau")).strip('"'))
        for node, attrs in G.nodes(data=True)
        if attrs.get("niveau") and str(attrs.get("niveau")).strip('"').isdigit()
    }
    G.remove_nodes_from([n for n in G.nodes() if n not in niveaux_temp])
    G.remove_nodes_from(
        [n for n in G.nodes() if niveaux_temp.get(n) == 10 and 'Reserves' in n])
    return G, niveaux_temp