OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2010 Google Inc. All Rights Reserved. | 2 # Copyright 2010 Google Inc. All Rights Reserved. |
3 # | 3 # |
4 # Licensed under the Apache License, Version 2.0 (the "License"); | 4 # Licensed under the Apache License, Version 2.0 (the "License"); |
5 # you may not use this file except in compliance with the License. | 5 # you may not use this file except in compliance with the License. |
6 # You may obtain a copy of the License at | 6 # You may obtain a copy of the License at |
7 # | 7 # |
8 # http://www.apache.org/licenses/LICENSE-2.0 | 8 # http://www.apache.org/licenses/LICENSE-2.0 |
9 # | 9 # |
10 # Unless required by applicable law or agreed to in writing, software | 10 # Unless required by applicable law or agreed to in writing, software |
(...skipping 1043 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1054 elif command == 'stats': | 1054 elif command == 'stats': |
1055 print http_archive.stats(options.command, options.host, options.full_path) | 1055 print http_archive.stats(options.command, options.host, options.full_path) |
1056 elif command == 'merge': | 1056 elif command == 'merge': |
1057 if not options.merged_file: | 1057 if not options.merged_file: |
1058 print 'Error: Must specify a merged file name (use --merged_file)' | 1058 print 'Error: Must specify a merged file name (use --merged_file)' |
1059 return | 1059 return |
1060 http_archive.merge(options.merged_file, args[2:]) | 1060 http_archive.merge(options.merged_file, args[2:]) |
1061 elif command == 'edit': | 1061 elif command == 'edit': |
1062 http_archive.edit(options.command, options.host, options.full_path) | 1062 http_archive.edit(options.command, options.host, options.full_path) |
1063 http_archive.Persist(replay_file) | 1063 http_archive.Persist(replay_file) |
| 1064 elif command == 'remap-host': |
| 1065 old_host = args[2] |
| 1066 new_host = args[3] |
| 1067 update_count = 0 |
| 1068 for req in http_archive: |
| 1069 if req.host == old_host: |
| 1070 req.host = new_host |
| 1071 update_count += 1 |
| 1072 updated_file_path = replay_file + '.updated' |
| 1073 http_archive.Persist(updated_file_path) |
| 1074 print 'Updated %i hosts of address %s to address %s' % ( |
| 1075 update_count, old_host, new_host) |
| 1076 print 'The updated WPR is saved as %s' % updated_file_path |
1064 else: | 1077 else: |
1065 option_parser.error('Unknown command "%s"' % command) | 1078 option_parser.error('Unknown command "%s"' % command) |
1066 return 0 | 1079 return 0 |
1067 | 1080 |
1068 | 1081 |
1069 if __name__ == '__main__': | 1082 if __name__ == '__main__': |
1070 sys.exit(main()) | 1083 sys.exit(main()) |
OLD | NEW |