add 'NPKMagicBytesError' for #1

Author: botlabsDev

npkpy/common.py

@@ -66,5 +66,9 @@ class NPKIdError(BaseException):
     pass
 
 
+class NPKMagicBytesError(BaseException):
+    pass
+
+
 class NPKError(BaseException):
     pass

npkpy/npk/npk.py

@@ -1,7 +1,7 @@
 import struct
 from pathlib import Path
 
-from npkpy.common import NPKError, NPKIdError
+from npkpy.common import NPKError, NPKIdError, NPKMagicBytesError
 from npkpy.npk.npk_constants import CNT_HANDLER
 from npkpy.npk.cnt_basic import BYTES_LEN_CNT_ID, BYTES_LEN_CNT_PAYLOAD_LEN
 from npkpy.npk.npk_file_basic import FileBasic
@@ -99,4 +99,4 @@ def __get_cnt(self, offset):
 
     def _check_magic_bytes(self, error_msg):
         if not self.pck_magic_bytes == MAGIC_BYTES:
-            raise NPKError(error_msg)
+            raise NPKMagicBytesError(error_msg)

tests/npk_test.py

@@ -3,7 +3,7 @@
 import unittest
 from pathlib import Path
 
-from npkpy.common import NPKError, NPKIdError
+from npkpy.common import NPKError, NPKIdError, NPKMagicBytesError
 from npkpy.npk.npk import Npk
 from npkpy.npk.pck_header import PckHeader
 from tests.constants import DummyHeaderCnt, MAGIC_BYTES, get_dummy_npk_binary
@@ -18,7 +18,7 @@ def setUp(self) -> None:
     def test_fileIsNoNpkFile(self):
         self.npkFile.write_bytes(b"NoMagicBytesAtHeadOfFile")
 
-        with self.assertRaises(NPKError) as e:
+        with self.assertRaises(NPKMagicBytesError) as e:
             _ = Npk(self.npkFile).pck_magic_bytes
         self.assertEqual(e.exception.args[0], "Magic bytes not found in Npk file")