# Importation du module arcpy pour la manipulation de données géospatiales
import arcpy
# Importation de la classe datetime du module datetime pour la gestion des dates
from datetime import datetime

# Définition de la fonction pour mettre à jour et supprimer des chantiers
def update_and_delete_chantiers(chantier, OBJECTID, Restriction=None, Date_fin=None, delete_expired=False):
    """
    Met a jour les champs 'Restriction' et 'Date_fin' pour un chantier specifique identifie par son numero,
    permettant a l'utilisateur de laisser des valeurs vides pour ne pas les mettre a jour.
    Supprime les travaux qui sont terminés si cette fonction est activée.
 
    Args:
        chantier (str): Chemin vers la classe d'entites 'chantiers'.
        OBJECTID (str): Numero du chantier a mettre a jour.
        Restriction (str): Nouvelle valeur pour le champ 'Restriction', optionnel.
        Date_fin (str): Nouvelle valeur pour le champ 'Date_fin', optionnel.
        delete_expired(boolean): Permet de supprimer les travaux terminés après avoir comparé les dates de fin du travaux avec la date du jours.
    """
    
    # Convertir OBJECTID en entier si possible, sinon None
    objectid_int = int(OBJECTID) if OBJECTID else None
    # Liste des champs à manipuler
    fields = ['OBJECTID', 'Restriction', 'Date_fin']
    # Date du jour
    today = datetime.now()
    # Variable pour suivre si un chantier a été mis à jour
    updated = False
    # Compteur pour le nombre de chantiers supprimés
    deleted_count = 0  

    # Ouvrir un curseur de mise à jour sur la couche chantier
    with arcpy.da.UpdateCursor(chantier, fields) as cursor:
        for row in cursor:
            # Mise à jour du chantier spécifié par OBJECTID
            if objectid_int and row[0] == objectid_int:
                if Restriction not in (None, '', '#', ' '):
                    row[1] = Restriction
                if Date_fin not in (None, '', '#', ' '):
                    row[2] = Date_fin
                cursor.updateRow(row)
                # Message de succès pour la mise à jour
                arcpy.AddMessage(f"Chantier {OBJECTID} mis à jour avec succès.")
                updated = True
 
            # Suppression des chantiers dont la date de fin est dépassée
            if delete_expired and row[2]:
                # Convertir la date de fin en datetime si c'est une chaîne de caractères
                date_fin_chantier = datetime.strptime(row[2], "%d/%m/%Y") if isinstance(row[2], str) else row[2]
                if date_fin_chantier < today:
                    cursor.deleteRow()
                    deleted_count += 1
 
    # Messages finaux
    if updated:
        arcpy.AddMessage(f"Chantier {OBJECTID} a été mis à jour.")
    if delete_expired:
        arcpy.AddMessage(f"{deleted_count} chantier(s) dont la date de fin est dépassée ont été supprimés.")
 
# Point d'entrée du script
if __name__ == "__main__":
    # Récupération des paramètres en entrée
    chantier = arcpy.GetParameterAsText(0)
    OBJECTID = arcpy.GetParameterAsText(1)  # Peut être vide si la mise à jour n'est pas nécessaire
    Restriction = arcpy.GetParameterAsText(2)
    Date_fin = arcpy.GetParameterAsText(3)
    delete_expired = arcpy.GetParameterAsText(4) == 'true'
 
    # Appel de la fonction avec les paramètres récupérés
    update_and_delete_chantiers(chantier, OBJECTID, Restriction, Date_fin, delete_expired)