From 969cc65ae5cea7a275736477dfc495ea50fd7f7c Mon Sep 17 00:00:00 2001 From: Mykola Grymalyuk Date: Sat, 27 May 2023 11:52:02 -0600 Subject: [PATCH] integrity_verification.py: Add extra error handling for missing/botched files --- resources/integrity_verification.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/resources/integrity_verification.py b/resources/integrity_verification.py index 0144f779f..b249d02d0 100644 --- a/resources/integrity_verification.py +++ b/resources/integrity_verification.py @@ -100,6 +100,18 @@ class ChunklistVerification: self.status = ChunklistStatus.FAILURE return + if not Path(self.file_path).exists(): + self.error_msg = f"File {self.file_path} does not exist" + self.status = ChunklistStatus.FAILURE + logging.info(self.error_msg) + return + + if not Path(self.file_path).is_file(): + self.error_msg = f"File {self.file_path} is not a file" + self.status = ChunklistStatus.FAILURE + logging.info(self.error_msg) + return + with self.file_path.open("rb") as f: for chunk in self.chunks: self.current_chunk += 1