Skip to content

Instantly share code, notes, and snippets.

@wasade
Created March 18, 2014 01:17
Show Gist options
  • Save wasade/9611804 to your computer and use it in GitHub Desktop.
Save wasade/9611804 to your computer and use it in GitHub Desktop.
PEP8 git pre-commit hook, adapted from https://gist.github.com/lentil/810399. The specific change was to stash any uncommited changed prior to executing pep8.
#!/usr/bin/env python
from __future__ import with_statement
import os
import re
import shutil
import subprocess
import sys
import tempfile
def system(*args, **kwargs):
kwargs.setdefault('stdout', subprocess.PIPE)
proc = subprocess.Popen(args, **kwargs)
out, err = proc.communicate()
return out
def main():
modified = re.compile('^[AM]+\s+(?P<name>.*\.py)', re.MULTILINE)
system('git', 'stash', '-q', '--keep-index')
files = system('git', 'status', '--porcelain')
files = modified.findall(files)
tempdir = tempfile.mkdtemp()
for name in files:
filename = os.path.join(tempdir, name)
filepath = os.path.dirname(filename)
if not os.path.exists(filepath):
os.makedirs(filepath)
with file(filename, 'w') as f:
system('git', 'show', ':' + name, stdout=f)
output = system('pep8', '.', cwd=tempdir)
shutil.rmtree(tempdir)
system('git', 'stash', 'pop', '-q')
if output:
print output,
sys.exit(1)
if __name__ == '__main__':
main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment