diff --git a/scylla_stress/__init__.py b/scylla_stress/__init__.py index 94fe914..1d7a004 100644 --- a/scylla_stress/__init__.py +++ b/scylla_stress/__init__.py @@ -1 +1,5 @@ +""" +Init file for scylla_stress +""" + from scylla_stress.scylla_stress import main diff --git a/scylla_stress/exceptions.py b/scylla_stress/exceptions.py index 2c5fecf..bebb082 100644 --- a/scylla_stress/exceptions.py +++ b/scylla_stress/exceptions.py @@ -2,9 +2,10 @@ Custom exceptions defined by the user """ + class RegexNotFound(Exception): """ - A class for custom exceptions + A class for custom Regex exception """ def __init__(self, value: str = ''): @@ -13,6 +14,9 @@ def __init__(self, value: str = ''): class DockerDaemonOff(Exception): + """ + A class for custom Docker Daemon Off exception + """ def __init__(self, value: str = ''): self.message = f"Docker daemon is probably not running or container name is invalid. Details:\n{value}" super().__init__(self.message)