Exemplo n.º 1
0
def main(argv=None):

    argv = argv or sys.argv[1:]
    op = OptionParser(usage=(
        "Usage: assetgen [<path/to/assetgen.yaml> ...] [options]\n\n"
        "Note:\n"
        "    If you don't specify assetgen.yaml file paths, then `git\n"
        "    ls-files *assetgen.yaml` will be used to detect all config\n"
        "    files in the current repository. So you need to be inside\n"
        "    a git repository's working tree."
        ))

    op.add_option(
        '-v', '--version', action='store_true',
        help="show program's version number and exit"
        )

    op.add_option(
        '--clean', action='store_true', help="remove all generated files"
        )

    op.add_option(
        '--debug', action='store_true', help="set debug mode"
        )

    op.add_option(
        '--extension', action='append', dest='path',
        help="specify a python extension file (may be repeated)"
        )

    op.add_option(
        '--force', action='store_true', help="force rebuild of all files"
        )

    op.add_option(
        '--profile', dest='name', default='default',
        help="specify a profile to use"
        )

    op.add_option(
        '--watch', action='store_true',
        help="keep running assetgen on a loop"
        )

    autocomplete(op)
    options, files = op.parse_args(argv)

    if options.version:
        print 'assetgen 0.1'
        sys.exit()

    if options.debug:
        global DEBUG
        DEBUG = True

    clean = options.clean
    extensions = options.path
    force = options.force
    profile = options.name
    watch = options.watch

    if extensions:
        scope = globals()
        for ext in extensions:
            execfile(ext, scope, {})

    if files:
        for file in files:
            if not isfile(file):
                exit("Could not find %s" % file)

    if not files:
        if not is_git():
            op.print_help()
            sys.exit()
        root = SCMConfig().root
        files = run_command(
            ['git', 'ls-files', '*assetgen.yaml'], cwd=root
            ).strip().splitlines()
        if not files:
            op.print_help()
            sys.exit()
        files = [join(root, file) for file in files]

    files = [realpath(file) for file in files]

    if watch:
        mtime_cache = {}
        for file in files:
            mtime_cache[file] = stat(file)[ST_MTIME]

    generators = [AssetGenRunner(file, profile, force) for file in files]

    if clean:
        for assetgen in generators:
            assetgen.clean()
        sys.exit()

    if watch:
        while 1:
            try:
                for assetgen in generators:
                    assetgen.run()
                for idx, file in enumerate(files):
                    mtime = stat(file)[ST_MTIME]
                    if mtime > mtime_cache[file]:
                        mtime_cache[file] = mtime
                        generators[idx] = AssetGenRunner(file, profile, force)
                sleep(1)
            except AppExit:
                sleep(3)
            except KeyboardInterrupt:
                break
    else:
        try:
            for assetgen in generators:
                assetgen.run()
        except AppExit:
            sys.exit(1)
