fixing and improving the nps computation

gip-inclusion · Nov 7, 2023 · 816bb19 · 816bb19
1 parent f6b7a95
commit 816bb19
Showing 1 changed file with 31 additions and 8 deletions.
diff --git a/dags/nps_gip.py b/dags/nps_gip.py
@@ -37,14 +37,37 @@ def store_gsheets(**kwargs):
         for name, pub_sheet_url in Variable.get("NPS_NAME_PUB_SHEET_URL_TUPLES", deserialize_json=True):
             print(f"reading {name=} at {pub_sheet_url=}")
             sheet_df = pd.read_csv(pub_sheet_url)
-            sheet_df.rename(
-                columns={
-                    "Submitted at": "Date",
-                    "Dates": "Date",
-                    sheet_df.columns[-1]: "Recommandation",
-                },
-                inplace=True,
-            )
+            column_name = "Quelle est la probabilité que vous recommandiez"
+            column_name += " La communauté de l'inclusion à un collègue, partenaire ou homologue ?"
+
+            # Identify the last column that contains an integer
+            last_integer_column = None
+            for column in sheet_df.columns:
+                if sheet_df[column].dtype == "int64":
+                    last_integer_column = column
+
+            if last_integer_column is not None:
+                # Rename the last integer column
+                sheet_df.rename(
+                    columns={
+                        "Submitted at": "Date",
+                        "Dates": "Date",
+                        last_integer_column: "Recommandation",
+                    },
+                    inplace=True,
+                )
+            else:
+                if column_name in sheet_df.columns:
+                    sheet_df.rename(
+                        columns={
+                            "Submitted at": "Date",
+                            "Dates": "Date",
+                            column_name: "Recommandation",
+                        },
+                        inplace=True,
+                    )
+                else:
+                    raise Exception("Our colleagues messed up the columns :(")
             sheet_df["Produit"] = name
             sheet_df = sheet_df[["Date", "Recommandation", "Produit"]]
             df = pd.concat([df, sheet_df])