Revision control
Copy as Markdown
# 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
import os
import re
from taskgraph.filter_tasks import filter_task
REPO_RE = r'((?P<owner>[\.\w-]+)[/:])?(?P<branch>[\.\w-]+)'
FIREFOX_IOS_BRANCH_RE = re.compile(r'\[firefox-ios:\s*' + REPO_RE + r'\]')
FIREFOX_ANDROID_BRANCH_RE = re.compile(r'\[firefox-android:\s*' + REPO_RE + r'\]')
def calc_branch_build_param(parameters):
title = os.environ.get("APPSERVICES_PULL_REQUEST_TITLE", "")
branch_build = {}
firefox_android_branch_match = FIREFOX_ANDROID_BRANCH_RE.search(title)
if firefox_android_branch_match:
branch_build['firefox-android'] = {
'owner': calc_owner(firefox_android_branch_match),
'branch': firefox_android_branch_match.group('branch'),
}
firefox_ios_branch_match = FIREFOX_IOS_BRANCH_RE.search(title)
if firefox_ios_branch_match:
branch_build['firefox-ios'] = {
'owner': calc_owner(firefox_ios_branch_match),
'branch': firefox_ios_branch_match.group('branch'),
}
return branch_build
def calc_owner(match):
if match.group('owner'):
return match.group('owner')
else:
return 'mozilla-mobile'
@filter_task("branch-build")
def filter_branch_build_tasks(full_task_graph, parameters, graph_config):
def should_keep_task(task):
task_branch_build = task.attributes.get("branch-build")
if task_branch_build is None:
# Don't filter out tasks without a `branch-build` attribute
return True
else:
# For tasks with a `branch-build` attribute, include them if there's a matching
# parameter
return task_branch_build in parameters.get('branch-build', {})
return [l for l, task in full_task_graph.tasks.items() if should_keep_task(task)]