mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-19 00:05:36 +00:00
28b0ebdf63
This changes templates so they return an entire context dict instead of only returning context based on their name. For example, now the 'path' template can set context for 'env'. A side effect of this is that there is no longer a 1-to-1 mapping of templates in tryselect and taskgraph. MozReview-Commit-ID: IHRXXi5mB4G --HG-- extra : rebase_source : 4d7e398e60598a5de7961fb126f1d05a0b983681
126 lines
4.1 KiB
Python
126 lines
4.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 os
|
|
import subprocess
|
|
import tempfile
|
|
from argparse import ArgumentParser
|
|
|
|
from .templates import all_templates
|
|
|
|
|
|
COMMON_ARGUMENT_GROUPS = {
|
|
'push': [
|
|
[['-m', '--message'],
|
|
{'const': 'editor',
|
|
'default': '{msg}',
|
|
'nargs': '?',
|
|
'help': 'Use the specified commit message, or create it in your '
|
|
'$EDITOR if blank. Defaults to computed message.',
|
|
}],
|
|
[['--no-push'],
|
|
{'dest': 'push',
|
|
'action': 'store_false',
|
|
'help': 'Do not push to try as a result of running this command (if '
|
|
'specified this command will only print calculated try '
|
|
'syntax and selection info).',
|
|
}],
|
|
[['--closed-tree'],
|
|
{'action': 'store_true',
|
|
'default': False,
|
|
'help': 'Push despite a closed try tree',
|
|
}],
|
|
],
|
|
'preset': [
|
|
[['--save'],
|
|
{'default': None,
|
|
'help': 'Save selection for future use with --preset.',
|
|
}],
|
|
[['--preset'],
|
|
{'default': None,
|
|
'help': 'Load a saved selection.',
|
|
}],
|
|
[['--list-presets'],
|
|
{'action': 'store_const',
|
|
'const': 'list_presets',
|
|
'dest': 'mod_presets',
|
|
'default': None,
|
|
'help': 'List available preset selections.',
|
|
}],
|
|
[['--edit-presets'],
|
|
{'action': 'store_const',
|
|
'const': 'edit_presets',
|
|
'dest': 'mod_presets',
|
|
'default': None,
|
|
'help': 'Edit the preset file.',
|
|
}],
|
|
],
|
|
'task': [
|
|
[['--full'],
|
|
{'action': 'store_true',
|
|
'default': False,
|
|
'help': "Use the full set of tasks as input to fzf (instead of "
|
|
"target tasks).",
|
|
}],
|
|
[['-p', '--parameters'],
|
|
{'default': None,
|
|
'help': "Use the given parameters.yml to generate tasks, "
|
|
"defaults to latest parameters.yml from mozilla-central",
|
|
}],
|
|
],
|
|
}
|
|
|
|
|
|
class BaseTryParser(ArgumentParser):
|
|
name = 'try'
|
|
common_groups = ['push', 'preset']
|
|
arguments = []
|
|
templates = []
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
ArgumentParser.__init__(self, *args, **kwargs)
|
|
|
|
group = self.add_argument_group("{} arguments".format(self.name))
|
|
for cli, kwargs in self.arguments:
|
|
group.add_argument(*cli, **kwargs)
|
|
|
|
for name in self.common_groups:
|
|
group = self.add_argument_group("{} arguments".format(name))
|
|
arguments = COMMON_ARGUMENT_GROUPS[name]
|
|
for cli, kwargs in arguments:
|
|
group.add_argument(*cli, **kwargs)
|
|
|
|
group = self.add_argument_group("template arguments")
|
|
self.templates = {t: all_templates[t]() for t in self.templates}
|
|
for template in self.templates.values():
|
|
template.add_arguments(group)
|
|
|
|
def validate(self, args):
|
|
if hasattr(args, 'message'):
|
|
if args.message == 'editor':
|
|
if 'EDITOR' not in os.environ:
|
|
self.error("must set the $EDITOR environment variable to use blank --message")
|
|
|
|
with tempfile.NamedTemporaryFile(mode='r') as fh:
|
|
subprocess.call([os.environ['EDITOR'], fh.name])
|
|
args.message = fh.read().strip()
|
|
|
|
if '{msg}' not in args.message:
|
|
args.message = '{}\n\n{}'.format(args.message, '{msg}')
|
|
|
|
def parse_known_args(self, *args, **kwargs):
|
|
args, remainder = ArgumentParser.parse_known_args(self, *args, **kwargs)
|
|
self.validate(args)
|
|
|
|
if self.templates:
|
|
args.templates = {}
|
|
for cls in self.templates.itervalues():
|
|
context = cls.context(**vars(args))
|
|
if context is not None:
|
|
args.templates.update(context)
|
|
|
|
return args, remainder
|