61 lines
1.8 KiB
Python
Executable file
61 lines
1.8 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# Import as few modules and define as few names as possible at global level before exec'ing the
|
|
# file, as all names will be seen by the exec'ed file. To prevent name collisions, all names
|
|
# defined here should begin with 'traceback_run_'
|
|
|
|
import sys,os,time
|
|
|
|
def traceback_run_init():
|
|
import os
|
|
sys.path[0] = 'test' if os.path.dirname(sys.argv[1]) == 'test' else '.'
|
|
|
|
if 'TMUX' in os.environ: del os.environ['TMUX']
|
|
os.environ['MMGEN_TRACEBACK'] = '1'
|
|
os.environ['PYTHONPATH'] = '.'
|
|
|
|
of = 'my.err'
|
|
try: os.unlink(of)
|
|
except: pass
|
|
|
|
return of
|
|
|
|
def traceback_run_process_exception():
|
|
import traceback,re
|
|
lines = traceback.format_exception(*sys.exc_info()) # returns a list
|
|
|
|
pat = re.compile('File "<string>"')
|
|
repl = f'File "{traceback_run_execed_file}"'
|
|
lines = [pat.sub(repl,line,count=1) for line in lines]
|
|
|
|
exc = lines.pop()
|
|
if exc.startswith('SystemExit:'):
|
|
lines.pop()
|
|
|
|
if False: # was: if os.getenv('MMGEN_DISABLE_COLOR'):
|
|
sys.stdout.write('{}{}'.format(''.join(lines),exc))
|
|
else:
|
|
yellow = lambda s: f'\033[33;1m{s}\033[0m'
|
|
red = lambda s: f'\033[31;1m{s}\033[0m'
|
|
sys.stdout.write('{}{}'.format(yellow(''.join(lines)),red(exc)))
|
|
|
|
open(traceback_run_outfile,'w').write(''.join(lines+[exc]))
|
|
|
|
traceback_run_outfile = traceback_run_init()
|
|
traceback_run_tstart = time.time()
|
|
|
|
try:
|
|
sys.argv.pop(0)
|
|
traceback_run_execed_file = sys.argv[0]
|
|
exec(open(sys.argv[0]).read())
|
|
except SystemExit as e:
|
|
if e.code != 0:
|
|
traceback_run_process_exception()
|
|
sys.exit(e.code)
|
|
except Exception as e:
|
|
traceback_run_process_exception()
|
|
retval = e.mmcode if hasattr(e,'mmcode') else e.code if hasattr(e,'code') else 1
|
|
sys.exit(retval)
|
|
|
|
blue = lambda s: s if os.getenv('MMGEN_DISABLE_COLOR') else '\033[34;1m{}\033[0m'.format(s)
|
|
sys.stderr.write(blue('Runtime: {:0.5f} secs\n'.format(time.time() - traceback_run_tstart)))
|