bin/delete-merged-git-branches

   1 #!/usr/bin/python3
   2
   3 import argparse
   4 import pprint
   5 import sys
   6
   7 from distutils.util import strtobool
   8 from tailslib.git import Git
   9
  10 # Functions
  11
  12
  13 def yes_no_input(prompt, default=True):
  14     if default:
  15         options = 'Y/n'
  16     else:
  17         options = 'N/y'
  18     sys.stdout.write('%s [%s] ' % (prompt, options))
  19     while True:
  20         answer = input().lower()
  21         if len(answer) == 0:
  22             return default
  23         else:
  24             try:
  25                 return strtobool(answer)
  26             except ValueError:
  27                 print("Please respond with 'y' or 'n'.")
  28
  29
  30 # Parse command-line arguments
  31
  32 parser = argparse.ArgumentParser(description='Delete merged Git branches.')
  33 parser.add_argument('--repo', type=str, dest='repo',
  34                     help='Path to an up-to-date (bare) Tails Git repository.')
  35 parser.add_argument('--remote', type=str, dest='remote', default='origin',
  36                     help='Push to the specified remote instead of "origin".')
  37 parser.add_argument('--batch', type=bool, dest='batch',
  38                     nargs='?', const=True, default=False,
  39                     help='Assume "yes" as answer to all prompts.')
  40 args = parser.parse_args()
  41
  42
  43 # Main
  44
  45 pp = pprint.PrettyPrinter()
  46 tailsgit = Git(args.repo)
  47 branches_to_delete = tailsgit.branches_to_delete()
  48
  49 if not branches_to_delete:
  50     print("No branch to delete was found.")
  51     sys.exit(0)
  52
  53 print("The following branches will be deleted:")
  54 pp.pprint(sorted(branches_to_delete))
  55 if not args.batch \
  56    and not yes_no_input("Remove these branches?", default=False):
  57     sys.exit(0)
  58
  59 tailsgit.push(args.remote, [':%s' % branch for branch in branches_to_delete])