Upload vacances_scolaires.py

vacances_scolaires.py

@@ -0,0 +1,159 @@
+"""
+Calendrier des vacances scolaires françaises par zone (A, B, C)
+Sources : Ministère de l'Éducation Nationale
+"""
+
+import pandas as pd
+from datetime import date
+from typing import Dict, List, Tuple
+
+# ============================================================
+# VACANCES SCOLAIRES 2023-2024
+# ============================================================
+VACANCES_2023_2024: Dict[str, List[Tuple[date, date]]] = {
+    "A": [
+        (date(2023, 10, 21), date(2023, 11, 5)),
+        (date(2023, 12, 23), date(2024, 1, 7)),
+        (date(2024, 2, 17), date(2024, 3, 3)),
+        (date(2024, 4, 13), date(2024, 4, 28)),
+        (date(2024, 7, 6), date(2024, 9, 1)),
+    ],
+    "B": [
+        (date(2023, 10, 21), date(2023, 11, 5)),
+        (date(2023, 12, 23), date(2024, 1, 7)),
+        (date(2024, 2, 24), date(2024, 3, 10)),
+        (date(2024, 4, 20), date(2024, 5, 5)),
+        (date(2024, 7, 6), date(2024, 9, 1)),
+    ],
+    "C": [
+        (date(2023, 10, 21), date(2023, 11, 5)),
+        (date(2023, 12, 23), date(2024, 1, 7)),
+        (date(2024, 2, 10), date(2024, 2, 25)),
+        (date(2024, 4, 6), date(2024, 4, 21)),
+        (date(2024, 7, 6), date(2024, 9, 1)),
+    ],
+}
+
+# ============================================================
+# VACANCES SCOLAIRES 2024-2025
+# ============================================================
+VACANCES_2024_2025: Dict[str, List[Tuple[date, date]]] = {
+    "A": [
+        (date(2024, 10, 19), date(2024, 11, 3)),
+        (date(2024, 12, 21), date(2025, 1, 5)),
+        (date(2025, 2, 8), date(2025, 2, 23)),
+        (date(2025, 4, 5), date(2025, 4, 20)),
+        (date(2025, 7, 5), date(2025, 8, 31)),
+    ],
+    "B": [
+        (date(2024, 10, 19), date(2024, 11, 3)),
+        (date(2024, 12, 21), date(2025, 1, 5)),
+        (date(2025, 2, 22), date(2025, 3, 9)),
+        (date(2025, 4, 19), date(2025, 5, 4)),
+        (date(2025, 7, 5), date(2025, 8, 31)),
+    ],
+    "C": [
+        (date(2024, 10, 19), date(2024, 11, 3)),
+        (date(2024, 12, 21), date(2025, 1, 5)),
+        (date(2025, 2, 15), date(2025, 3, 2)),
+        (date(2025, 4, 12), date(2025, 4, 27)),
+        (date(2025, 7, 5), date(2025, 8, 31)),
+    ],
+}
+
+ALL_VACANCES = {
+    "2023-2024": VACANCES_2023_2024,
+    "2024-2025": VACANCES_2024_2025,
+}
+
+# ============================================================
+# MAPPING DR → ZONE DE VACANCES (À ADAPTER)
+# ============================================================
+DR_TO_ZONE = {
+    # === ZONE A ===
+    "Besancon": "A",
+    "Bordeaux": "A",
+    "Clermont-Ferrand": "A",
+    "Dijon": "A",
+    "Grenoble": "A",
+    "Lyon": "A",
+    "Limoges": "A",
+    "Poitiers": "A",
+    # === ZONE B ===
+    "Aix-Marseille": "B",
+    "Amiens": "B",
+    "Caen": "B",
+    "Lille": "B",
+    "Nantes": "B",
+    "Nice": "B",
+    "Orleans-Tours": "B",
+    "Reims": "B",
+    "Rennes": "B",
+    "Rouen": "B",
+    "Strasbourg": "B",
+    # === ZONE C ===
+    "Creteil": "C",
+    "Montpellier": "C",
+    "Nancy-Metz": "C",
+    "Paris": "C",
+    "Toulouse": "C",
+    "Versailles": "C",
+    # --- Ajoute tes codes DR ici ---
+    "AFC": "C",
+    "AFL": "B",
+}
+
+
+def get_zone_for_dr(dr: str) -> str:
+    return DR_TO_ZONE.get(dr, "C")
+
+
+def is_vacances(d: date, zone: str, vacances: Dict[str, List[Tuple[date, date]]]) -> bool:
+    for debut, fin in vacances.get(zone, []):
+        if debut <= d <= fin:
+            return True
+    return False
+
+
+def get_periode_vacances(d: date, vacances: Dict[str, List[Tuple[date, date]]]) -> str:
+    for zone in ["A", "B", "C"]:
+        for debut, fin in vacances.get(zone, []):
+            if debut <= d <= fin:
+                m = d.month
+                if m in [10, 11]: return "Toussaint"
+                elif m in [12, 1]: return "Noel"
+                elif m in [2, 3]: return "Hiver"
+                elif m in [4, 5]: return "Printemps"
+                elif m in [7, 8]: return "Ete"
+    return "Hors_vacances"
+
+
+def add_vacances_columns(df: pd.DataFrame, date_col: str = "Date", dr_col: str = "DR") -> pd.DataFrame:
+    """Ajoute zone_vacances, is_vacances_zone, periode_vacances au DataFrame."""
+    df = df.copy()
+    df["Date"] = pd.to_datetime(df[date_col]).dt.tz_localize(None)
+    df["zone_vacances"] = df[dr_col].apply(get_zone_for_dr)
+    
+    def get_annee_scolaire(d):
+        if d.month >= 9:
+            return f"{d.year}-{d.year+1}"
+        else:
+            return f"{d.year-1}-{d.year}"
+    
+    df["annee_scolaire"] = df["Date"].apply(get_annee_scolaire)
+    
+    def check_vacances(row):
+        d = row["Date"].date()
+        vac = ALL_VACANCES.get(row["annee_scolaire"], {})
+        return is_vacances(d, row["zone_vacances"], vac)
+    
+    df["is_vacances_zone"] = df.apply(check_vacances, axis=1)
+    
+    def get_periode(row):
+        d = row["Date"].date()
+        vac = ALL_VACANCES.get(row["annee_scolaire"], {})
+        return get_periode_vacances(d, vac)
+    
+    df["periode_vacances"] = df.apply(get_periode, axis=1)
+    
+    return df