forked from mirrors/gecko-dev
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
# 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/.
|
|
|
|
"""
|
|
Utilities for generating a decision task from :file:`.taskcluster.yml`.
|
|
"""
|
|
|
|
|
|
import os
|
|
|
|
import jsone
|
|
import slugid
|
|
import yaml
|
|
|
|
from .templates import merge
|
|
from .time import current_json_time
|
|
from .vcs import find_hg_revision_push_info
|
|
|
|
|
|
def make_decision_task(params, root, context, head_rev=None):
|
|
"""Generate a basic decision task, based on the root .taskcluster.yml"""
|
|
with open(os.path.join(root, ".taskcluster.yml"), "rb") as f:
|
|
taskcluster_yml = yaml.safe_load(f)
|
|
|
|
if not head_rev:
|
|
head_rev = params["head_rev"]
|
|
|
|
if params["repository_type"] == "hg":
|
|
pushlog = find_hg_revision_push_info(params["repository_url"], head_rev)
|
|
|
|
hg_push_context = {
|
|
"pushlog_id": pushlog["pushid"],
|
|
"pushdate": pushlog["pushdate"],
|
|
"owner": pushlog["user"],
|
|
}
|
|
else:
|
|
hg_push_context = {}
|
|
|
|
slugids = {}
|
|
|
|
def as_slugid(name):
|
|
# https://github.com/taskcluster/json-e/issues/164
|
|
name = name[0]
|
|
if name not in slugids:
|
|
slugids[name] = slugid.nice()
|
|
return slugids[name]
|
|
|
|
# provide a similar JSON-e context to what mozilla-taskcluster provides:
|
|
# https://docs.taskcluster.net/reference/integrations/mozilla-taskcluster/docs/taskcluster-yml
|
|
# but with a different tasks_for and an extra `cron` section
|
|
context = merge(
|
|
{
|
|
"repository": {
|
|
"url": params["repository_url"],
|
|
"project": params["project"],
|
|
"level": params["level"],
|
|
},
|
|
"push": merge(
|
|
{
|
|
"revision": params["head_rev"],
|
|
# remainder are fake values, but the decision task expects them anyway
|
|
"comment": " ",
|
|
},
|
|
hg_push_context,
|
|
),
|
|
"now": current_json_time(),
|
|
"as_slugid": as_slugid,
|
|
},
|
|
context,
|
|
)
|
|
|
|
rendered = jsone.render(taskcluster_yml, context)
|
|
if len(rendered["tasks"]) != 1:
|
|
raise Exception("Expected .taskcluster.yml to only produce one cron task")
|
|
task = rendered["tasks"][0]
|
|
|
|
task_id = task.pop("taskId")
|
|
return (task_id, task)
|