Simplify the install and uninstall commands and improve tests.

This commit is contained in:
Anthony Sottile
2014-03-23 18:33:18 -07:00
parent 48bbc68b35
commit ecdacd474b
7 changed files with 82 additions and 52 deletions

View File

@@ -1,9 +1,7 @@
import functools
import os
import os.path
import pkg_resources
import re
import stat
from plumbum import local
from pre_commit.util import memoize_by_cwd
@@ -24,23 +22,6 @@ def get_root():
return _get_root_new()
@memoize_by_cwd
def get_pre_commit_path():
return os.path.join(get_root(), '.git/hooks/pre-commit')
def create_pre_commit():
path = get_pre_commit_path()
pre_commit_file = pkg_resources.resource_filename('pre_commit', 'resources/pre-commit.sh')
local.path(path).write(local.path(pre_commit_file).read())
original_mode = os.stat(path).st_mode
os.chmod(path, original_mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
def remove_pre_commit():
local.path(get_pre_commit_path()).delete()
def get_head_sha(git_repo_path):
with local.cwd(git_repo_path):
return local['git']['rev-parse', 'HEAD']().strip()