Remove branches which are no more on remote

git-remove-stale

#!/usr/bin/env python

import subprocess
import re
import argparse

parser = argparse.ArgumentParser(description="Remove local branches, which are no longer available in the remote")
parser.add_argument("--do-it", action='store_true', help="Remove branches")
parser.add_argument("--remote", default="origin", help="Remote name")
args = parser.parse_args()

def find_local_branches():
  branches = subprocess.check_output(["git", "branch"])
  correct_branches = [];
  for line in branches.splitlines():
    # take out the active branch, we are only intersted in the name of the
    # branch
    branch_name = re.sub(r"\*", "", line).strip()

    # find out what is the remote of the branch
    try:
      remoteName = subprocess.check_output(["git", "config", "--get", "branch.%s.remote" % branch_name])
    except Exception as e:
      # Branch has no config"
      remoteName = ""

    remoteName = remoteName.strip();

    if "%s" % remoteName == args.remote:
      correct_branches.append(branch_name)

  return correct_branches


def find_remote_branches():
  branches = subprocess.check_output(["git", "branch", "-r"])
  correct_branches = [];

  for line in branches.splitlines():
    branch_name = line.strip();
    result = re.search(r"%s/([^\s]*)" % args.remote, branch_name)
    if result:
      correct_branches.append(result.group(1))

  return correct_branches


def remove_branches(branches):
  for branch_name in branches:
    if (args.do_it):
      print "Removing %s" % branch_name
      return_code = subprocess.call(["git", "branch", "-d", branch_name]);
      if return_code != 0:
        print "Unable to remove branch"
    else:
      print "Will remove %s" % branch_name

def find_to_remove(local=[], remote=[]):
  will_remove = []

  for branch_name in local_branches:
    try:
      index = remote_branches.index(branch_name)
    except ValueError:
      will_remove.append(branch_name)

  return will_remove


# test whether this is the git repo
try:
  subprocess.check_output(["git", "rev-parse", "--show-toplevel"]);
  is_git = True
except Exception:
  is_git = False

if is_git:
  # walk through the local branches
  # if local branch is not in remote branch list
  # prepare for removing
  remote_branches = find_remote_branches()
  local_branches = find_local_branches() 
  to_remove = find_to_remove(local=local_branches, remote=remote_branches);

  if to_remove:
    remove_branches(to_remove)
  else:
    print "No stale branches are found"