remove if file exists

meedan · Jan 3, 2024 · b00b83a · b00b83a
1 parent 3373365
commit b00b83a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lib/model/video.py b/lib/model/video.py
@@ -57,6 +57,7 @@ def process(self, video: schemas.Message) -> schemas.GenericItem:
             )
             s3.upload_file_to_s3(self.tmk_bucket(), self.tmk_file_path(video_filename))
         finally:
-            os.remove(self.tmk_file_path(video_filename))
-            os.remove(temp_file_name)
+            for file_path in [self.tmk_file_path(video_filename), temp_file_name]:
+                if os.path.exists(file_path):
+                    os.remove(file_path)
         return dict(**video.dict(), **{"folder": self.tmk_bucket(), "filepath": self.tmk_file_path(video_filename), "hash_value": hash_value})