mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-13 12:30:08 -06:00
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import argparse
|
|
import sys
|
|
|
|
from pre_commit import color
|
|
from pre_commit import commands
|
|
from pre_commit.runner import Runner
|
|
from pre_commit.util import entry
|
|
|
|
|
|
@entry
|
|
def run(argv):
|
|
parser = argparse.ArgumentParser()
|
|
|
|
subparsers = parser.add_subparsers(dest='command')
|
|
|
|
subparsers.add_parser('install', help='Intall the pre-commit script.')
|
|
|
|
subparsers.add_parser('uninstall', help='Uninstall the pre-commit script.')
|
|
|
|
subparsers.add_parser('clean', help='Clean out pre-commit files.')
|
|
|
|
subparsers.add_parser('autoupdate', help='Auto-update hooks config.')
|
|
|
|
run_parser = subparsers.add_parser('run', help='Run hooks.')
|
|
run_parser.add_argument('hook', nargs='?', help='A single hook-id to run')
|
|
run_parser.add_argument(
|
|
'--all-files', '-a', action='store_true', default=False,
|
|
help='Run on all the files in the repo. Implies --no-stash.',
|
|
)
|
|
run_parser.add_argument(
|
|
'--color', default='auto', type=color.use_color,
|
|
help='Whether to use color in output. Defaults to `auto`',
|
|
)
|
|
run_parser.add_argument(
|
|
'--no-stash', default=False, action='store_true',
|
|
help='Use this option to prevent auto stashing of unstaged files.',
|
|
)
|
|
run_parser.add_argument('--verbose', '-v', action='store_true', default=False)
|
|
|
|
help = subparsers.add_parser('help', help='Show help for a specific command.')
|
|
help.add_argument('help_cmd', nargs='?', help='Command to show help for.')
|
|
|
|
# Argparse doesn't really provide a way to use a `default` subparser
|
|
if len(argv) == 0:
|
|
argv = ['run']
|
|
args = parser.parse_args(argv)
|
|
|
|
runner = Runner.create()
|
|
|
|
if args.command == 'install':
|
|
return commands.install(runner)
|
|
elif args.command == 'uninstall':
|
|
return commands.uninstall(runner)
|
|
elif args.command == 'clean':
|
|
return commands.clean(runner)
|
|
elif args.command == 'autoupdate':
|
|
return commands.autoupdate(runner)
|
|
elif args.command == 'run':
|
|
return commands.run(runner, args)
|
|
elif args.command == 'help':
|
|
if args.help_cmd:
|
|
parser.parse_args([args.help_cmd, '--help'])
|
|
else:
|
|
parser.parse_args(['--help'])
|
|
else:
|
|
raise NotImplementedError(
|
|
'Command {0} not implemented.'.format(args.command)
|
|
)
|
|
|
|
raise AssertionError(
|
|
'Command {0} failed to exit with a returncode'.format(args.command)
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(run())
|