format.sh: allow selecting file paths
This commit is contained in:
parent
de71a71c13
commit
a6129a82bd
1 changed files with 12 additions and 4 deletions
16
format.sh
16
format.sh
|
@ -4,16 +4,24 @@ yapf_args=('--recursive' '--parallel')
|
||||||
autoflake_args=('--recursive' '--remove-unused-variables' '--remove-all-unused-imports' '--expand-star-imports' '--remove-duplicate-keys')
|
autoflake_args=('--recursive' '--remove-unused-variables' '--remove-all-unused-imports' '--expand-star-imports' '--remove-duplicate-keys')
|
||||||
|
|
||||||
format() {
|
format() {
|
||||||
yapf "${yapf_args[@]}" .
|
files=("$@")
|
||||||
autoflake "${autoflake_args[@]}" .
|
if [[ -z "${files[@]}" ]]; then
|
||||||
|
files=(".")
|
||||||
|
fi
|
||||||
|
|
||||||
|
yapf "${yapf_args[@]}" "${files[@]}"
|
||||||
|
autoflake "${autoflake_args[@]}" "${files[@]}"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if [[ "$1" == "--check" ]]; then
|
if [[ "$1" == "--check" ]]; then
|
||||||
yapf_args+=('--diff')
|
yapf_args+=('--diff')
|
||||||
[[ "$(format | tee /dev/stderr | wc -c)" == "0" ]]
|
shift
|
||||||
|
[[ "$(format "$@" | tee /dev/stderr | wc -c)" == "0" ]]
|
||||||
else
|
else
|
||||||
yapf_args+=('--in-place')
|
yapf_args+=('--in-place')
|
||||||
autoflake_args+=('--in-place')
|
autoflake_args+=('--in-place')
|
||||||
format
|
format "$@"
|
||||||
fi
|
fi
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue