| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2014 The Chromium Authors. All rights reserved. | 2 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 """Provides a short mapping of all the branches in your local repo, organized | 6 """Provides a short mapping of all the branches in your local repo, organized |
| 7 by their upstream ('tracking branch') layout. | 7 by their upstream ('tracking branch') layout. |
| 8 | 8 |
| 9 Example: | 9 Example: |
| 10 origin/master | 10 origin/master |
| 11 cool_feature | 11 cool_feature |
| 12 dependent_feature | 12 dependent_feature |
| 13 other_dependent_feature | 13 other_dependent_feature |
| 14 other_feature | 14 other_feature |
| 15 | 15 |
| 16 Branches are colorized as follows: | 16 Branches are colorized as follows: |
| 17 * Red - a remote branch (usually the root of all local branches) | 17 * Red - a remote branch (usually the root of all local branches) |
| 18 * Cyan - a local branch which is the same as HEAD | 18 * Cyan - a local branch which is the same as HEAD |
| 19 * Note that multiple branches may be Cyan, if they are all on the same | 19 * Note that multiple branches may be Cyan, if they are all on the same |
| 20 commit, and you have that commit checked out. | 20 commit, and you have that commit checked out. |
| 21 * Green - a local branch | 21 * Green - a local branch |
| 22 * Blue - a 'branch-heads' branch | 22 * Blue - a 'branch-heads' branch |
| 23 * Magenta - a tag | 23 * Magenta - a tag |
| 24 * Magenta '{NO UPSTREAM}' - If you have local branches which do not track any | 24 * Magenta '{NO UPSTREAM}' - If you have local branches which do not track any |
| 25 upstream, then you will see this. | 25 upstream, then you will see this. |
| 26 """ | 26 """ |
| 27 | 27 |
| 28 import argparse | 28 import argparse |
| 29 import collections | 29 import collections |
| 30 import os |
| 30 import sys | 31 import sys |
| 31 import subprocess2 | 32 import subprocess2 |
| 32 | 33 |
| 33 from third_party import colorama | 34 from third_party import colorama |
| 34 from third_party.colorama import Fore, Style | 35 from third_party.colorama import Fore, Style |
| 35 | 36 |
| 36 from git_common import current_branch, upstream, tags, get_branches_info | 37 from git_common import current_branch, upstream, tags, get_branches_info |
| 37 from git_common import get_git_version, MIN_UPSTREAM_TRACK_GIT_VERSION, hash_one | 38 from git_common import get_git_version, MIN_UPSTREAM_TRACK_GIT_VERSION, hash_one |
| 38 from git_common import run | 39 from git_common import run |
| 39 | 40 |
| (...skipping 224 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 264 if self.show_subject: | 265 if self.show_subject: |
| 265 line.append(run('log', '-n1', '--format=%s', branch)) | 266 line.append(run('log', '-n1', '--format=%s', branch)) |
| 266 | 267 |
| 267 self.output.append(line) | 268 self.output.append(line) |
| 268 | 269 |
| 269 for child in sorted(self.__parent_map.pop(branch, ())): | 270 for child in sorted(self.__parent_map.pop(branch, ())): |
| 270 self.__append_branch(child, depth=depth + 1) | 271 self.__append_branch(child, depth=depth + 1) |
| 271 | 272 |
| 272 | 273 |
| 273 def main(argv): | 274 def main(argv): |
| 274 colorama.init() | 275 colorama.init(wrap="TERM" not in os.environ) |
| 275 if get_git_version() < MIN_UPSTREAM_TRACK_GIT_VERSION: | 276 if get_git_version() < MIN_UPSTREAM_TRACK_GIT_VERSION: |
| 276 print >> sys.stderr, ( | 277 print >> sys.stderr, ( |
| 277 'This tool will not show all tracking information for git version ' | 278 'This tool will not show all tracking information for git version ' |
| 278 'earlier than ' + | 279 'earlier than ' + |
| 279 '.'.join(str(x) for x in MIN_UPSTREAM_TRACK_GIT_VERSION) + | 280 '.'.join(str(x) for x in MIN_UPSTREAM_TRACK_GIT_VERSION) + |
| 280 '. Please consider upgrading.') | 281 '. Please consider upgrading.') |
| 281 | 282 |
| 282 parser = argparse.ArgumentParser( | 283 parser = argparse.ArgumentParser( |
| 283 description='Print a a tree of all branches parented by their upstreams') | 284 description='Print a a tree of all branches parented by their upstreams') |
| 284 parser.add_argument('-v', action='count', | 285 parser.add_argument('-v', action='count', |
| (...skipping 16 matching lines...) Expand all Loading... |
| 301 mapper.start() | 302 mapper.start() |
| 302 print mapper.output.as_formatted_string() | 303 print mapper.output.as_formatted_string() |
| 303 return 0 | 304 return 0 |
| 304 | 305 |
| 305 if __name__ == '__main__': | 306 if __name__ == '__main__': |
| 306 try: | 307 try: |
| 307 sys.exit(main(sys.argv[1:])) | 308 sys.exit(main(sys.argv[1:])) |
| 308 except KeyboardInterrupt: | 309 except KeyboardInterrupt: |
| 309 sys.stderr.write('interrupted\n') | 310 sys.stderr.write('interrupted\n') |
| 310 sys.exit(1) | 311 sys.exit(1) |
| OLD | NEW |