reset_patched_repos.py

#!/usr/bin/env python

import argparse
import json
import subprocess


def reset_patched_repos(dirs):
  for patch_dir, repo in dirs.iteritems():
    args = ['git', '-C', repo, 'am', '--abort']
    subprocess.call(args) # ignore failure
    args = ['git', '-C', repo, 'checkout', 'refs/patches/upstream-head']
    subprocess.call(args) # ignore failure


def parse_args():
  parser = argparse.ArgumentParser(description='Reset patched repos')
  parser.add_argument('config', nargs='+',
                      type=argparse.FileType('r'),
                      help='patches\' config(s) in the JSON format')
  return parser.parse_args()


def main():
  configs = parse_args().config
  for config_json in configs:
    reset_patched_repos(json.load(config_json))


if __name__ == '__main__':
  main()