use try since write will fail on aws lambda

ericz1803 · Nov 15, 2023 · 3e51f03 · 3e51f03
1 parent 6c5239a
commit 3e51f03
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/capitoltrades/CapitolTrades.py b/capitoltrades/CapitolTrades.py
@@ -148,7 +148,11 @@ def update_data(self):
         print(f"Merged {len(new_data)} new records and {len(self.data)} old records into {len(merged_data)} total records.")
 
         script_dir = os.path.dirname(__file__)
-        with open(os.path.join(script_dir, self.__pkl_path), "wb") as fp:
-            pickle.dump(merged_data, fp)
+        try:
+            with open(os.path.join(script_dir, self.__pkl_path), "wb") as fp:
+                pickle.dump(merged_data, fp)
+        except Exception as e:
+            logging.error("Error saving pickle file: " + str(e))
+
         self.data = merged_data