mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 12:24:32 +00:00
d6735db364
WA2 only supported a single config file but the way WA3's configuration parser works there can be as many and the user needs. They will be prioritied in the order they are specified. e.g in `wa run agenda.yaml -c 1.yaml -c 2.yaml` `1.yaml` will be applied first and `2.yaml` will be applied on top of that.
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
# Copyright 2014-2015 ARM Limited
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
import textwrap
|
|
|
|
from wlauto.core.plugin import Plugin
|
|
from wlauto.utils.doc import format_body
|
|
from wlauto.core.version import get_wa_version
|
|
|
|
|
|
def init_argument_parser(parser):
|
|
parser.add_argument('-c', '--config', help='specify an additional config.py', action='append')
|
|
parser.add_argument('-v', '--verbose', action='count',
|
|
help='The scripts will produce verbose output.')
|
|
parser.add_argument('--version', action='version', version='%(prog)s {}'.format(get_wa_version()))
|
|
return parser
|
|
|
|
|
|
class Command(Plugin):
|
|
"""
|
|
Defines a Workload Automation command. This will be executed from the command line as
|
|
``wa <command> [args ...]``. This defines the name to be used when invoking wa, the
|
|
code that will actually be executed on invocation and the argument parser to be used
|
|
to parse the reset of the command line arguments.
|
|
|
|
"""
|
|
kind = "command"
|
|
help = None
|
|
usage = None
|
|
description = None
|
|
epilog = None
|
|
formatter_class = None
|
|
|
|
def __init__(self, subparsers):
|
|
super(Command, self).__init__()
|
|
self.group = subparsers
|
|
parser_params = dict(help=(self.help or self.description), usage=self.usage,
|
|
description=format_body(textwrap.dedent(self.description), 80),
|
|
epilog=self.epilog)
|
|
if self.formatter_class:
|
|
parser_params['formatter_class'] = self.formatter_class
|
|
self.parser = subparsers.add_parser(self.name, **parser_params)
|
|
init_argument_parser(self.parser) # propagate top-level options
|
|
self.initialize(None)
|
|
|
|
def initialize(self, context):
|
|
"""
|
|
Perform command-specific initialisation (e.g. adding command-specific options to the command's
|
|
parser). ``context`` is always ``None``.
|
|
|
|
"""
|
|
pass
|
|
|
|
def execute(self, args):
|
|
"""
|
|
Execute this command.
|
|
|
|
:args: An ``argparse.Namespace`` containing command line arguments (as returned by
|
|
``argparse.ArgumentParser.parse_args()``. This would usually be the result of
|
|
invoking ``self.parser``.
|
|
|
|
"""
|
|
raise NotImplementedError()
|