fix stack trace linking

This commit is contained in:
Aaron Munger 2023-11-06 09:43:57 -08:00 committed by Aaron Munger
parent e0e31d3b61
commit 6afac1d3f7
2 changed files with 14 additions and 2 deletions

View file

@ -31,7 +31,7 @@ export function formatStackTrace(stack: string) {
const formatSequence = /\u001b\[.+?m/g;
const fileRegex = /File\s+(?:\u001b\[.+?m)?(.+):(\d+)/;
const lineNumberRegex = /((?:\u001b\[.+?m)?[ ->]*?)(\d+)(.*)/;
const lineNumberRegex = /^((?:\u001b\[.+?m)?[ ->]*?)(\d+)(.*)/;
const cellRegex = /(?<prefix>Cell\s+(?:\u001b\[.+?m)?In\s*\[(?<executionCount>\d+)\],\s*)(?<lineLabel>line (?<lineNumber>\d+)).*/;
// older versions of IPython ~8.3.0
const inputRegex = /(?<prefix>Input\s+?(?:\u001b\[.+?m)(?<cellLabel>In\s*\[(?<executionCount>\d+)\]))(?<postfix>.*)/;
@ -41,7 +41,7 @@ function isIpythonStackTrace(stack: string) {
}
function stripFormatting(text: string) {
return text.replace(formatSequence, '');
return text.replace(formatSequence, '').trim();
}
type cellLocation = { kind: 'cell'; path: string };

View file

@ -88,4 +88,16 @@ suite('StackTraceHelper', () => {
assert.ok(!/<a href=.*>\d<\/a>/.test(formatted), formatted);
});
test('IPython stack without line numbers are not linkified', () => {
const stack =
'\u001b[1;36m Cell \u001b[1;32mIn[6], line 1\u001b[1;36m\u001b[0m\n' +
'\u001b[1;33m print(\u001b[0m\n' +
'\u001b[1;37m ^\u001b[0m\n' +
'\u001b[1;31mSyntaxError\u001b[0m\u001b[1;31m:\u001b[0m incomplete input\n';
const formattedLines = formatStackTrace(stack).split('\n');
assert.ok(/<a href='vscode-notebook-cell.*>/.test(formattedLines[0]), 'line should contain a link: ' + formattedLines[0]);
formattedLines.slice(1).forEach(line => assert.ok(!/<a href=.*>/.test(line), 'line should not contain a link: ' + line));
});
});