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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import re
from typing import Literal, Union
from taskgraph.util.attributes import _match_run_on
INTEGRATION_PROJECTS = {
"autoland",
}
TRUNK_PROJECTS = INTEGRATION_PROJECTS | {"mozilla-central", "comm-central"}
# Mapping of project to list of branches that should be considered "release"
# level. A value of `True` means all branches are considered release
# (used by hg.mozilla.org based projects).
PROJECT_RELEASE_BRANCHES: dict[str, Union[list[str], Literal[True]]] = {
# https://github.com/mozilla-firefox/firefox
"firefox": [
"main",
"beta",
"release",
"esr115",
"esr128",
"esr140",
],
"mozilla-central": True,
"mozilla-beta": True,
"mozilla-release": True,
"mozilla-esr115": True,
"mozilla-esr128": True,
"mozilla-esr140": True,
"comm-central": True,
"comm-beta": True,
"comm-release": True,
"comm-esr115": True,
"comm-esr128": True,
"comm-esr140": True,
# bug 1845368: pine is a permanent project branch used for testing
# nightly updates
"pine": True,
# bug 1877483: larch has similar needs for nightlies
"larch": True,
# maple is also an L3 branch: https://phabricator.services.mozilla.com/D184833
"maple": True,
# bug 1988213: cypress project branch
"cypress": True,
}
RELEASE_PROJECTS = set(PROJECT_RELEASE_BRANCHES)
RELEASE_PROMOTION_PROJECTS = {
"jamun",
"maple",
"try",
"try-comm-central",
} | RELEASE_PROJECTS
TEMPORARY_PROJECTS = set({
# When using a "Disposable Project Branch" you can specify your branch here. e.g.:
"oak",
})
TRY_PROJECTS = {
"staging-firefox", # https://github.com/mozilla-releng/staging-firefox
"try",
"try-comm-central",
}
ALL_PROJECTS = RELEASE_PROMOTION_PROJECTS | TRUNK_PROJECTS | TEMPORARY_PROJECTS
RUN_ON_PROJECT_ALIASES = {
# key is alias, value is lambda to test it against
"all": lambda params: True,
"integration": lambda params: (
params["project"] in INTEGRATION_PROJECTS or params["project"] == "toolchains"
),
"release": lambda params: (
release_level(params) == "production" or params["project"] == "toolchains"
),
"trunk": lambda params: (
params["project"] in TRUNK_PROJECTS or params["project"] == "toolchains"
),
"trunk-only": lambda params: params["project"] in TRUNK_PROJECTS,
"autoland": lambda params: params["project"] in ("autoland", "toolchains"),
"autoland-only": lambda params: params["project"] == "autoland",
"mozilla-central": lambda params: params["project"]
in ("mozilla-central", "toolchains"),
"mozilla-central-only": lambda params: params["project"] == "mozilla-central",
}
_COPYABLE_ATTRIBUTES = (
"accepted-mar-channel-ids",
"artifact_map",
"artifact_prefix",
"build_platform",
"build_type",
"l10n_chunk",
"locale",
"mar-channel-id",
"maven_packages",
"nightly",
"required_signoffs",
"shippable",
"shipping_phase",
"shipping_product",
"signed",
"stub-installer",
"update-channel",
)
def match_run_on_projects(params, run_on_projects):
"""Determine whether the given project is included in the `run-on-projects`
parameter, applying expansions for things like "integration" mentioned in
the attribute documentation."""
aliases = RUN_ON_PROJECT_ALIASES.keys()
run_aliases = set(aliases) & set(run_on_projects)
if run_aliases:
if any(RUN_ON_PROJECT_ALIASES[alias](params) for alias in run_aliases):
return True
return params["project"] in run_on_projects
def match_run_on_hg_branches(hg_branch, run_on_hg_branches):
"""Determine whether the given project is included in the `run-on-hg-branches`
parameter. Allows 'all'."""
if "all" in run_on_hg_branches:
return True
for expected_hg_branch_pattern in run_on_hg_branches:
if re.match(expected_hg_branch_pattern, hg_branch):
return True
return False
match_run_on_repo_type = _match_run_on
def copy_attributes_from_dependent_job(dep_job, denylist=()):
return {
attr: dep_job.attributes[attr]
for attr in _COPYABLE_ATTRIBUTES
if attr in dep_job.attributes and attr not in denylist
}
def sorted_unique_list(*args):
"""Join one or more lists, and return a sorted list of unique members"""
combined = set().union(*args)
return sorted(combined)
def release_level(params):
"""
Whether this is a staging release or not.
:return str: One of "production" or "staging".
"""
if branches := PROJECT_RELEASE_BRANCHES.get(params.get("project")):
if branches is True:
return "production"
m = re.match(r"refs/heads/(\S+)$", params["head_ref"])
if m is not None and m.group(1) in branches:
return "production"
return "staging"
def is_try(params):
"""
Determine whether this graph is being built on a try project or for
`mach try fuzzy`.
"""
return "try" in params["project"] or params["try_mode"] == "try_select"
def task_name(task):
if task.label.startswith(task.kind + "-"):
return task.label[len(task.kind) + 1 :]
raise AttributeError(f"Task {task.label} does not have a name.")
|
