view tests/filtertraceback.py @ 44436:09f3e003fc2a

phabricator: avoid a stacktrace when command arguments are missing

Previously, the TypeError wasn't properly converted to a SignatureError when
improper arguments were supplied to the inner function, because the stack depth
is 2 inside the vcrcommand decorator.  The `__name__` and `__doc__` attributes
need to be reassigned to the new wrapper so that the help summary is available.

Differential Revision: https://phab.mercurial-scm.org/D8209

#!/usr/bin/env python

# Filters traceback lines from stdin.

from __future__ import absolute_import, print_function

import sys

state = 'none'

for line in sys.stdin:
    if state == 'none':
        if line.startswith('Traceback '):
            state = 'tb'

    elif state == 'tb':
        if line.startswith('  File '):
            state = 'file'
            continue

        elif not line.startswith(' '):
            state = 'none'

    elif state == 'file':
        # Ignore lines after "  File "
        state = 'tb'
        continue

    print(line, end='')