src/git_stats/branch.py

   1 #!/usr/bin/env python
   2
   3 import os
   4 import sys
   5
   6 from optparse import OptionParser
   7 from git import Repo
   8
   9 from git_stats import parse
  10
  11 def belongsTo(commit):
  12   """
  13   """
  14
  15   git = Repo(".").git
  16
  17   result = git.for_each_ref("refs/heads", format="%(objectname)")
  18
  19   branches = result.split('\n')
  20
  21   if commit in branches:
  22     print("Easy as pie, that's a branch head!")
  23     return
  24
  25   results = {}
  26   rev_lists = {}
  27
  28   for branch in branches:
  29     result = git.rev_list(branch)
  30     commits = result.split('\n')
  31
  32     # No need to keep going if the commit is not part of this branch
  33     if not commit in commits:
  34       continue
  35
  36     result = git.rev_list("--no-merges", branch)
  37     no_merges = result.split('\n')
  38
  39     # Make the ommits into a set
  40     commits = set(commits)
  41     no_merges = set(no_merges)
  42
  43     # Store all commits for later use
  44     rev_lists[branch] = commits
  45
  46     # The merges are 'all_commits - no_merges'
  47     merges = commits.difference(no_merges)
  48
  49     dilution = 0
  50
  51     # Check all the merges to see if they diluted this branch
  52     for merge in merges:
  53       result = git.cat_file("-p", merge)
  54       log = result.split('\n')
  55
  56       # We really only have to check the first parent
  57       # If it's not in there, it is a dilution (for this branch at least)
  58       for line in log:
  59         if not line.startswith("parent "):
  60           continue
  61
  62         parent = line[7:]
  63         break
  64
  65       result = git.rev_list(parent)
  66       children = result.split('\n')
  67
  68       # Diluted iff the commit is not a child of the primary parent
  69       if not commit in children:
  70         dilution += 1
  71
  72     results[branch] =  dilution
  73
  74   print(results)
  75
  76   min = -1
  77   champs = []
  78
  79   for branch in results.iteritems():
  80     if min == -1:
  81       min = branch[1]
  82
  83     if branch[1] > min:
  84       continue
  85
  86     # Clear the old ones if we have a better set
  87     if branch[1] < min:
  88       champs = []
  89       min = branch[1]
  90
  91     champs.append(branch[0])
  92
  93   result = []
  94
  95   # Loop over all the champs and exclude any subsets
  96   for champ in champs:
  97     champlist = rev_lists[champ]
  98     for otherchamp in champs:
  99       # Don't compare to self
 100       if champ == otherchamp:
 101         continue
 102
 103       otherlist = rev_lists[otherchamp]
 104       if champlist.issubset(otherlist):
 105         break
 106     else:
 107       result.append(champ)
 108
 109   print(result)
 110
 111
 112 def branchcontains(branch, commit):
 113   """returns whether the specified branch contains the specified commit.
 114
 115   params:
 116     branch: the branch.
 117     commit: the commit.
 118
 119   returns:
 120     whether the branch contains the commit.
 121   """
 122
 123   git = git(".")
 124   arg = branch + ".." + commit
 125   result = git.rev_list(arg)
 126
 127   if result:
 128     # if there is a difference between these sets, the commit is not in the branch
 129     return false
 130   else:
 131     # there is no difference between the two, thus the branch contains the commit
 132     return true
 133
 134 def branchList(commitFilter, includeRemotes=False):
 135   """
 136   """
 137
 138   git = Repo(".").git
 139
 140   args = []
 141
 142   if includeRemotes:
 143     args.append("-a")
 144
 145   if commitFilter:
 146     args.append("--contains")
 147     args.append(commitFilter)
 148
 149   result = git.branch(*args)
 150
 151   branches = result.split('\n')
 152
 153   result = []
 154
 155   for branch in branches:
 156     result.append(branch[2:])
 157
 158   return result
 159
 160 def dispatch(*args):
 161   """Dispatches branch related commands
 162   """
 163
 164   progname = os.path.basename(sys.argv[0]) + " branch"
 165
 166   parser = OptionParser(option_class=parse.GitOption, prog=progname)
 167
 168
 169   parser.add_option(
 170       "-b", "--belongs-to",
 171       type="commit",
 172       metavar="COMMIT",
 173       help="find out which branch the specified commit belongs to")
 174
 175   parser.add_option(
 176       "-c", "--contains",
 177       type="commit",
 178       help="show only branches that contain the specified commit")
 179
 180   parser.add_option(
 181       "-r", "--remotes",
 182       action="store_true",
 183       help="include remotes in the listing")
 184
 185   (options, args) = parser.parse_args(list(args))
 186
 187   if options.belongs_to:
 188     result = belongsTo(options.belongs_to)
 189   else:
 190     result = branchList(commitFilter=options.contains,
 191                         includeRemotes=options.remotes)
 192
 193     print("Matching branches:")
 194
 195     for branch in result:
 196       print(branch)
 197