diff options
author | Nate Sesti <sestinj@gmail.com> | 2023-06-16 09:22:08 -0700 |
---|---|---|
committer | Nate Sesti <sestinj@gmail.com> | 2023-06-16 09:22:08 -0700 |
commit | 06910ad84e8da33fa6d5fe66cd69293ce8986d84 (patch) | |
tree | ff43ae2d4033856f6f543acd7ab550078604cf8c | |
parent | f36cd52da559d34753ceb597235d5519686c83cf (diff) | |
download | sncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.tar.gz sncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.tar.bz2 sncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.zip |
improved traceback parser
-rw-r--r-- | continuedev/src/continuedev/libs/util/traceback_parsers.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/libs/util/traceback_parsers.py b/continuedev/src/continuedev/libs/util/traceback_parsers.py index 4e7a24e0..1885cc79 100644 --- a/continuedev/src/continuedev/libs/util/traceback_parsers.py +++ b/continuedev/src/continuedev/libs/util/traceback_parsers.py @@ -1,6 +1,11 @@ +PYTHON_TRACEBACK_PREFIX = "Traceback (most recent call last):" + + def get_python_traceback(output: str) -> str: - if "Traceback (most recent call last):" in output or "SyntaxError" in output: - return output + if PYTHON_TRACEBACK_PREFIX in output: + return PYTHON_TRACEBACK_PREFIX + output.split(PYTHON_TRACEBACK_PREFIX)[-1] + elif "SyntaxError" in output: + return "SyntaxError" + output.split("SyntaxError")[-1] else: return None |