traceback.py 835 B

12345678910111213141516171819202122232425262728293031323334
  1. #!/usr/bin/env python
  2. import sys,traceback,os
  3. sys.path.insert(0,'.')
  4. if 'TMUX' in os.environ: del os.environ['TMUX']
  5. os.environ['MMGEN_TRACEBACK'] = '1'
  6. tb_source = open(sys.argv[1])
  7. tb_file = os.path.join(os.environ['PWD'],'my.err')
  8. def process_exception(es):
  9. l = traceback.format_exception(*es)
  10. l_save = l[:]
  11. exc = l.pop()
  12. if exc[:11] == 'SystemExit:': l.pop()
  13. def red(s): return '{e}[31;1m{}{e}[0m'.format(s,e='\033')
  14. def yellow(s): return '{e}[33;1m{}{e}[0m'.format(s,e='\033')
  15. sys.stdout.write('{}{}'.format(yellow(''.join(l)),red(exc)))
  16. with open(tb_file,'w') as f:
  17. f.write(''.join(l_save))
  18. try:
  19. sys.argv.pop(0)
  20. exec tb_source
  21. except SystemExit:
  22. # pass
  23. e = sys.exc_info()
  24. if int(str(e[1])) != 0:
  25. process_exception(e)
  26. sys.exit(int(str(e[1])))
  27. except:
  28. e = sys.exc_info()
  29. process_exception(e)
  30. sys.exit(1)