Exemplo n.º 2
0
def main(argv=None):
    """Handle the bolt command line call."""

    if argv is None:
        argv = sys.argv[1:]

    op = OptionParser(
        usage="bolt <command-1> <command-2> ... [options]",
        )

    op.add_option(
        '-v', '--version', action='store_true', default=False,
        help="show program's version number and exit"
        )

    op.add_option(
        '-f', dest='file', default="Boltfile",
        help="set the name or path of the bolt file [Boltfile]"
        )

    op.add_option(
        '-d', dest='defaults_file', default=_rc_path(),
        help="set the path of the defaults file [~/.bolt.yaml]"
        )

    op.add_option(
        '-i',  dest='identity', action='append', default=None,
        help="path to SSH private key file(s) -- may be repeated"
        )

    op.add_option(
        '--hide', metavar='LEVELS',
        help="comma-separated list of output levels to hide"
        )

    op.add_option(
        '--show', metavar='LEVELS',
        help="comma-separated list of output levels to show"
        )

    op.add_option(
        '--disable', metavar='HOOKS',
        help="comma-separated list of hooks to disable"
        )

    op.add_option(
        '--enable', metavar='HOOKS',
        help="comma-separated list of hooks to enable"
        )

    op.add_option(
        '--list', action='store_true', default=False,
        help="show the list of available tasks and exit"
        )

    op.add_option(
        '--no-pty', action='store_true', default=False,
        help="do not use pseudo-terminal in run/sudo"
        )

    options, args = op.parse_args(argv)
    setup_defaults(options.defaults_file)

    # Load the Boltfile.
    runner = init_task_runner(options.file, getcwd())

    # Autocompletion support.
    autocomplete_items = runner.tasks.keys()
    if 'autocomplete' in env:
        autocomplete_items += env.autocomplete

    autocomplete(op, ListCompleter(autocomplete_items))

    if options.version:
        print("bolt %s" % __version__)
        sys.exit()

    if options.no_pty:
        env.always_use_pty = False

    if options.identity:
        env.key_filename = options.identity

    split_string = lambda s: filter(None, map(str.strip, s.split(',')))

    # Handle output levels.
    if options.show:
        for level in split_string(options.show):
            output[level] = True

    if options.hide:
        for level in split_string(options.hide):
            output[level] = False

    if output.debug:
        print("Using Boltfile: %s" % runner.path)

    # Handle hooks related options.
    if options.disable:
        for hook in split_string(options.disable):
            DISABLED_HOOKS.append(hook)

    if options.enable:
        for hook in split_string(options.enable):
            ENABLED_HOOKS.append(hook)

    if options.list:
        print('\n'.join(sorted(runner.tasks)))
        sys.exit()

    tasks = []
    idx = 0

    # Parse command line arguments.
    for task in args:

        # Initialise variables.
        _args = []
        _kwargs = {}
        _ctx = None

        # Handle +env flags.
        if task.startswith('+'):
            if ':' in task:
                name, value = task[1:].split(':', 1)
                env[name] = value
            else:
                env[task[1:]] = True
            continue

        # Handle @context specifiers.
        if task.startswith('@'):
            if not idx:
                continue
            ctx = (task[1:],)
            existing = tasks[idx-1][3]
            if existing:
                new = list(existing)
                new.extend(ctx)
                ctx = tuple(new)
            tasks[idx-1][3] = ctx
            continue

        # Handle tasks with parameters.
        if ':' in task:
            task, argstr = task.split(':', 1)
            for pair in _escape_split(',', argstr):
                k, _, v = pair.partition('=')
                if _:
                    _kwargs[k] = v
                else:
                    _args.append(k)

        idx += 1
        task_name = task.replace('_', '-')

        if task_name not in runner.tasks:
            abort("Task not found:\n\n%s" % indent(task))

        tasks.append([task_name, _args, _kwargs, _ctx])

    if not tasks:
        runner.display_listing()
        sys.exit()

    runner.run(tasks)
