iframe-proxy | Sunbelt Computer Software

129 lines (103 loc) · 4.29 KB
# -*- coding: utf-8 -*-
"""Console script for grumpy_tools."""
from StringIO import StringIO
from pkg_resources import resource_filename, Requirement, DistributionNotFound
import logging
logger = logging.getLogger(__package__)
import click
import click_log
from click_log.core import ColorFormatter
from click_default_group import DefaultGroup
ColorFormatter.colors['info'] = {'fg': 'green'}
click_log.basic_config(logger)
logger.propagate = True
from . import grumpc, grumprun, pydeps
@click.group('grumpy', cls=DefaultGroup, default='run', default_if_no_args=True)
@click_log.simple_verbosity_option(logger, default='WARNING')
def main(args=None):
    Console script for grumpy_tools.
    The default command `grumpy run` will ran if no other selected.
    It mimics the CPython options, when possible and applicable.
    Please take a look on `grumpy run --help` for its implemented options.
    Example: all the following lines outputs Hello on the STDOUT\n
        $ python -c 'print("Hello")'\n
        $ grumpy -c 'print("Hello")'\n
        $ grumpy run -c 'print("Hello")'
@main.command('transpile')
@click.argument('script', type=click.File('rb'))
@click.option('-m', '-modname', '--modname', default='__main__', help='Python module name')
@click.option('--pep3147', is_flag=True, help='Put the transpiled outputs on a __pycache__ folder')
def transpile(script=None, modname=None, pep3147=False):
    Translates the python SCRIPT file to Go, then prints to stdout
    _ensure_gopath(raises=False)
    output = grumpc.main(stream=script, modname=modname, pep3147=pep3147)
    click.echo(output)
    sys.exit(0)
@main.command('run')
@click.argument('file', required=False, type=click.File('rb'))
@click.option('-c', '--cmd', help='Program passed in as string')
@click.option('-m', '-modname', '--modname', help='Run run library module as a script')
@click.option('--keep-main', is_flag=True,
              help='Do not clear the temporary folder containing the transpilation result of main script')
def run(file=None, cmd=None, modname=None, keep_main=False, pep3147=True):
    _ensure_gopath()
    if modname:
        stream = None
    elif file:
        stream = StringIO(file.read())
    elif cmd:
        stream = StringIO(cmd)
    else:   # Read from STDIN
        stdin = click.get_text_stream('stdin')
        if stdin.isatty():  # Interactive terminal -> REPL
            click.echo('Python REPL is not (yet) available.'
                       '\nTry to pipe in your code, or pass --help for more options',
                       err=True)
            sys.exit(1)
        else:               # Piped terminal
            stream = StringIO(stdin.read())
    if stream is not None:
        stream.seek(0)
        stream.name = '__main__.py'
    result = grumprun.main(stream=stream, modname=modname, pep3147=pep3147, clean_tempfolder=(not keep_main))
    sys.exit(result)
@main.command('depends')
@click.argument('script')
@click.option('-m', '-modname', '--modname', default='__main__', help='Python module name')
def depends(script=None, modname=None):
    Discover with modules are needed to run the 'script' provided
    _ensure_gopath()
    package_dir = os.path.dirname(os.path.abspath(script))
    output = pydeps.main(script=script, modname=modname, package_dir=package_dir)
    for dep in output:
        click.echo(dep)
    sys.exit(0)
def _ensure_gopath(raises=True):
    environ_gopath = os.environ.get('GOPATH', '')
        runtime_gopath = resource_filename(
            Requirement.parse('grumpy-runtime'),
            'grumpy_runtime/data/gopath',
    except DistributionNotFound:
        runtime_gopath = None
    if runtime_gopath and runtime_gopath not in environ_gopath:
        gopaths = environ_gopath.split(':') + [runtime_gopath]
        new_gopath = os.pathsep.join([p for p in gopaths if p])  # Filter empty ones
        if new_gopath:
            os.environ['GOPATH'] = new_gopath
    if raises and not runtime_gopath and not environ_gopath:
        raise click.ClickException("Could not found the Grumpy Runtime 'data/gopath' resource.\n"
                                   "Is 'grumpy-runtime' package installed?")
    logger.info("GOPATH: %s", os.environ.get("GOPATH"))
if __name__ == "__main__":
    import sys
    sys.exit(main())
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

cli.py

Latest commit

History

cli.py

File metadata and controls
