26 lines
1 KiB
Python
26 lines
1 KiB
Python
|
"""
|
||
|
|
Utilitaires de parcours de l'arborescence de catégories (profondeur
|
||
|
|
illimitée). Partagés entre le routeur categories (affichage de l'arbre) et
|
||
|
|
le routeur materiels (construction du <select> indenté).
|
||
|
|
"""
|
||
|
|
|
||
|
|
from sqlmodel import Session, select
|
||
|
|
|
||
|
|
from app.models import Categorie
|
||
|
|
|
||
|
|
|
||
|
|
def categories_racines(session: Session) -> list[Categorie]:
|
||
|
|
"""Catégories de premier niveau (sans parent)."""
|
||
|
|
return list(session.exec(select(Categorie).where(Categorie.parent_id.is_(None))).all())
|
||
|
|
|
||
|
|
|
||
|
|
def aplatir(categories: list[Categorie], profondeur: int = 0) -> list[tuple[int, Categorie]]:
|
||
|
|
"""Aplatit l'arborescence en une liste (profondeur, categorie), dans
|
||
|
|
l'ordre d'affichage (chaque nœud juste avant ses enfants). Sert à
|
||
|
|
construire un <select> indenté sans avoir besoin de JavaScript."""
|
||
|
|
resultat: list[tuple[int, Categorie]] = []
|
||
|
|
for categorie in categories:
|
||
|
|
resultat.append((profondeur, categorie))
|
||
|
|
resultat.extend(aplatir(categorie.enfants, profondeur + 1))
|
||
|
|
return resultat
|