Exemplo n.º 3
0
def main(argv=None):

    argv = argv or sys.argv[1:]
    op = OptionParser(usage=(
        "Usage: assetgen [<path/to/assetgen.yaml> ...] [options]\n\n"
        "Note:\n"
        "    If you don't specify assetgen.yaml file paths, then `git\n"
        "    ls-files *assetgen.yaml` will be used to detect all config\n"
        "    files in the current repository. So you need to be inside\n"
        "    a git repository's working tree.\n\n"
        "    And if you specify a URL as a `source`, then it will be\n"
        "    downloaded to ~/.assetgen -- you can override this by\n"
        "    setting the env variable $ASSETGEN_DOWNLOADS_DIRECTORY"))

    op.add_option('-v',
                  '--version',
                  action='store_true',
                  help="show program's version number and exit")

    op.add_option('--clean',
                  action='store_true',
                  help="remove all generated files")

    op.add_option('--debug', action='store_true', help="set debug mode")

    op.add_option('--extension',
                  action='append',
                  dest='path',
                  help="specify a python extension file (may be repeated)")

    op.add_option('--force',
                  action='store_true',
                  help="force rebuild of all files")

    op.add_option('--nuke',
                  action='store_true',
                  help="remove all generated and downloaded files")

    op.add_option('--profile',
                  dest='name',
                  default='default',
                  help="specify a profile to use")

    op.add_option('--watch',
                  action='store_true',
                  help="keep running assetgen on a loop")

    autocomplete(op)
    options, files = op.parse_args(argv)

    if options.version:
        print 'assetgen 0.2.2'
        sys.exit()

    if options.debug:
        global DEBUG
        DEBUG = True

    clean = options.clean
    extensions = options.path
    force = options.force
    nuke = options.nuke
    profile = options.name
    watch = options.watch

    if extensions:
        scope = globals()
        for ext in extensions:
            execfile(ext, scope, {})

    if files:
        for file in files:
            if not isfile(file):
                exit("Could not find %s" % file)

    if not files:
        if not is_git():
            op.print_help()
            sys.exit()
        root = SCMConfig().root
        files = run_command(['git', 'ls-files', '*assetgen.yaml'],
                            cwd=root).strip().splitlines()
        if not files:
            op.print_help()
            sys.exit()
        files = [join(root, file) for file in files]

    files = [realpath(file) for file in files]

    if watch:
        mtime_cache = {}
        for file in files:
            mtime_cache[file] = stat(file)[ST_MTIME]

    generators = [AssetGenRunner(file, profile, force) for file in files]

    if nuke:
        if isdir(DOWNLOADS_PATH):
            print "=> Removing:", DOWNLOADS_PATH
            rmtree(DOWNLOADS_PATH)
        clean = 1

    if clean:
        for assetgen in generators:
            assetgen.clean()
        sys.exit()

    if watch:
        while 1:
            try:
                for assetgen in generators:
                    assetgen.run()
                for idx, file in enumerate(files):
                    mtime = stat(file)[ST_MTIME]
                    if mtime > mtime_cache[file]:
                        mtime_cache[file] = mtime
                        generators[idx] = AssetGenRunner(file, profile, force)
                sleep(1)
            except AppExit:
                sleep(3)
            except KeyboardInterrupt:
                break
    else:
        try:
            for assetgen in generators:
                assetgen.run()
        except AppExit:
            sys.exit(1)
Exemplo n.º 4
0
def main(argv=None):

    argv = argv or sys.argv[1:]
    op = OptionParser(usage=(
        "Usage: assetgen [<path/to/assetgen.yaml> ...] [options]\n\n"
        "Note:\n"
        "    If you don't specify assetgen.yaml file paths, then `git\n"
        "    ls-files *assetgen.yaml` will be used to detect all config\n"
        "    files in the current repository. So you need to be inside\n"
        "    a git repository's working tree.\n\n"
        "    And if you specify a URL as a `source`, then it will be\n"
        "    downloaded to ~/.assetgen -- you can override this by\n"
        "    setting the env variable $ASSETGEN_DOWNLOADS"
        ))

    op.add_option(
        '-v', '--version', action='store_true',
        help="show program's version number and exit"
        )

    op.add_option(
        '--clean', action='store_true', help="remove all generated files"
        )

    op.add_option(
        '--debug', action='store_true', help="set debug mode"
        )

    op.add_option(
        '--extension', action='append', dest='path',
        help="specify a python extension file (may be repeated)"
        )

    op.add_option(
        '--force', action='store_true', help="force rebuild of all files"
        )

    op.add_option(
        '--nuke', action='store_true',
        help="remove all generated and downloaded files"
        )

    op.add_option(
        '--profile', dest='name', default='default',
        help="specify a profile to use"
        )

    op.add_option(
        '--watch', action='store_true',
        help="keep running assetgen on a loop"
        )

    autocomplete(op)
    options, files = op.parse_args(argv)

    if options.version:
        print 'assetgen %s' % __release__
        sys.exit()

    if options.debug:
        global DEBUG
        DEBUG = True

    clean = options.clean
    extensions = options.path
    force = options.force
    nuke = options.nuke
    profile = options.name
    watch = options.watch

    if extensions:
        scope = globals()
        for ext in extensions:
            execfile(ext, scope, {})

    if files:
        for file in files:
            if not isfile(file):
                exit("Could not find %s" % file)

    if not files:
        if not is_git():
            op.print_help()
            sys.exit()
        root = SCMConfig().root
        files = run_command(
            ['git', 'ls-files', '*assetgen.yaml'], cwd=root
            ).strip().splitlines()
        if not files:
            op.print_help()
            sys.exit()
        files = [join(root, file) for file in files]

    files = [realpath(file) for file in files]

    if watch:
        change_checker = FileChangeDetector()
        for file in files:
            change_checker.mark_clean(file)


    generators = [AssetGenRunner(file, profile, force, nuke) for file in files]

    if nuke:
        if isdir(DOWNLOADS_PATH):
            log.info("Removing: %s" % DOWNLOADS_PATH)
            rmtree(DOWNLOADS_PATH)
        clean = 1

    if clean:
        for assetgen in generators:
            assetgen.clean()
        sys.exit()

    if watch:
        while 1:
            try:
                for assetgen in generators:
                    assetgen.run()
                for idx, file in enumerate(files):
                    if change_checker.is_changed(file):
                        generators[idx] = AssetGenRunner(file, profile, force)
                        change_checker.mark_clean(file)
                sleep(1)
            except AppExit:
                sleep(3)
            except KeyboardInterrupt:
                break
    else:
        try:
            for assetgen in generators:
                assetgen.run()
        except AppExit:
            sys.exit(1)
