Add "archive" command to git_cl.py.

git_cl.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 # Copyright (C) 2008 Evan Martin <martine@danga.com>
 
 """A git-command for integrating reviews on Rietveld and Gerrit."""
 
 from distutils.version import LooseVersion
@@ skipping 2929 matching lines @@
   # Silence upload.py otherwise it becomes unwieldly.
   upload.verbosity = 0
 
   if fine_grained:
     # Process one branch synchronously to work through authentication, then
     # spawn processes to process all the other branches in parallel.
     if changes:
       fetch = lambda cl: (cl, cl.GetStatus())
       yield fetch(changes[0])
 
+      if len(changes) == 0:
+        # Exit early if there was only one branch to fetch.
+        return
+
       changes_to_fetch = changes[1:]
       pool = ThreadPool(
           min(max_processes, len(changes_to_fetch))
               if max_processes is not None
               else len(changes_to_fetch))
 
       fetched_cls = set()
       it = pool.imap_unordered(fetch, changes_to_fetch).__iter__()
       while True:
         try:
@@ skipping 106 matching lines @@
 
   print
   print 'Upload complete for dependent branches!'
   for dependent_branch in dependents:
     upload_status = 'failed' if failures.get(dependent_branch) else 'succeeded'
     print '  %s : %s' % (dependent_branch, upload_status)
   print
 
   return 0
 
+def CMDcleanup(parser, args):
+  """Archives and deletes branches associated with closed changelists."""
+  parser.add_option(
+      '-j', '--maxjobs', action='store', type=int,
+      help='The maximum number of jobs to use when retrieving review status')
+
+  parser.add_option(
+      '-f', action='store_true', dest='force', default=False,
+      help='Bypasses the confirmation prompt.')
+
+  auth.add_auth_options(parser)
+  options, args = parser.parse_args(args)
+  if args:
+    parser.error('Unsupported args: %s' % ' '.join(args))
+  auth_config = auth.extract_auth_config_from_options(options)
+
+  branches = RunGit(['for-each-ref', '--format=%(refname)', 'refs/heads'])
+  if not branches:
+    print 'No local branch found.'
+    return 0
+
+  print 'Finding all branches associated with closed issues...'
+  changes = [Changelist(branchref=b, auth_config=auth_config)
+              for b in branches.splitlines()]
+  alignment = max(5, max(len(c.GetBranch()) for c in changes))
+  statuses = get_cl_statuses(changes,
+                             fine_grained=True,
+                             max_processes=options.maxjobs)
+  proposal = [(cl.GetBranch(),
+               'git-cl-closed-%s-%s' % (cl.GetBranch(), cl.GetIssue()))
+              for cl, status in statuses
+              if status == 'closed']
+
+  if len(proposal) == 0:
+    print 'No branches with closed codereview issues found.'
+    return 0
+
+  current_branch = RunGit(['symbolic-ref', 'HEAD', '--short']).strip()

[tandrii(chromium), 2016/05/31 20:18:32]

use existing func:
current_branch = GetCurrentBranch()

[Kevin M, 2016/05/31 20:26:35]

Done.

+  print "Current branch: " + current_branch
+
+  if any(branch == current_branch for branch, _ in proposal):

[tandrii(chromium), 2016/05/31 20:18:33]

I'd print all the branches with closed issues first, and then do this check.
WDYT?

[Kevin M, 2016/05/31 20:26:35]

Done.

+    print 'You are currently using a branch associated with a closed'

[tandrii(chromium), 2016/05/31 20:18:33]

s/using a branch associated/on a branch %s associated

and %s is current_branch. And I'd not print it above, as in normal case, it's
not important to the command.

[Kevin M, 2016/05/31 20:26:35]

Done.

+    print 'codereview issue, so cleanup cannot proceed. Please change to '

[tandrii(chromium), 2016/05/31 20:18:33]

use 1 print statement (basically, call) and don't format this yourself:

print ('You are currently using a branch associated with a closed '
       'codereview issue, so cleanup cannot proceed. ....
       '...')

[Kevin M, 2016/05/31 20:26:35]

Done.

+    print 'another branch and run \'git cl cleanup\' again.'
+    return 1
+
+  print '\nBranches with closed issues that will be tagged and deleted:'
+  print '%*s | %s' % (alignment, 'Branch name', 'Archival tag name')
+  for next_item in proposal:
+    print '%*s   %s' % (alignment, next_item[0], next_item[1])
+
+  if not options.force:
+    if ask_for_data('\nProceed with deletion (Y/N)? ').lower() != 'y':
+      print 'Aborted.'
+      return 1
+
+  for branch, tagname in proposal:
+    RunGit(['tag', tagname, branch])
+    RunGit(['branch', '-D', branch])
+  print '\nJob\'s done!'
+
+  return 0
 
 def CMDstatus(parser, args):
   """Show status of changelists.
 
   Colors are used to tell the state of the CL unless --fast is used:
     - Red      not sent for review or broken
     - Blue     waiting for review
     - Yellow   waiting for you to reply to review
     - Green    LGTM'ed
     - Magenta  in the commit queue
@@ skipping 1801 matching lines @@
 if __name__ == '__main__':
   # These affect sys.stdout so do it outside of main() to simplify mocks in
   # unit testing.
   fix_encoding.fix_encoding()
   setup_color.init()
   try:
     sys.exit(main(sys.argv[1:]))
   except KeyboardInterrupt:
     sys.stderr.write('interrupted\n')
     sys.exit(1)