Add script to cleanup old gitlab branches (#5795)
This commit is contained in:
parent
a7a204ebca
commit
1ae794e5e4
4 changed files with 67 additions and 0 deletions
2
scripts/gitlab-branch-cleanup/.gitignore
vendored
Normal file
2
scripts/gitlab-branch-cleanup/.gitignore
vendored
Normal file
|
@ -0,0 +1,2 @@
|
|||
openrc
|
||||
venv
|
24
scripts/gitlab-branch-cleanup/README.md
Normal file
24
scripts/gitlab-branch-cleanup/README.md
Normal file
|
@ -0,0 +1,24 @@
|
|||
# gitlab-branch-cleanup
|
||||
|
||||
Cleanup old branches in a GitLab project
|
||||
|
||||
## Installation
|
||||
|
||||
```shell
|
||||
pip install -r requirements.txt
|
||||
python main.py --help
|
||||
```
|
||||
|
||||
## Usage
|
||||
|
||||
```console
|
||||
$ export GITLAB_API_TOKEN=foobar
|
||||
$ python main.py kargo-ci/kubernetes-sigs-kubespray
|
||||
Deleting branch pr-5220-containerd-systemd from 2020-02-17 ...
|
||||
Deleting branch pr-5561-feature/cinder_csi_fixes from 2020-02-17 ...
|
||||
Deleting branch pr-5607-add-flatcar from 2020-02-17 ...
|
||||
Deleting branch pr-5616-fix-typo from 2020-02-17 ...
|
||||
Deleting branch pr-5634-helm_310 from 2020-02-18 ...
|
||||
Deleting branch pr-5644-patch-1 from 2020-02-15 ...
|
||||
Deleting branch pr-5647-master from 2020-02-17 ...
|
||||
```
|
40
scripts/gitlab-branch-cleanup/main.py
Normal file
40
scripts/gitlab-branch-cleanup/main.py
Normal file
|
@ -0,0 +1,40 @@
|
|||
import gitlab
|
||||
import argparse
|
||||
import os
|
||||
import sys
|
||||
from datetime import timedelta, datetime, timezone
|
||||
|
||||
|
||||
from pprint import pprint
|
||||
|
||||
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()
|
1
scripts/gitlab-branch-cleanup/requirements.txt
Normal file
1
scripts/gitlab-branch-cleanup/requirements.txt
Normal file
|
@ -0,0 +1 @@
|
|||
python-gitlab
|
Loading…
Reference in a new issue