Add PoC for on-demand hooks repo download

This makes pre-commit skip all config entries not listing certain hook

Resolves #637
This commit is contained in:
Sviatoslav Sydorenko 2017-10-21 02:12:29 +03:00
parent 39d5205e31
commit a46b3a1231
No known key found for this signature in database
GPG key ID: 7072133B5779DD62
2 changed files with 25 additions and 6 deletions

View file

@ -57,6 +57,8 @@ def _add_hook_type_option(parser):
def _add_run_options(parser): def _add_run_options(parser):
parser.add_argument('hook', nargs='?', help='A single hook-id to run') parser.add_argument('hook', nargs='?', help='A single hook-id to run')
parser.add_argument('--verbose', '-v', action='store_true', default=False) parser.add_argument('--verbose', '-v', action='store_true', default=False)
parser.add_argument('--only', '-i', action='store_true', default=False,
help='Save some bandwidth')
parser.add_argument( parser.add_argument(
'--origin', '-o', '--origin', '-o',
help="The origin branch's commit_id when using `git push`.", help="The origin branch's commit_id when using `git push`.",
@ -227,7 +229,11 @@ def main(argv=None):
with error_handler(): with error_handler():
add_logging_handler(args.color) add_logging_handler(args.color)
runner = Runner.create(args.config) runner = Runner.create(
args.config,
filter_repos=args.only,
filter_hook=args.hook,
)
git.check_for_cygwin_mismatch() git.check_for_cygwin_mismatch()
if args.command == 'install': if args.command == 'install':

View file

@ -15,20 +15,26 @@ class Runner(object):
repository under test. repository under test.
""" """
def __init__(self, git_root, config_file, store_dir=None): def __init__(self, git_root, config_file, store_dir=None,
filter_repos=False, filter_hook=None):
self.git_root = git_root self.git_root = git_root
self.config_file = config_file self.config_file = config_file
self._store_dir = store_dir self._store_dir = store_dir
self._filter_repos = filter_repos
self._filter_hook = filter_hook
@classmethod @classmethod
def create(cls, config_file): def create(cls, config_file, filter_repos=False, filter_hook=None):
"""Creates a Runner by doing the following: """Creates a Runner by doing the following:
- Finds the root of the current git repository - Finds the root of the current git repository
- chdir to that directory - chdir to that directory
""" """
root = git.get_root() root = git.get_root()
os.chdir(root) os.chdir(root)
return cls(root, config_file) return cls(
root, config_file,
filter_repos=filter_repos, filter_hook=filter_hook
)
@cached_property @cached_property
def git_dir(self): def git_dir(self):
@ -45,8 +51,15 @@ class Runner(object):
@cached_property @cached_property
def repositories(self): def repositories(self):
"""Returns a tuple of the configured repositories.""" """Returns a tuple of the configured repositories."""
repos = self.config['repos'] filtered_repos = repos = self.config['repos']
return tuple(Repository.create(x, self.store) for x in repos) if self._filter_repos and self._filter_hook:
filtered_repos = (
r for r in repos
if self._filter_hook in (
h['id'] for h in r['hooks']
)
)
return tuple(Repository.create(x, self.store) for x in filtered_repos)
def get_hook_path(self, hook_type): def get_hook_path(self, hook_type):
return os.path.join(self.git_dir, 'hooks', hook_type) return os.path.join(self.git_dir, 'hooks', hook_type)