OLD | NEW |
(Empty) | |
| 1 # Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. |
| 4 |
| 5 """ Changes builder names in the expected factory configurations. """ |
| 6 |
| 7 |
| 8 import os |
| 9 import json |
| 10 import re |
| 11 import subprocess |
| 12 import sys |
| 13 |
| 14 |
| 15 EXPECTED_DIR = os.path.join(os.path.dirname(__file__), 'expected') |
| 16 if os.name == 'nt': |
| 17 SVN = 'svn.bat' |
| 18 else: |
| 19 SVN = 'svn' |
| 20 |
| 21 |
| 22 def CheckName(builder_name): |
| 23 """ Make sure that the builder name contains no illegal characters. """ |
| 24 if not re.match('^[\w\-_\.]+$', builder_name): |
| 25 raise ValueError('"%s" is not a valid builder name.' % builder_name) |
| 26 |
| 27 |
| 28 def ExistingBuilders(): |
| 29 """ List the existing set of builder names. """ |
| 30 builders = os.listdir(EXPECTED_DIR) |
| 31 builders.sort() |
| 32 for builder_file in builders: |
| 33 if not builder_file.startswith('.'): |
| 34 yield builder_file |
| 35 |
| 36 |
| 37 def CreateMappingTemplate(mapping_file): |
| 38 """ Creates a JSON file containing a template for mapping old builder names to |
| 39 new builder names. """ |
| 40 mapping = {} |
| 41 for builder_file in ExistingBuilders(): |
| 42 mapping[builder_file] = '' |
| 43 with open(mapping_file, 'w') as f: |
| 44 json.dump(mapping, f, indent=4, sort_keys=True) |
| 45 print 'Created mapping template file in %s' % mapping_file |
| 46 |
| 47 |
| 48 def Rename(old_name, new_name): |
| 49 """ "Rename" a builder. Runs "svn mv" to create the new expectations file, |
| 50 then replaces all instances of the old builder name with the new in that file. |
| 51 """ |
| 52 print 'Changing "%s" to "%s".' % (old_name, new_name) |
| 53 old_file_path = os.path.join(EXPECTED_DIR, old_name) |
| 54 new_file_path = os.path.join(EXPECTED_DIR, new_name) |
| 55 |
| 56 # Verify the new file names. |
| 57 CheckName(old_name) |
| 58 CheckName(new_name) |
| 59 if not os.path.isfile(old_file_path): |
| 60 raise Exception('Config file for "%s" does not exist!' % old_file_path) |
| 61 if os.path.isfile(new_file_path): |
| 62 raise Exception('Config file for "%s" already exists!' % new_file_path) |
| 63 |
| 64 # Read the old builder configuration. |
| 65 with open(old_file_path) as old_file: |
| 66 old_file_contents = old_file.read() |
| 67 |
| 68 # Use "svn mv" to create the new file so that the diff only shows the changes. |
| 69 subprocess.call([SVN, 'mv', old_file_path, new_file_path]) |
| 70 |
| 71 # Replace any instances of old_name with new_name in file_contents. |
| 72 new_file_contents = old_file_contents.replace(old_name, new_name) |
| 73 |
| 74 # Write the new builder configuration. |
| 75 with open(new_file_path, 'w') as new_file: |
| 76 new_file.write(new_file_contents) |
| 77 |
| 78 |
| 79 def Usage(err_msg=None): |
| 80 if err_msg: |
| 81 print 'Error: %s\n' % err_msg |
| 82 print """rename_builders.py: Rename builders and update expectations files. |
| 83 |
| 84 Options: |
| 85 -h, --help Display this message. |
| 86 -t, --create-mapping-template Create a JSON template file for the mapping. |
| 87 |
| 88 Example usage: |
| 89 |
| 90 Create a mapping template file to be filled in manually. File name is optional, |
| 91 default is "mapping.json". |
| 92 $ rename_builders.py -t [filename] |
| 93 |
| 94 Map builders using a mapping file. |
| 95 $ rename_builders.py <filename> |
| 96 |
| 97 Interactive. Given each old builder name, prompts for the new name. |
| 98 $ rename_builders.py |
| 99 |
| 100 """ |
| 101 sys.exit(1) |
| 102 |
| 103 |
| 104 def main(argv): |
| 105 if len(argv) > 1: |
| 106 if argv[1] == '--help' or argv[1] == '-h': |
| 107 Usage() |
| 108 if argv[1] == '--create-mapping-template' or argv[1] == '-t': |
| 109 if len(argv) == 3: |
| 110 mapping_file = argv[2] |
| 111 elif len(argv) == 2: |
| 112 mapping_file = 'mapping.json' |
| 113 else: |
| 114 Usage('Too many arguments provided.') |
| 115 CreateMappingTemplate(mapping_file) |
| 116 return |
| 117 else: |
| 118 if len(argv) != 2: |
| 119 Usage('Too many arguments provided.') |
| 120 if not os.path.isfile(argv[1]): |
| 121 Usage('Please provide a mapping file.') |
| 122 with open(argv[1]) as f: |
| 123 mapping = json.load(f) |
| 124 for builder_file in ExistingBuilders(): |
| 125 if not builder_file in mapping: |
| 126 raise Exception('Your mapping file contains no mapping for "%s"' % \ |
| 127 builder_file) |
| 128 else: |
| 129 print ('No mapping file provided. Next time you can provide a ' |
| 130 'JSON-formatted file which maps old names to new names.') |
| 131 mapping = {} |
| 132 for builder_file in ExistingBuilders(): |
| 133 new_name = raw_input('New name for %s: ' % builder_file) |
| 134 mapping[builder_file] = new_name |
| 135 break |
| 136 |
| 137 for old_name, new_name in mapping.iteritems(): |
| 138 Rename(old_name, new_name) |
| 139 |
| 140 |
| 141 if '__main__' == __name__: |
| 142 sys.exit(main(sys.argv)) |
OLD | NEW |