check-docstyle.sh

directories="src tests"
separate_files="setup.py"
pass=0
fail=0

function prepare_venv() {
    VIRTUALENV=`which virtualenv`
    if [ $? -eq 1 ]; then
        # python34 which is in CentOS does not have virtualenv binary
        VIRTUALENV=`which virtualenv-3`
    fi

    ${VIRTUALENV} -p python3 venv && source venv/bin/activate && python3 `which pip3` install pydocstyle
}

# run the pydocstyle for all files that are provided in $1
function check_files() {
    for source in $1
    do
        echo $source
        pydocstyle --count $source
        if [ $? -eq 0 ]
        then
            echo "    Pass"
            let "pass++"
        elif [ $? -eq 2 ]
        then
            echo "    Illegal usage (should not happen)"
            exit 2
        else
            echo "    Fail"
            let "fail++"
        fi
    done
}


echo "----------------------------------------------------"
echo "Checking documentation strings in all sources stored"
echo "in following directories:"
echo $directories
echo "----------------------------------------------------"
echo

[ "$NOVENV" == "1" ] || prepare_venv || exit 1

# checks for the whole directories
for directory in $directories
do
    files=`find $directory -path $directory/venv -prune -o -name '*.py' -print`

    check_files "$files"
done


echo
echo "----------------------------------------------------"
echo "Checking documentation strings in the following files"
echo $separate_files
echo "----------------------------------------------------"

check_files "$separate_files"


if [ $fail -eq 0 ]
then
    echo "All checks passed for $pass source files"
else
    let total=$pass+$fail
    echo "Documentation strings should be added and/or fixed in $fail source files out of $total files"
    exit 1
fi