Extend ConfigFileNotFoundError

Signed-off-by: Václav Valíček <valicek1994@gmail.com>
This commit is contained in:
Václav Valíček 2022-06-26 01:15:44 +02:00
parent 97054b4fcc
commit 7293289a1a
Signed by: valicek
GPG Key ID: FF05BDCA0C73BB31
2 changed files with 8 additions and 6 deletions

View File

@ -5,16 +5,18 @@ import os
class ConfigFileNotFoundError(OSError): class ConfigFileNotFoundError(OSError):
def __str__(self): def __str__(self):
if len(self.args) == 2: if len(self.args) in [1, 2]:
path: str = "" path: str = ""
filename: str = ""
if isinstance(self.args[0], PosixPath): if isinstance(self.args[0], PosixPath):
path = str(PosixPath(self.args[0])) path = str(PosixPath(self.args[0]))
else: else:
path = str(self.args[0]) path = str(self.args[0])
if len(self.args) == 2:
filename: str = ""
filename = str(self.args[1]) filename = str(self.args[1])
path = os.path.join(path, filename) path = os.path.join(path, filename)
return f"Config file does not exist: {path}" return f"Config file does not exist: {path}"
else: else:
return super.__str__(self) return super.__str__(self)

View File

@ -14,9 +14,9 @@ def test_return():
def test_str_single_arg(): def test_str_single_arg():
x = ConfigFileNotFoundError("/tmp") x = ConfigFileNotFoundError("/tmp/cfg.cfg")
s = x.__str__() s = x.__str__()
assert s == "ConfigFileNotFoundError('/tmp')" assert s == "Config file does not exist: /tmp/cfg.cfg"
def test_return_empty(): def test_return_empty():