Gate index-autometa-tables cron to prod to avoid prod/staging index race

Author: louije

cron/index-autometa-tables/CRON.md

@@ -6,6 +6,8 @@ timeout: 3600
 
 Recrée les index d'`autometa_tables_db` et rafraîchit les statistiques (`ANALYZE`). Les tables sont reconstruites périodiquement par le DAG `populate_matometa_db` de pilotage-airflow (`if_exists="replace"`), ce qui supprime les index ; ce cron les recrée après le chargement. La liste des index vit dans `cron.py` (`INDEXES`).
 
+Prod uniquement : prod et staging visent la **même** `autometa_tables_db`. Deux runs concurrents (un par environnement) font courir `CREATE INDEX IF NOT EXISTS` et provoquent des collisions `pg_class` (`UniqueViolation`). Le script ne s'exécute donc que si `AUTOMETA_ENV == "prod"` ; ailleurs il sort immédiatement.
+
 Tolérance aux pannes : chaque statement s'exécute en AUTOCOMMIT ; un échec isolé (table absente, colonne renommée) est logué sans bloquer les suivants. Le script raise en fin de run si au moins un statement a échoué — `web.cron` remonte `failure` à Sentry et Slack.
 
 Si le DAG termine après l'heure du cron, les index de la journée ne réapparaissent qu'au run suivant (ou via un déclenchement manuel depuis `/cron`).

cron/index-autometa-tables/cron.py

@@ -106,6 +106,12 @@ def main() -> None:
         logger.info("AUTOMETA_TABLES_DATABASE_URL not configured; skipping")
         return
 
+    # Why: prod et staging partagent la même autometa_tables_db ; deux runs concurrents font
+    # courir CREATE INDEX IF NOT EXISTS (collision pg_class). Seul prod entretient les index.
+    if config.AUTOMETA_ENV != "prod":
+        logger.info("AUTOMETA_ENV=%s (not prod); skipping to avoid racing prod on the shared DB", config.AUTOMETA_ENV)
+        return
+
     # Why: AUTOCOMMIT — chaque statement est indépendant, un échec n'avorte pas les suivants.
     engine = create_engine(
         config.AUTOMETA_TABLES_DATABASE_URL,

tests/test_index_autometa_tables.py

@@ -67,8 +67,24 @@ def test_main_skips_without_database_url(mocker):
     create_engine.assert_not_called()
 
 
+@pytest.mark.parametrize(
+    ("env_value", "should_run"),
+    [("prod", True), ("staging", False), ("live", False), ("dev", False)],
+)
+def test_main_runs_only_on_prod(mocker, env_value, should_run):
+    mocker.patch.object(cron.config, "AUTOMETA_TABLES_DATABASE_URL", "postgresql://example")
+    mocker.patch.object(cron.config, "AUTOMETA_ENV", env_value)
+    engine, _ = make_engine(mocker)
+    create_engine = mocker.patch.object(cron, "create_engine", return_value=engine)
+
+    cron.main()
+
+    assert create_engine.called is should_run
+
+
 def test_main_executes_all_statements(mocker):
     mocker.patch.object(cron.config, "AUTOMETA_TABLES_DATABASE_URL", "postgresql://example")
+    mocker.patch.object(cron.config, "AUTOMETA_ENV", "prod")
     engine, conn = make_engine(mocker)
     mocker.patch.object(cron, "create_engine", return_value=engine)
 
@@ -79,6 +95,7 @@ def test_main_executes_all_statements(mocker):
 
 def test_main_continues_after_failures_then_raises(mocker):
     mocker.patch.object(cron.config, "AUTOMETA_TABLES_DATABASE_URL", "postgresql://example")
+    mocker.patch.object(cron.config, "AUTOMETA_ENV", "prod")
 
     def execute(clause):
         if '"candidats"' in str(clause) and "CREATE INDEX" in str(clause):