From 3e51f03e399f7d2e0e75b4221485c6662a30cec6 Mon Sep 17 00:00:00 2001 From: Eric Zhou Date: Wed, 15 Nov 2023 15:17:39 -0800 Subject: [PATCH] use try since write will fail on aws lambda --- capitoltrades/CapitolTrades.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/capitoltrades/CapitolTrades.py b/capitoltrades/CapitolTrades.py index 09fd4b8..d0e3be5 100644 --- 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 \ No newline at end of file