gecko-dev/taskcluster/taskgraph/util/treeherder.py
Tom Prince 7b13402d7e Bug 1456234: [taskgraph]: Add a helper for adding a suffix to a possibly grouped treeherder symbol; r=dustin.
Differential Revision: https://phabricator.services.mozilla.com/D1286

--HG--
extra : rebase_source : 3cb1080e5a94e3e9e5b5a3f6909b0e8d636de09c
extra : source : 4a182c3447fa01c75fa8a231e067bebd83683da9
2018-05-15 13:00:18 -06:00

32 lines
1.1 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/.
from __future__ import absolute_import, print_function, unicode_literals
import re
def split_symbol(treeherder_symbol):
"""Split a symbol expressed as grp(sym) into its two parts. If no group is
given, the returned group is '?'"""
groupSymbol = '?'
symbol = treeherder_symbol
if '(' in symbol:
groupSymbol, symbol = re.match(r'([^(]*)\(([^)]*)\)', symbol).groups()
return groupSymbol, symbol
def join_symbol(group, symbol):
"""Perform the reverse of split_symbol, combining the given group and
symbol. If the group is '?', then it is omitted."""
if group == '?':
return symbol
return '{}({})'.format(group, symbol)
def add_suffix(treeherder_symbol, chunk):
"""Add a suffix to a treeherder symbol that may contain a group."""
group, symbol = split_symbol(treeherder_symbol)
symbol += str(chunk)
return join_symbol(group, symbol)