pre-commit.githook

#!/bin/bash
# vim: et ts=4 sw=4 ft=sh

# Interesting post on max line length:
# http://stackoverflow.com/questions/88942/why-should-python-pep-8-specify-a-maximum-line-length-of-79-characters

PEP8_CMD='flake8'
PEP8_OPTIONS='--max-line-length=120'


RED=$(echo -e $"\033[1;31m")
YELLOW=$(echo -e $"\033[0;33m")
CYAN=$(echo -e $"\033[0;36m")
RESET=$(echo -e $"\033[0;0m")
BRIGHTYELLOW=$(echo -e $"\033[1;33m")
WHITE=$(echo -e $"\033[1;37m")

RE="s/\([^:]*\):\([0-9]*\):\([0-9]*\): \([EW][0-9]*\) \(.*\)/$WHITE[$CYAN\1$RESET $BRIGHTYELLOW\2:\3$WHITE] $RED\4 $YELLOW\5$RESET/g"
STATUS=0

_get_files() {
    local i
    unset FILES
    while IFS= read -r -d $'\0' file; do
        FILES[i++]="$file"
    done < <(git diff --name-only --diff-filter=ACMR --staged -z "$1")
}

# Flake8
if ! RESULT=$(flake8 $(git status -s | grep -E '\.py$' | cut -c 4-)); then
    echo "${RED}There are PEP8 issues in your code:${RESET}"
    STATUS=1
fi
if [[ -n "$RESULT" ]] ; then
    echo "$RESULT" | sed -e "$RE"
    echo
fi


# isort
if ! RESULT=$(python -c 'import sys; from isort.hooks import git_hook; sys.exit(git_hook(strict=True))'); then
    STATUS=1
    echo "${RED}There are isort issues in your code:${RESET}"
    echo "$RESULT"
    echo
    echo "${CYAN}Run this command to sort the imports:${RESET}"
    echo "git diff --staged --name-only --diff-filter '*.py' | xargs isort"
fi


if [[ $STATUS != 0 ]] ; then
    # claim stdin back
    exec < /dev/tty
    echo
    read -r -p "${RED}Do you wish to commit it anyway ${CYAN}[${WHITE}y${CYAN}/${WHITE}N${CYAN}]${RESET}? " yn
    case $yn in
        [Yy]* ) exit 0;;
        [Nn]* ) exit $STATUS;;
        * ) exit $STATUS;;
    esac
    # close stdin
    exec <&-
fi

exit $STATUS