Exemplo n.º 5
0
def main(argv=None, show_help=False):

    argv = argv or sys.argv[1:]

    # Set the script name to ``redpill`` so that OptionParser error messages
    # don't display a potentially confusing ``redpill.py`` to end users.
    sys.argv[0] = 'redpill'

    major_listing = '\n'.join(
        "    %-10s %s"
        % (cmd, MAJOR_COMMANDS[cmd].__doc__) for cmd in sorted(MAJOR_COMMANDS)
        )

    mini_listing = '\n'.join(
        "    %-10s %s"
        % (cmd, MINI_COMMANDS[cmd].__doc__) for cmd in sorted(MINI_COMMANDS)
        )

    usage = ("""%s\nUsage: redpill <command> [options]
    \nCommands:
    \n%s\n\n%s
    \nSee `redpill help <command>` for more info on a specific command.""" %
    (__doc__, major_listing, mini_listing))

    autocomplete(
        OptionParser(add_help_option=False),
        ListCompleter(AUTOCOMPLETE_COMMANDS.keys()),
        subcommands=AUTOCOMPLETE_COMMANDS
        )

    if not argv:
        show_help = True
    else:
        command = argv[0]
        argv = argv[1:]
        if command in ['help', '-h', '--help']:
            if argv:
                command = argv[0]
                argv = ['--help']
                if command in MINI_COMMANDS:
                    help = MINI_COMMANDS[command].__doc__
                    print "Usage: redpill %s\n\n    %s\n" % (command, help)
                    sys.exit()
            else:
                show_help = True
        elif command in ['-v', '--version']:
            version()
            sys.exit()
        elif command in MINI_COMMANDS:
            MINI_COMMANDS[command]()
            sys.exit()

    if show_help:
        print usage
        sys.exit()

    if command in MAJOR_COMMANDS:
        return MAJOR_COMMANDS[command](argv)

    # We support git-command like behaviour. That is, if there's an external
    # binary named ``redpill-foo`` available on the ``$PATH``, then running ``redpill
    # foo`` will automatically delegate to it.
    try:
        output, retcode = run_command(
            ['redpill-%s' % command] + argv, retcode=True, redirect_stdout=False,
            redirect_stderr=False
            )
    except CommandNotFound:
        exit("ERROR: Unknown command %r" % command)

    if retcode:
        sys.exit(retcode)