1 files changed, 58 insertions, 0 deletions
diff --git a/grpc/tools/distrib/isort_code.sh b/grpc/tools/distrib/isort_code.sh
new file mode 100755
index 00000000..08b4e5a1
--- /dev/null
+++ b/grpc/tools/distrib/isort_code.sh
@@ -0,0 +1,58 @@
+#!/bin/bash
+# Copyright 2021 The gRPC Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -ex
+
+ACTION=${1:---overwrite-in-place}
+[[ $ACTION == '--overwrite-in-place' ]] || [[ $ACTION == '--diff' ]]
+
+if [[ $ACTION == '--diff' ]]; then
+    ACTION="--diff --check"
+fi
+
+# Change to root
+cd "$(dirname "${0}")/../.."
+
+DIRS=(
+    'examples/python'
+    'src/python'
+    'test'
+    'tools'
+    'setup.py'
+)
+
+VIRTUALENV=isort_virtual_environment
+
+python3 -m virtualenv $VIRTUALENV
+PYTHON=${VIRTUALENV}/bin/python
+"$PYTHON" -m pip install isort==5.9.2
+
+$PYTHON -m isort $ACTION \
+  --force-sort-within-sections \
+  --force-single-line-imports --single-line-exclusions=typing \
+  --src "examples/python/data_transmission" \
+  --src "examples/python/async_streaming" \
+  --src "tools/run_tests/xds_k8s_test_driver" \
+  --src "src/python/grpcio_tests" \
+  --src "tools/run_tests" \
+  --project "examples" \
+  --project "src" \
+  --thirdparty "grpc" \
+  --skip-glob "third_party/*" \
+  --skip-glob "*/env/*" \
+  --skip-glob "*pb2*.py" \
+  --skip-glob "**/site-packages/**/*" \
+  --dont-follow-links \
+  "${DIRS[@]}"