From bd6e57b3c61f929e2915ae3d8363ea9d9cc04068 Mon Sep 17 00:00:00 2001 From: onyx-and-iris Date: Mon, 7 Aug 2023 16:31:08 +0100 Subject: [PATCH] define message attribute for VBANCMD error classes override str magic method --- vban_cmd/error.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/vban_cmd/error.py b/vban_cmd/error.py index 4c635c6..a655e15 100644 --- a/vban_cmd/error.py +++ b/vban_cmd/error.py @@ -1,6 +1,13 @@ class VBANCMDError(Exception): """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): """Exception raised when connection/timeout errors occur"""