import gitlab import argparse import os import sys from datetime import timedelta, datetime, timezone parser = argparse.ArgumentParser( description='Cleanup old branches in a GitLab project') parser.add_argument('--api', default='https://gitlab.com/', help='URL of GitLab API, defaults to gitlab.com') parser.add_argument('--age', type=int, default=30, help='Delete branches older than this many days') parser.add_argument('--prefix', default='pr-', help='Cleanup only branches with names matching this prefix') parser.add_argument('--dry-run', action='store_true', help='Do not delete anything') parser.add_argument('project', help='Path of the GitLab project') args = parser.parse_args() limit = datetime.now(timezone.utc) - timedelta(days=args.age) if os.getenv('GITLAB_API_TOKEN', '') == '': print("Environment variable GITLAB_API_TOKEN is required.") sys.exit(2) gl = gitlab.Gitlab(args.api, private_token=os.getenv('GITLAB_API_TOKEN')) gl.auth() p = gl.projects.get(args.project) for b in p.branches.list(all=True): date = datetime.fromisoformat(b.commit['created_at']) if date < limit and not b.protected and not b.default and b.name.startswith(args.prefix): print("Deleting branch %s from %s ..." % (b.name, date.date().isoformat())) if not args.dry_run: b.delete()