#!/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)