You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
#!/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.
|
|
|
|
"""Lists branches with closed and abandoned issues."""
|
|
|
|
import optparse
|
|
import os
|
|
import sys
|
|
import urllib2
|
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
DEPOT_TOOLS_DIR = os.path.dirname(BASE_DIR)
|
|
sys.path.insert(0, DEPOT_TOOLS_DIR)
|
|
|
|
import git_cl
|
|
|
|
|
|
def get_branches():
|
|
"""Get list of all local git branches."""
|
|
branches = [l.split() for l in git_cl.RunGit(
|
|
["for-each-ref",
|
|
"--format=%(refname:short) %(upstream:short)",
|
|
"refs/heads"]).splitlines()]
|
|
return [Branch(*b) for b in branches]
|
|
|
|
def get_change_count(start, end):
|
|
return int(git_cl.RunGit(["rev-list", "%s..%s" % (start, end), "--count" ]))
|
|
|
|
|
|
class Branch(git_cl.Changelist):
|
|
def __init__(self, name, upstream=None):
|
|
git_cl.Changelist.__init__(self, branchref=name)
|
|
self._upstream = upstream
|
|
self._distance = None
|
|
self._issue_status = None
|
|
|
|
def GetStatus(self):
|
|
if not self._issue_status:
|
|
if self.GetIssue():
|
|
try:
|
|
issue_properties = self.RpcServer().get_issue_properties(
|
|
self.GetIssue(), None)
|
|
if issue_properties['closed']:
|
|
self._issue_status = 'closed'
|
|
else:
|
|
self._issue_status = 'pending'
|
|
except urllib2.HTTPError, e:
|
|
if e.code == 404:
|
|
self._issue_status = 'abandoned'
|
|
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()
|
|
if args:
|
|
parser.error('Unsupported arg: %s' % args)
|
|
|
|
branches = get_branches()
|
|
filtered = { 'closed' : [],
|
|
'empty' : [],
|
|
'pending' : [],
|
|
'abandoned' : [],
|
|
'no-issue' : []}
|
|
|
|
for branch in branches:
|
|
filtered[branch.GetStatus()].append(branch)
|
|
|
|
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
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|