rebase all branches with branch

git-rebase-all.py

#!/usr/bin/env python

import argparse
import os
import subprocess
import sys


def rebase(base, branch, abort=True):
    try:
        subprocess.check_call([
            'git',
            'rebase',
            base,
            branch.strip(),
        ])
    except subprocess.CalledProcessError:
        if abort:
            subprocess.check_call([
                'git',
                'rebase',
                '--abort',
            ])
            return False
        else:
            raise
    return True


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-n', '--no-abort', dest='abort', action='store_false', default=True,
            help='do not rebase --abort if rebase requires intervention ')
    parser.add_argument('--base', type=str, default='master')
    args = parser.parse_args()

    results = {True: [], False: []}
    branches = subprocess.check_output(['git', 'branch'])
    for branch in branches.splitlines():
        br = branch[2:]  # trim first two spaces or space+asterisk
        success = rebase(args.base, br, args.abort)
        results[success].append(br.decode('utf-8'))

    print("successfully rebased {:d} branches".format(len(results[True])))
    failed = len(results[False])
    print("failed to rebase {:d} branch{}: {}".format(failed, '' if failed == 1 else 'es', ', '.join(results[False])))


if __name__ == '__main__':
    main()