summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-06-16 09:22:08 -0700
committerNate Sesti <sestinj@gmail.com>2023-06-16 09:22:08 -0700
commit06910ad84e8da33fa6d5fe66cd69293ce8986d84 (patch)
treeff43ae2d4033856f6f543acd7ab550078604cf8c
parentf36cd52da559d34753ceb597235d5519686c83cf (diff)
downloadsncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.tar.gz
sncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.tar.bz2
sncontinue-06910ad84e8da33fa6d5fe66cd69293ce8986d84.zip
improved traceback parser
-rw-r--r--continuedev/src/continuedev/libs/util/traceback_parsers.py9
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