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.2"

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"
    git = Popen(cmd, shell=True, stdout=PIPE, cwd=os.getcwd())
    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 3 2014-05-23 21:11:10
+++ revision 4 2014-05-27 13:45:26
@@ -5,6 +5,10 @@
 #-*- mode: python -*-
 
 """Git pre-commit hook: reject large files"""
+
+__author__   = "Albert-Jan Roskam"
+__email__    = "@".join(["fomcl", "yahoo" + ".com"])
+__version__  = "1.0.2"
 
 import sys
 import os
@@ -21,7 +25,7 @@
     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 max(sizes) > bytes_cut_off:
+    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

History