commit 2ba05f1672dc958412ab29fa994c722bfca1cff0
author Peter Wen <wnwen@chromium.org> Mon Jun 15 16:22:10 2020 +0000
committer LUCI CQ <infra-scoped@luci-project-accounts.iam.gserviceaccount.com> Mon Jun 15 16:22:10 2020 +0000
tree 64e4813d7ba8420e7ce2491d8ca2a33e89b93eb2
parent 23bd174fea451b2ef774eebdfd02fa298131a148

Support formatting python3 scripts

Use vpython or vpython3 to invoke yapf depending on the shebang line for
each file.

Bug: 1062053
Change-Id: I497fa48c08aa6052a2b06df7cc7f18e2154ab898
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/2241792
Commit-Queue: Edward Lesmes <ehmaldonado@chromium.org>
Reviewed-by: Edward Lesmes <ehmaldonado@chromium.org>
Auto-Submit: Peter Wen <wnwen@chromium.org>

git_cl.py

diff --git a/git_cl.py b/git_cl.py
index fe92b2c..49a9f12 100755
--- a/git_cl.py
+++ b/git_cl.py
@@ -4931,7 +4931,13 @@
       if not yapf_style:
         yapf_style = 'pep8'
 
-      cmd = [yapf_tool, '--style', yapf_style, f]
+      with open(f, 'r') as py_f:
+        if 'python3' in py_f.readline():
+          vpython_script = 'vpython3'
+        else:
+          vpython_script = 'vpython'
+
+      cmd = [vpython_script, yapf_tool, '--style', yapf_style, f]
 
       has_formattable_lines = False
       if not opts.full: