forked from cms-sw/cms-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
add-externals-gh-labels.py
executable file
·103 lines (95 loc) · 4.53 KB
/
add-externals-gh-labels.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
#!/usr/bin/env python
from github import Github
from os.path import expanduser, dirname, abspath, join
from githublabels import LABEL_TYPES, COMMON_LABELS, COMPARISON_LABELS, CMSSW_BUILD_LABELS
from categories import COMMON_CATEGORIES, EXTERNAL_CATEGORIES, EXTERNAL_REPOS, CMSSW_REPOS, CMSDIST_REPOS, CMSSW_CATEGORIES
from datetime import datetime
from socket import setdefaulttimeout
from github_utils import api_rate_limits
from sys import argv
setdefaulttimeout(120)
SCRIPT_DIR = dirname(abspath(argv[0]))
def setRepoLabels (gh, repo_name, all_labels, dryRun=False):
repos = []
if not "/" in repo_name:
user = gh.get_user(repo_name)
for repo in user.get_repos():
repos.append(repo)
else:
repos.append(gh.get_repo(repo_name))
api_rate_limits(gh)
for repo in repos:
print "Checking repository ", repo.full_name
cur_labels = {}
for lab in repo.get_labels():
cur_labels [lab.name]=lab
api_rate_limits(gh)
for lab in all_labels:
if not lab in cur_labels:
print " Creating new label ",lab,"=>",all_labels[lab]
if not dryRun:
repo.create_label(lab, all_labels[lab])
api_rate_limits(gh)
elif cur_labels[lab].color != all_labels[lab]:
if not dryRun:
cur_labels[lab].edit(lab, all_labels[lab])
api_rate_limits(gh)
print " Label ",lab," color updatd: ",cur_labels[lab].color ," => ",all_labels[lab]
if __name__ == "__main__":
from optparse import OptionParser
parser = OptionParser(usage="%prog [-n|--dry-run] [-e|--externals] [-c|--cmssw] [-d|--cmsdist] [-a|--all]")
parser.add_option("-n", "--dry-run", dest="dryRun", action="store_true", help="Do not modify Github", default=False)
parser.add_option("-e", "--externals", dest="externals", action="store_true", help="Only process CMS externals repositories", default=False)
parser.add_option("-u", "--userss", dest="users", action="store_true", help="Only process Users externals repositories", default=False)
parser.add_option("-c", "--cmssw", dest="cmssw", action="store_true", help="Only process "+",".join(CMSSW_REPOS)+" repository", default=False)
parser.add_option("-d", "--cmsdist", dest="cmsdist", action="store_true", help="Only process "+",".join(CMSDIST_REPOS)+" repository", default=False)
parser.add_option("-a", "--all", dest="all", action="store_true", help="Process all CMS repository i.e. externals, cmsdist and cmssw", default=False)
opts, args = parser.parse_args()
if opts.all:
opts.externals = True
opts.cmssw = True
opts.cmsdist = True
elif (not opts.externals) and (not opts.cmssw) and (not opts.cmsdist) and (not opts.users):
parser.error("Too few arguments, please use either -e, -c, -u or -d")
import repo_config
gh = Github(login_or_token=open(expanduser(repo_config.GH_TOKEN)).read().strip())
api_rate_limits(gh)
if opts.externals:
all_labels = COMMON_LABELS
for cat in COMMON_CATEGORIES+EXTERNAL_CATEGORIES:
for lab in LABEL_TYPES:
all_labels[cat+"-"+lab]=LABEL_TYPES[lab]
for repo_name in EXTERNAL_REPOS:
setRepoLabels (gh, repo_name, all_labels, opts.dryRun)
if opts.cmssw:
all_labels = COMMON_LABELS
for lab in COMPARISON_LABELS:
all_labels[lab] = COMPARISON_LABELS[lab]
for lab in CMSSW_BUILD_LABELS:
all_labels[lab] = CMSSW_BUILD_LABELS[lab]
for cat in COMMON_CATEGORIES+CMSSW_CATEGORIES.keys():
for lab in LABEL_TYPES:
all_labels[cat+"-"+lab]=LABEL_TYPES[lab]
for repo_name in CMSSW_REPOS:
setRepoLabels (gh, repo_name, all_labels, opts.dryRun)
if opts.cmsdist:
all_labels = COMMON_LABELS
for cat in COMMON_CATEGORIES+CMSSW_CATEGORIES.keys():
for lab in LABEL_TYPES:
all_labels[cat+"-"+lab]=LABEL_TYPES[lab]
for repo_name in CMSDIST_REPOS:
setRepoLabels (gh, repo_name, all_labels, opts.dryRun)
if opts.users:
from glob import glob
for rconf in glob(join(SCRIPT_DIR,"repos","*","*","repo_config.py")):
repo_data = join(rconf.split("/")[-4:-1])
exec 'from '+".".join(repo_data)+' import categories, repo_config'
print repo_config.GH_TOKEN
gh = Github(login_or_token=open(expanduser(repo_config.GH_TOKEN)).read().strip())
all_labels = COMMON_LABELS
for lab in COMPARISON_LABELS:
all_labels[lab] = COMPARISON_LABELS[lab]
for cat in categories.COMMON_CATEGORIES+categories.CMSSW_CATEGORIES.keys():
for lab in LABEL_TYPES:
all_labels[cat+"-"+lab]=LABEL_TYPES[lab]
setRepoLabels (gh, "/".join(repo_data[-2:]), all_labels, opts.dryRun)