From a3800b2db57cc9b2d223a969cbf7d7bf33d07225 Mon Sep 17 00:00:00 2001
From: "groby@chromium.org"
 <groby@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>
Date: Thu, 2 May 2013 01:41:35 +0000
Subject: [PATCH] Allow branches without upstream.

Clean up formatting a bit.

Review URL: https://chromiumcodereview.appspot.com/14602017

git-svn-id: svn://svn.chromium.org/chrome/trunk/tools/depot_tools@197773 0039d316-1c4b-4281-b951-d872f2087c98
---
 git_utils/git-tree-prune | 53 +++++++++++++++++++++++-----------------
 1 file changed, 30 insertions(+), 23 deletions(-)

diff --git a/git_utils/git-tree-prune b/git_utils/git-tree-prune
index e8fb2812f..556183d87 100755
--- a/git_utils/git-tree-prune
+++ b/git_utils/git-tree-prune
@@ -30,7 +30,7 @@ def get_change_count(start, end):
 
 
 class Branch(git_cl.Changelist):
-  def __init__(self, name, upstream):
+  def __init__(self, name, upstream=None):
     git_cl.Changelist.__init__(self, branchref=name)
     self._upstream = upstream
     self._distance = None
@@ -52,23 +52,36 @@ class Branch(git_cl.Changelist):
       else:
         self._issue_status = 'no-issue'
       if (self._issue_status != 'pending'
+         and self._upstream
          and not self.GetDistance()[0]
          and not self._upstream.startswith("origin/")):
         self._issue_status = 'empty'
     return self._issue_status
     
   def GetDistance(self):
+    if self._upstream is None:
+      return None;
     if not self._distance:
       self._distance = [get_change_count(self._upstream, self.GetBranch()),
           get_change_count(self.GetBranch(), self._upstream)]
     return self._distance
 
   def GetDistanceInfo(self):
+    if not self._upstream:
+      return "<No upstream branch>"
     formatted_dist = ", ".join(["%s %d" % (x,y)
         for (x,y) in zip(["ahead","behind"], self.GetDistance()) if y])
     return "[%s%s]" % (
         self._upstream, ": " + formatted_dist if formatted_dist else "")
 
+def print_branches(title, fmt, branches):
+  if branches:
+    print title
+    for branch in branches:
+      print fmt.format(branch=branch.GetBranch(),
+                       issue=branch.GetIssue(),
+                       distance=branch.GetDistanceInfo())
+
 def main():
   parser = optparse.OptionParser(usage=sys.modules['__main__'].__doc__)
   options, args = parser.parse_args()
@@ -85,28 +98,22 @@ def main():
   for branch in branches:
     filtered[branch.GetStatus()].append(branch)
 
-  print "# Branches with closed issues"
-  for branch in filtered['closed']:
-    print "git branch -D %s # Issue %s is closed." % (branch.GetBranch(),
-                                                        branch.GetIssue())
-
-  print "# Empty branches"
-  for branch in filtered['empty']:
-    print "git branch -D %s # Empty." % (branch.GetBranch())
-
-  print "\n# Pending Branches"
-  for branch in filtered['pending']:
-    print "# Branch %s - Issue %s - %s" % (
-        branch.GetBranch(), branch.GetIssue(), branch.GetDistanceInfo())
-
-  print "\n# Branches with abandoned issues"
-  for branch in filtered['abandoned']:
-    print "# Branch %s - was issue %s - %s" % (
-        branch.GetBranch(), branch.GetIssue(), branch.GetDistanceInfo())
-
-  print "\n# Branches without associated issues"
-  for branch in filtered['no-issue']:
-    print "# Branch %s - %s" % (branch.GetBranch(), branch.GetDistanceInfo())
+  print_branches("# Branches with closed issues",
+                 "git branch -D {branch} # Issue {issue} is closed.",
+                 filtered['closed'])
+  print_branches("\n# Empty branches",
+                 "git branch -D {branch} # Empty.",
+                 filtered['empty'])
+  print_branches("\n# Pending Branches",
+                 "# Branch {branch} - Issue {issue} - {distance}",
+                 filtered['pending']);
+  print_branches("\n# Branches with abandoned issues",
+                 "# Branch {branch} - was issue {issue} - {distance}",
+                 filtered['abandoned'])
+
+  print_branches("\n# Branches without associated issues",
+                 "# Branch {branch} - {distance}",
+                 filtered['no-issue'])
 
   return 0