Affichage et tris de données avec Python

#!/usr/bin/env python

# -*- coding: utf-8 -*-

## Import des modules utiles au bon fonctionnement du programme Python

import folium, os, webbrowser, csv

## CONSTANTES

Latitude_centre =     		#exemple 49.5

Longitude_centre =    		#exemple 1.5

Zoom_depart =              	#sur openstreetmap.org, le nombre entier après map=

Fichier_html = 'xxxx.html'  	#nom du fichier html généré avec l'extension .html

Fichier_csv = 'yyyy.csv'        #nom de votre fichier CSV avec l'extension .csv

Separateur = '?'                #le séparateur utilisé dans votre fichier CSV

## FONCTIONS

def carte(Carte, Fichier_csv, Fichier_html, Separateur):

    """Fonction qui réalise une carte au format HTML avec des tuiles OpenStreeMap

    Elle est centrée sur la latitude et longitude définie dans les constantes

    Elle est affichée avec le zoom défini dans les constantes

    Elle utilise le fichier CSV défini dans les constantes et affiche tous les enregistrements du fichier

    Elle crée le fichier HTML défini dans les constantes

    Elle utilse le séparateur défini dans les constantes""" 

    with open(Fichier_csv, newline='') as fichier:

        lecture = csv.DictReader(fichier, delimiter= Separateur)

        MesEtabs = list(lecture)        #Je transforme en liste pour manipuler les établissements

        for etab in MesEtabs:

            if etab['latitude'] != '':   #Si le champ lat ou long n'est pas vide

                latitude = float(etab['latitude'])   #Transformation en float de la latitude

                longitude = float(etab['longitude'])

                folium.Marker(  location=[latitude, longitude],

                                    tooltip=etab['Nom_etablissement'],                     #Au survol, nom de l'établissement

                                    popup=f"{etab['Nom_etablissement']} à {etab['Nom_commune']}", #J'affiche le nom de l'établissement et la commune

                                    icon=folium.Icon(icon="home", color="green"),              #Icone home dans le marqueur

                                    ).add_to(Carte)

    Carte.save(Fichier_html)

def carte_triee(Carte, Fichier_csv, Fichier_html, Separateur):

    """Fonction qui réalise une carte au format HTML avec des tuiles OpenStreeMap

    Elle est centrée sur la latitude et longitude définie dans les constantes

    Elle est affichée avec le zoom défini dans les constantes

    Elle utilise le fichier CSV défini dans les constantes et affiche les lycées en vert et les collèges en bleu

    Elle crée le fichier HTML défini dans les constantes

    Elle utilse le séparateur défini dans les constantes""" 

    with open(Fichier_csv, newline='') as fichier:

        lecture = csv.DictReader(fichier, delimiter= Separateur)

        MesEtabs = list(lecture)        #Je transforme en liste pour manipuler les établissements

        for etab in MesEtabs:

            if etab['latitude'] != '':   #Si le champ lat ou long n'est pas vide

                latitude = float(etab['latitude'])   #Transformation en float de la latitude

                longitude = float(etab['longitude'])

                if etab['Type_etablissement'] == 'Lycée':

                    folium.Marker(  location=[latitude, longitude],

                                    tooltip=etab['Nom_etablissement'],                     #Au survol, nom de l'établissement

                                    popup=f"{etab['Nom_etablissement']} à {etab['Nom_commune']}", #J'affiche le nom de l'établissement et la commune

                                    icon=folium.Icon(icon="home", color="green"),              #Icone home dans le marqueur

                                    ).add_to(Carte)

                elif etab['Type_etablissement'] == 'Collège':

                    folium.Marker(  location=[latitude, longitude],

                                    tooltip=etab['Nom_etablissement'],                     #Au survol, nom de l'établissement

                                    popup=f"{etab['Nom_etablissement']} à {etab['Nom_commune']}", #J'affiche le nom de l'établissement et la commune

                                    icon=folium.Icon(icon="home", color="blue"),              #Icone home dans le marqueur

                                    ).add_to(Carte)

                else: #EREA

                    folium.Marker(  location=[latitude, longitude],

                                    tooltip=etab['Nom_etablissement'],                     #Au survol, nom de l'établissement

                                    popup=f"{etab['Nom_etablissement']} à {etab['Nom_commune']}", #J'affiche le nom de l'établissement et la commune

                                    icon=folium.Icon(icon="home", color="purple"),              #Icone home dans le marqueur

                                    ).add_to(Carte)

    Carte.save(Fichier_html)

## PROGRAMME PRINCIPAL

reponse = input("""Quelle carte souhaitez-vous consulter ?\n

    - une carte avec tous les établissements (taper 1),

    - une carte avec la distinction collèges et les lycées (taper 2) ?\n

    Réponse : """)

if reponse == '1':

    MaCarte = folium.Map(location=(Latitude_centre, Longitude_centre), zoom_start= Zoom_depart)

    carte(MaCarte, Fichier_csv, Fichier_html, Separateur)

else:

    MaCarte = folium.Map(location=(Latitude_centre, Longitude_centre), zoom_start= Zoom_depart)

    carte_triee(MaCarte, Fichier_csv, Fichier_html, Separateur)

webbrowser.open('file://' + os.path.realpath(Fichier_html))        #Ouverture automatique du fichier HTML