Welcome, guest | Sign In | My Account | Store | Cart
albertjan@debian ~/Desktop/test_repo $ git config --global init.templatedir ~/Desktop/git_template_dir
albertjan@debian ~/Desktop/test_repo $ cd ~/Desktop/git_template_dir
albertjan@debian ~/Desktop/git_template_dir $ cat hooks/pre-commit
#!/usr/bin/python
#-*- mode: python -*-

"""Git pre-commit hook: reject large files"""

__author__   = "Albert-Jan Roskam"
__email__    = "@".join(["fomcl", "yahoo" + ".com"])
__version__  = "1.0.3"

import sys
import os
import re
from subprocess import Popen, PIPE

def git_filesize_hook(megabytes_cutoff=5, verbose=False):
    """Git pre-commit hook: Return error if the maximum file size in the HEAD
    revision exceeds <megabytes_cutoff>, succes (0) otherwise. You can bypass
    this hook by specifying '--no-verify' as an option in 'git commit'."""
    if verbose: print os.getcwd()
    cmd = "git ls-tree --full-tree -r -l HEAD"
    kwargs = dict(args=cmd, shell=True, stdout=PIPE, cwd=os.getcwd())
    if sys.platform.startswith("win"):
        del kwargs["cwd"]
        cmd = "pushd && \"%s\"" % os.getcwd() + cmd + " && popd"
    git = Popen(**kwargs)
    get_size = lambda item: int(re.split(" +", item)[3].split("\t")[0])          
    sizes = map(get_size, git.stdout.readlines())
    bytes_cut_off = megabytes_cutoff * 2 ** 20
    if sizes and max(sizes) > bytes_cut_off:
        return ("ERROR: your commit contains at least one file "
                "that is larger than %d bytes" % bytes_cut_off)
    return 0

if __name__ == "__main__":
    sys.exit(git_filesize_hook(0.000001, True))

albertjan@debian ~/Desktop/git_template_dir $ cd -
/home/albertjan/Desktop/test_repo
albertjan@debian ~/Desktop/test_repo $ git init  ## this also fetches my own pre-commit hook from template_dir
Initialized empty Git repository in /home/albertjan/Desktop/test_repo/.git/
albertjan@debian ~/Desktop/test_repo $ touch foo.txt
albertjan@debian ~/Desktop/test_repo $ git add foo.txt
albertjan@debian ~/Desktop/test_repo $ ls -l .git/hooks
total 4
-rw-r--r-- 1 albertjan albertjan 1468 May 22 14:49 pre-commit
albertjan@debian ~/Desktop/test_repo $ git commit -a -m "commit"   ##### hook does not yet work
[master (root-commit) dc82f3d] commit
 0 files changed
 create mode 100644 foo.txt
albertjan@debian
~/Desktop/test_repo $ chmod +x .git/hooks/pre-commit
albertjan@debian ~/Desktop/test_repo $ echo "blaah\n" >> foo.txt
albertjan@debian ~/Desktop/test_repo $ git commit -a -m "commit"  ##### now the hook does its job
/home/albertjan/Desktop/test_repo
ERROR: your commit contains at least one file that is larger than 1 bytes

Diff to Previous Revision

--- revision 4 2014-05-27 13:45:26
+++ revision 5 2015-01-05 10:11:21
@@ -8,7 +8,7 @@
 
 __author__   = "Albert-Jan Roskam"
 __email__    = "@".join(["fomcl", "yahoo" + ".com"])
-__version__  = "1.0.2"
+__version__  = "1.0.3"
 
 import sys
 import os
@@ -21,7 +21,11 @@
     this hook by specifying '--no-verify' as an option in 'git commit'."""
     if verbose: print os.getcwd()
     cmd = "git ls-tree --full-tree -r -l HEAD"
-    git = Popen(cmd, shell=True, stdout=PIPE, cwd=os.getcwd())
+    kwargs = dict(args=cmd, shell=True, stdout=PIPE, cwd=os.getcwd())
+    if sys.platform.startswith("win"):
+        del kwargs["cwd"]
+        cmd = "pushd && \"%s\"" % os.getcwd() + cmd + " && popd"
+    git = Popen(**kwargs)
     get_size = lambda item: int(re.split(" +", item)[3].split("\t")[0])          
     sizes = map(get_size, git.stdout.readlines())
     bytes_cut_off = megabytes_cutoff * 2 ** 20

History