Splits exceptions in common + client

This commit is contained in:
Sergi Delgado Segura
2020-04-09 17:49:46 +02:00
parent b3841c1da6
commit 80e72f089d
4 changed files with 44 additions and 48 deletions

View File

@@ -1,32 +1,4 @@
class BasicException(Exception):
def __init__(self, msg, **kwargs):
self.msg = msg
self.kwargs = kwargs
def __str__(self):
if len(self.kwargs) > 2:
params = "".join("{}={}, ".format(k, v) for k, v in self.kwargs.items())
# Remove the extra 2 characters (space and comma) and add all data to the final message.
message = self.msg + " ({})".format(params[:-2])
else:
message = self.msg
return message
def to_json(self):
response = {"error": self.msg}
response.update(self.kwargs)
return response
class InvalidParameter(BasicException):
"""Raised when a command line parameter is invalid (either missing or wrong)"""
class InvalidKey(BasicException):
"""Raised when there is an error loading the keys"""
from common.exceptions import BasicException
class TowerConnectionError(BasicException):