12345678910111213141516171819202122232425262728293031323334 |
- #!/usr/bin/env python
- import sys,traceback,os
- sys.path.insert(0,'.')
- if 'TMUX' in os.environ: del os.environ['TMUX']
- os.environ['MMGEN_TRACEBACK'] = '1'
- tb_source = open(sys.argv[1])
- tb_file = os.path.join(os.environ['PWD'],'my.err')
- def process_exception(es):
- l = traceback.format_exception(*es)
- l_save = l[:]
- exc = l.pop()
- if exc[:11] == 'SystemExit:': l.pop()
- def red(s): return '{e}[31;1m{}{e}[0m'.format(s,e='\033')
- def yellow(s): return '{e}[33;1m{}{e}[0m'.format(s,e='\033')
- sys.stdout.write('{}{}'.format(yellow(''.join(l)),red(exc)))
- with open(tb_file,'w') as f:
- f.write(''.join(l_save))
- try:
- sys.argv.pop(0)
- exec tb_source
- except SystemExit:
- # pass
- e = sys.exc_info()
- if int(str(e[1])) != 0:
- process_exception(e)
- sys.exit(int(str(e[1])))
- except:
- e = sys.exc_info()
- process_exception(e)
- sys.exit(1)
|