From 523d8eaa26fc04c1b6437607079fe9bd7d084d51 Mon Sep 17 00:00:00 2001 From: Jani Nikula Date: Sun, 1 Sep 2024 14:25:55 +0300 Subject: [PATCH] parser: add support for returning just basename in clang diagnostics This will be useful for parser tests, as we'll be able to reuse this instead of duplicating the logic for filename or line being None. --- src/hawkmoth/parser.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/hawkmoth/parser.py b/src/hawkmoth/parser.py index 01ea40b7..e8f86ddf 100644 --- a/src/hawkmoth/parser.py +++ b/src/hawkmoth/parser.py @@ -65,12 +65,14 @@ class ParserError: line: int message: str - def get_message(self): + def get_message(self, basename=False): if self.filename: + filename = os.path.basename(self.filename) if basename else self.filename + if self.line is not None: - return f'{self.filename}:{self.line}: {self.message}' + return f'{filename}:{self.line}: {self.message}' else: - return f'{self.filename}: {self.message}' + return f'{filename}: {self.message}' else: return f'{self.message}'