final picklist fix part 2

team4099 · Mar 3, 2024 · f39b437 · f39b437
1 parent bd9e577
commit f39b437
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/page_managers/picklist_manager.py b/src/page_managers/picklist_manager.py
@@ -281,7 +281,7 @@ def write_to_notion(self, dataframe: DataFrame) -> None:
                     parent={"type": "database_id", "database_id": db_id},
                     properties={
                         column: {
-                            "number": dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]
+                            "number": data if (data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
                         } for column in dataframe.columns if column != "Team Number"
                     } | {
                         "Team Name": {"id": "title", "title": [{"text": {"content": team_name}}]},
@@ -316,7 +316,7 @@ def write_to_notion(self, dataframe: DataFrame) -> None:
                     parent={"type": "database_id", "database_id": db_id},
                     properties={
                        column: {
-                           "number": dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]
+                           "number": data if (data := dataframe[dataframe["Team Number"] == team_name][column].tolist()[0]) else 0
                        } for column in dataframe.columns if column != "Team Number"
                     } | {
                        "Team Name": {"id": "title", "title": [{"text": {"content": team_name}}]},