Added stage tests and tidied up rescue

This commit is contained in:
Joel Collins 2020-11-06 15:34:34 +00:00
parent 70437212b1
commit 0989a50558
4 changed files with 69 additions and 36 deletions

View file

@ -18,19 +18,13 @@ class Source:
return self._message
class WarningSource:
def __init__(self, message):
self.message = message
class WarningSource(Source):
@property
def message(self):
print(bcolors.WARNING + self._message + bcolors.ENDC)
return bcolors.WARNING + self._message + bcolors.ENDC
class ErrorSource:
def __init__(self, message):
self.message = message
class ErrorSource(Source):
@property
def message(self):
print(bcolors.FAIL + self._message + bcolors.ENDC)
return bcolors.FAIL + self._message + bcolors.ENDC