define message attribute for VBANCMD error classes

override str magic method
This commit is contained in:
onyx-and-iris 2023-08-07 16:31:08 +01:00
parent eed036ca03
commit bd6e57b3c6

View File

@ -1,6 +1,13 @@
class VBANCMDError(Exception): class VBANCMDError(Exception):
"""Base VBANCMD Exception class. Raised when general errors occur""" """Base VBANCMD Exception class. Raised when general errors occur"""
def __init__(self, msg):
self.message = msg
super().__init__(self.message)
def __str__(self):
return f"{type(self).__name__}: {self.message}"
class VBANCMDConnectionError(VBANCMDError): class VBANCMDConnectionError(VBANCMDError):
"""Exception raised when connection/timeout errors occur""" """Exception raised when connection/timeout errors occur"""