Lets you run sort-headers.py on all files changed since some git branch.

tools/sort-headers.py

 #!/usr/bin/python
 # Copyright (c) 2011 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Given a filename as an argument, sort the #include/#imports in that file.
 
 Shows a diff and prompts for confirmation before doing the deed.

[(unused - use chromium), 2011/10/31 18:19:56]

Maybe add "Works great with tools/git/for-all-touched-files.py" here?

[Jói, 2011/10/31 18:32:08]

Done.

 """
 
 import optparse
 import os
 import sys
 import termios
 import tty
 
 def YesNo(prompt):
   """Prompts with a yes/no question, returns True if yes."""
@@ skipping 45 matching lines @@
       while IsInclude(line):
         headerblock.append(line)
         line = infile.next()
       for header in sorted(headerblock, key=IncludeCompareKey):
         outfile.write(header)
       # Intentionally fall through, to write the line that caused
       # the above while loop to exit.
     outfile.write(line)
 
 
+def DiffAndConfirm(filename, should_confirm):

[(unused - use chromium), 2011/10/31 18:19:56]

needs docstring

[Jói, 2011/10/31 18:32:08]

Done.

+  fixfilename = filename + '.new'
+  infile = open(filename, 'r')
+  outfile = open(fixfilename, 'w')
+  SortHeader(infile, outfile)
+  infile.close()
+  outfile.close()  # Important so the below diff gets the updated contents.
+
+  try:
+    diff = os.system('diff -u %s %s' % (filename, fixfilename))
+    if diff >> 8 == 0:  # Check exit code.
+      print '%s: no change' % filename
+      return
+
+    if not should_confirm or YesNo('Use new file (y/N)?'):
+      os.rename(fixfilename, filename)
+  finally:
+    try:
+      os.remove(fixfilename)
+    except OSError:
+      # If the file isn't there, we don't care.
+      pass
+
+
 def main():
   parser = optparse.OptionParser(usage='%prog filename1 filename2 ...')
-  opts, args = parser.parse_args()
+  parser.add_option('-f', '--force', action='store_false', default=True,
+                    dest='should_confirm',
+                    help='Turn off confirmation prompt.')
+  opts, filenames = parser.parse_args()
 
-  if len(args) < 1:
+  if len(filenames) < 1:
     parser.print_help()
     sys.exit(1)
 
-  for filename in args:
-    fixfilename = filename + '.new'
-    infile = open(filename, 'r')
-    outfile = open(fixfilename, 'w')
-    SortHeader(infile, outfile)
-    infile.close()
-    outfile.close()  # Important so the below diff gets the updated contents.
-
-    try:
-      diff = os.system('diff -u %s %s' % (filename, fixfilename))
-      if diff >> 8 == 0:  # Check exit code.
-        print '%s: no change' % filename
-        continue
-
-      if YesNo('Use new file (y/N)?'):
-        os.rename(fixfilename, filename)
-    finally:
-      try:
-        os.remove(fixfilename)
-      except OSError:
-        # If the file isn't there, we don't care.
-        pass
+  for filename in filenames:
+    DiffAndConfirm(filename, opts.should_confirm)
 
 
 if __name__ == '__main__':
   main()