from aio_proxy.response.formatters.elus import format_elus
from aio_proxy.response.unite_legale_model import CollectiviteTerritoriale


def format_collectivite_territoriale(
    colter_code=None,
    colter_code_insee=None,
    colter_elus=None,
    colter_niveau=None,
):
    if colter_code is None:
        return None
    else:
        return CollectiviteTerritoriale(
            code=colter_code,
            code_insee=colter_code_insee,
            elus=format_elus(colter_elus),  # Format elus if provided
            niveau=colter_niveau,
        ).dict()
