111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
""" SCM hooks. Integration with git and mercurial. """
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import sys
|
|
from os import path as op, chmod
|
|
from subprocess import Popen, PIPE
|
|
|
|
from .main import LOGGER, process_paths
|
|
from .config import parse_options, setup_logger
|
|
|
|
|
|
try:
|
|
from configparser import ConfigParser # noqa
|
|
except ImportError: # Python 2
|
|
from ConfigParser import ConfigParser
|
|
|
|
|
|
def run(command):
|
|
""" Run a shell command.
|
|
|
|
:return str: Stdout
|
|
|
|
"""
|
|
p = Popen(command.split(), stdout=PIPE, stderr=PIPE)
|
|
(stdout, stderr) = p.communicate()
|
|
return (p.returncode, [line.strip() for line in stdout.splitlines()],
|
|
[line.strip() for line in stderr.splitlines()])
|
|
|
|
|
|
def git_hook():
|
|
""" Run pylama after git commit. """
|
|
_, files_modified, _ = run("git diff-index --cached --name-only HEAD")
|
|
|
|
options = parse_options()
|
|
setup_logger(options)
|
|
candidates = list(map(str, files_modified))
|
|
if candidates:
|
|
process_paths(options, candidates=candidates)
|
|
|
|
|
|
def hg_hook(ui, repo, node=None, **kwargs):
|
|
""" Run pylama after mercurial commit. """
|
|
seen = set()
|
|
paths = []
|
|
if len(repo):
|
|
for rev in range(repo[node], len(repo)):
|
|
for file_ in repo[rev].files():
|
|
file_ = op.join(repo.root, file_)
|
|
if file_ in seen or not op.exists(file_):
|
|
continue
|
|
seen.add(file_)
|
|
paths.append(file_)
|
|
|
|
options = parse_options()
|
|
setup_logger(options)
|
|
if paths:
|
|
process_paths(options, candidates=paths)
|
|
|
|
|
|
def install_git(path):
|
|
""" Install hook in Git repository. """
|
|
hook = op.join(path, 'pre-commit')
|
|
with open(hook, 'w') as fd:
|
|
fd.write("""#!/usr/bin/env python
|
|
import sys
|
|
from pylama.hook import git_hook
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(git_hook())
|
|
""")
|
|
chmod(hook, 484)
|
|
|
|
|
|
def install_hg(path):
|
|
""" Install hook in Mercurial repository. """
|
|
hook = op.join(path, 'hgrc')
|
|
if not op.isfile(hook):
|
|
open(hook, 'w+').close()
|
|
|
|
c = ConfigParser()
|
|
c.readfp(open(hook, 'r'))
|
|
if not c.has_section('hooks'):
|
|
c.add_section('hooks')
|
|
|
|
if not c.has_option('hooks', 'commit'):
|
|
c.set('hooks', 'commit', 'python:pylama.hooks.hg_hook')
|
|
|
|
if not c.has_option('hooks', 'qrefresh'):
|
|
c.set('hooks', 'qrefresh', 'python:pylama.hooks.hg_hook')
|
|
|
|
c.write(open(hook, 'w+'))
|
|
|
|
|
|
def install_hook(path):
|
|
""" Auto definition of SCM and hook installation. """
|
|
git = op.join(path, '.git', 'hooks')
|
|
hg = op.join(path, '.hg')
|
|
if op.exists(git):
|
|
install_git(git)
|
|
LOGGER.warn('Git hook has been installed.')
|
|
|
|
elif op.exists(hg):
|
|
install_hg(hg)
|
|
LOGGER.warn('Mercurial hook has been installed.')
|
|
|
|
else:
|
|
LOGGER.error('VCS has not found. Check your path.')
|
|
sys.exit(1)
|
|
|
|
# pylama:ignore=F0401,E1103,D210,F0001
|