| OLD | NEW | 
|---|
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be | 
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. | 
| 4 | 4 | 
| 5 import os | 5 import os | 
| 6 import sys | 6 import sys | 
| 7 import logging | 7 import logging | 
| 8 import shutil | 8 import shutil | 
| 9 | 9 | 
|  | 10 sys.path.insert(1, os.path.join(sys.path[0], '..', '..', 'python')) | 
|  | 11 import google.path_utils | 
|  | 12 | 
| 10 import diff_util | 13 import diff_util | 
| 11 | 14 | 
| 12 sys.path.insert(1, os.path.join(sys.path[0], '..', '..', 'python')) |  | 
| 13 from google import path_utils |  | 
| 14 |  | 
| 15 def DoPresubmitMain(argv, original_filename, backup_filename, script_name, | 15 def DoPresubmitMain(argv, original_filename, backup_filename, script_name, | 
| 16                     prettyFn): | 16                     prettyFn): | 
| 17   """Execute presubmit/pretty printing for the target file. | 17   """Execute presubmit/pretty printing for the target file. | 
| 18 | 18 | 
| 19   Args: | 19   Args: | 
| 20     argv: command line arguments | 20     argv: command line arguments | 
| 21     original_filename: The filename to read from. | 21     original_filename: The filename to read from. | 
| 22     backup_filename: When pretty printing, move the old file contents here. | 22     backup_filename: When pretty printing, move the old file contents here. | 
| 23     script_name: The name of the script to run for pretty printing. | 23     script_name: The name of the script to run for pretty printing. | 
| 24     prettyFn: A function which takes the original xml content and produces | 24     prettyFn: A function which takes the original xml content and produces | 
| 25         pretty printed xml. | 25         pretty printed xml. | 
| 26 | 26 | 
| 27   Returns: | 27   Returns: | 
| 28     An exit status.  Non-zero indicates errors. | 28     An exit status.  Non-zero indicates errors. | 
| 29   """ | 29   """ | 
| 30   logging.basicConfig(level=logging.INFO) | 30   logging.basicConfig(level=logging.INFO) | 
| 31   presubmit = ('--presubmit' in argv) | 31   presubmit = ('--presubmit' in argv) | 
| 32 | 32 | 
| 33   # If there is a description xml in the current working directory, use that. | 33   # If there is a description xml in the current working directory, use that. | 
| 34   # Otherwise, use the one residing in the same directory as this script. | 34   # Otherwise, use the one residing in the same directory as this script. | 
| 35   xml_dir = os.getcwd() | 35   xml_dir = os.getcwd() | 
| 36   if not os.path.isfile(os.path.join(xml_dir, original_filename)): | 36   if not os.path.isfile(os.path.join(xml_dir, original_filename)): | 
| 37     xml_dir = path_utils.ScriptDir() | 37     xml_dir = google.path_utils.ScriptDir() | 
| 38 | 38 | 
| 39   xml_path = os.path.join(xml_dir, original_filename) | 39   xml_path = os.path.join(xml_dir, original_filename) | 
| 40 | 40 | 
| 41   # Save the original file content. | 41   # Save the original file content. | 
| 42   logging.info('Loading %s...', os.path.relpath(xml_path)) | 42   logging.info('Loading %s...', os.path.relpath(xml_path)) | 
| 43   with open(xml_path, 'rb') as f: | 43   with open(xml_path, 'rb') as f: | 
| 44     original_xml = f.read() | 44     original_xml = f.read() | 
| 45 | 45 | 
| 46   # Check there are no CR ('\r') characters in the file. | 46   # Check there are no CR ('\r') characters in the file. | 
| 47   if '\r' in original_xml: | 47   if '\r' in original_xml: | 
| (...skipping 22 matching lines...) Expand all  Loading... | 
| 70     sys.exit(1) | 70     sys.exit(1) | 
| 71 | 71 | 
| 72   logging.info('Creating backup file: %s', backup_filename) | 72   logging.info('Creating backup file: %s', backup_filename) | 
| 73   shutil.move(xml_path, os.path.join(xml_dir, backup_filename)) | 73   shutil.move(xml_path, os.path.join(xml_dir, backup_filename)) | 
| 74 | 74 | 
| 75   with open(xml_path, 'wb') as f: | 75   with open(xml_path, 'wb') as f: | 
| 76     f.write(pretty) | 76     f.write(pretty) | 
| 77   logging.info('Updated %s. Don\'t forget to add it to your changelist', | 77   logging.info('Updated %s. Don\'t forget to add it to your changelist', | 
| 78                xml_path) | 78                xml_path) | 
| 79   sys.exit(0) | 79   sys.exit(0) | 
| OLD | NEW | 
|---|