From 4e755bbd2fc7265c077df0b36ff4a8f3970e84e3 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 19 Jul 2022 18:12:41 +0530 Subject: [PATCH] Island: Move SizeError to its own file --- .../monkey_island/cc/server_utils/encryption/__init__.py | 1 + .../cc/server_utils/encryption/encryption_key_types.py | 7 +------ monkey/monkey_island/cc/server_utils/encryption/errors.py | 6 ++++++ 3 files changed, 8 insertions(+), 6 deletions(-) create mode 100644 monkey/monkey_island/cc/server_utils/encryption/errors.py diff --git a/monkey/monkey_island/cc/server_utils/encryption/__init__.py b/monkey/monkey_island/cc/server_utils/encryption/__init__.py index 04381ee35..3ef328289 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/__init__.py +++ b/monkey/monkey_island/cc/server_utils/encryption/__init__.py @@ -15,3 +15,4 @@ from .data_store_encryptor import ( reset_datastore_encryptor, ) from .encryption_key_types import EncryptionKey32Bytes +from .errors import SizeError diff --git a/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py index 175dbad75..3f7c2252f 100644 --- a/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py +++ b/monkey/monkey_island/cc/server_utils/encryption/encryption_key_types.py @@ -1,9 +1,4 @@ -class SizeError(Exception): - """ - Raised when an argument is not of the expected size during object creation. - """ - - pass +from . import SizeError class EncryptionKey32Bytes(bytes): diff --git a/monkey/monkey_island/cc/server_utils/encryption/errors.py b/monkey/monkey_island/cc/server_utils/encryption/errors.py new file mode 100644 index 000000000..0364ee084 --- /dev/null +++ b/monkey/monkey_island/cc/server_utils/encryption/errors.py @@ -0,0 +1,6 @@ +class SizeError(Exception): + """ + Raised when an argument is not of the expected size during object creation. + """ + + pass