forked from quantumlib/Cirq
-
Notifications
You must be signed in to change notification settings - Fork 0
/
format-incremental
executable file
·111 lines (103 loc) · 3.9 KB
/
format-incremental
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/usr/bin/env bash
################################################################################
# Formats lines of python code that have been modified.
#
# Usage:
# check/format-incremental [BASE_REVISION] [--apply]
#
# Without '--apply', the diff that would be applied is printed and the exit
# status is 1 if there are any changes or else 0 if no changes are needed.
#
# With '--apply', the exit status is 0 and the changed files are actually
# reformated.
#
# Note that sometimes yapf will format unchanged lines. In particular, it
# completely ignores the changed line ranges when fixing indentation.
#
# You can specify a base git revision to compare against (i.e. to use when
# determining whether or not a line is considered to have "changed"). For
# example, you can compare against 'origin/master' or 'HEAD~1'.
#
# If you don't specify a base revision, the following defaults will be tried, in
# order, until one exists:
#
# 1. upstream/master
# 2. origin/master
# 3. master
#
# If none exists, the script fails.
################################################################################
# Get the working directory to the repo root.
cd "$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$(git rev-parse --show-toplevel)"
# Parse arguments.
only_print=1
rev=""
for arg in $@; do
if [[ "${arg}" == "--apply" ]]; then
only_print=0
elif [ -z "${rev}" ]; then
if [ "$(git cat-file -t ${arg} 2> /dev/null)" != "commit" ]; then
echo -e "\e[31mNo revision '${arg}'.\e[0m" >&2
exit 1
fi
rev="${arg}"
else
echo -e "\e[31mToo many arguments. Expected [revision] [--apply].\e[0m" >&2
exit 1
fi
done
# Figure out which branch to compare against.
if [ -z "${rev}" ]; then
if [ "$(git cat-file -t upstream/master 2> /dev/null)" == "commit" ]; then
rev=upstream/master
elif [ "$(git cat-file -t origin/master 2> /dev/null)" == "commit" ]; then
rev=origin/master
elif [ "$(git cat-file -t master 2> /dev/null)" == "commit" ]; then
rev=master
else
echo -e "\e[31mNo default revision found to compare against. Argument #1 must be what to diff against (e.g. 'origin/master' or 'HEAD~1').\e[0m" >&2
exit 1
fi
fi
base="$(git merge-base ${rev} HEAD)"
if [ "$(git rev-parse ${rev})" == "${base}" ]; then
echo -e "Comparing against revision '${rev}'." >&2
else
echo -e "Comparing against revision '${rev}' (merge base ${base})." >&2
rev="${base}"
fi
# Get the _test version of changed python files.
needed_changes=0
changed_files=$(git diff --name-only ${rev} -- | grep "\.py$" | grep -v "_pb2\.py$")
esc=$(printf '\033')
for changed_file in ${changed_files}; do
# Extract changed line ranges from diff output.
changed_line_ranges=$( \
git diff --unified=0 "${rev}" -- "${changed_file}" \
| perl -ne 'chomp(); if (/@@ -\d+(,\d+)? \+(\d+)(,)?(\d+)? @@/) {$end=$2+($4 or 1)-1; print "--lines=$2-$end "}' \
)
if [[ "${changed_line_ranges}" != "--lines=0-0 " ]]; then
# Do the formatting.
results=$(yapf --style=google --diff "${changed_file}" ${changed_line_ranges})
# Print colorized error messages.
if [ ! -z "${results}" ]; then
needed_changes=1
if (( only_print == 0 )); then
$(yapf --style=google --in-place "${changed_file}" ${changed_line_ranges})
else
echo -e "\n\e[31mChanges in ${changed_file} require formatting:\e[0m\n${results}" \
| sed "s/^\(+ .*\)$/${esc}[32m\\1${esc}[0m/" \
| sed "s/^\(- .*\)$/${esc}[31m\\1${esc}[0m/"
fi
fi
fi
done
if (( needed_changes == 0 )); then
echo -e "\e[32mNo formatting needed on changed lines\e[0m."
elif (( only_print == 1 )); then
echo -e "\e[31mSome formatting needed on changed lines\e[0m."
exit 1
else
echo -e "\e[32mReformatted changed lines\e[0m."
fi