mmgen-wallet/scripts/traceback_run.py

#!/usr/bin/env python3
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]).read()
tb_file = os.path.join(os.environ['PWD'],'my.err')

try: os.unlink(os.path.join(repo_root,tb_file))
except: pass

def process_exception():
	l = traceback.format_exception(*sys.exc_info())
	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 as e:
	if e.code != 0:
		process_exception()
	sys.exit(e.code)
except Exception as e:
	process_exception()
	sys.exit(e.mmcode if hasattr(e,'mmcode') else e.code if hasattr(e,'code') else 1)
else:
	print('else: '+repr(sys.exc_info()))
finally:
	print('finally: '+repr(sys.exc_info()))