diff --git a/monkey/monkey_island/cc/repository/__init__.py b/monkey/monkey_island/cc/repository/__init__.py index 3a8c552e3..52e6f6873 100644 --- a/monkey/monkey_island/cc/repository/__init__.py +++ b/monkey/monkey_island/cc/repository/__init__.py @@ -1,4 +1,4 @@ -from .errors import RetrievalError, StorageError +from .errors import RemovalError, RetrievalError, StorageError from .file_storage import FileNotFoundError, IFileRepository, LocalStorageFileRepository from .i_agent_binary_repository import IAgentBinaryRepository from .agent_binary_repository import AgentBinaryRepository diff --git a/monkey/monkey_island/cc/repository/errors.py b/monkey/monkey_island/cc/repository/errors.py index 586f23cb2..aeb9fa23c 100644 --- a/monkey/monkey_island/cc/repository/errors.py +++ b/monkey/monkey_island/cc/repository/errors.py @@ -1,3 +1,11 @@ +class RemovalError(RuntimeError): + """ + Raised when a repository encounters an error while attempting to remove data. + """ + + pass + + class RetrievalError(RuntimeError): """ Raised when a repository encounters an error while attempting to retrieve data.