def eval_and_exit(code: str):
""" Execute the code and exit according to the last expression:
- False: exitcode 1
- True/None: exitcode 0
- Else: print it and exitcode 0
"""
from io import StringIO
from pysh.transforms import Compiler
compiler = Compiler(EVAL_TRANSFORMS)
fn = compiler.compile(StringIO(code), 'pysh-eval')
try:
result = fn()
except Exception as ex:
print('{}: {}'.format(ex.__class__.__name__, ex), file=sys.stderr)
raise SystemExit(3)
if result is False:
raise SystemExit(1)
if result is not True and result is not None:
print('{}'.format(result))
raise SystemExit(0)
def main(argv=None):
version = '{} ({} {} - {} {})'.format(pysh.__version__,
platform.python_implementation(),
platform.python_version(),
platform.system(),
platform.machine())
#TODO: When compiling code from stdin a traceback won't show the lines
# since Python cannot open stdin again. We should investigate if
# we can keep a copy around and use it for tracebacks.
#TODO: `--transform foo --transform bar` yields ['foo', 'bar', 'bar']
opts = dict(version=pysh.__version__, program=PurePath(sys.argv[0]).name)
doc = re.sub(r'%([A-Za-z_]+)%', lambda m: opts[m.group(1)], __doc__)
args = docopt(doc, help=False, version=version, argv=argv)
# print(args)
if args['--help']:
symbol = args['SYMBOL']
if symbol:
if symbol not in pysh.__dict__:
print('Unknown symbol {}'.format(symbol), file=sys.stderr)
raise SystemExit(1)
help(pysh.__dict__[symbol])
else:
print(doc, file=sys.stderr)
raise SystemExit(0)
# Tune log level based on flags
if args['--verbose']:
logging.getLogger('').setLevel(logging.INFO)
elif args['--debug']:
logging.getLogger('').setLevel(logging.DEBUG)
elif args['--quiet']:
logging.getLogger('').setLevel(logging.CRITICAL)
if args['--eval'] is not None:
eval_and_exit(args['--eval'])
if not args['FILE']:
args['FILE'] = '/dev/stdin'
with open(args['FILE']) as fd:
transforms = list(SCRIPT_TRANSFORMS)
for t in args['--transform']:
#TODO: implement properly :)
if t.startswith('-'):
transforms = [x for x in transforms if not x.endswith(t[1:])]
else:
transforms.append(t)
from pysh.transforms import Compiler
compiler = Compiler(transforms)
fn = compiler.compile(fd, args['FILE'])
result = fn()
if result: # in case autoexpr is used
print(result)
def lex(modules, code):
comp = Compiler(modules)
return comp.lex(StringIO(dedent(code))).read()
def auto(modules, code):
comp = Compiler(list(modules) + ['autoreturn'])
func = comp.compile(dedent(code))
return func()
def comp(modules, code):
comp = Compiler(modules)
return comp.compile(StringIO(dedent(code)))
def parse(modules, code):
comp = Compiler(modules)
return comp.parse(StringIO(dedent(code)))