ca9eefdf12
Restructure creation of subcommands to pass a full subparser object to the command class to allow more complex parser creation to facilitate adding argument groups and mutually exclusive groups. Change-Id: Ic6f7e3834cf8bd29b87d5d6cafdb8a9f041f4cf6
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
#
|
|
# Copyright (c) 2011 OpenStack LLC.
|
|
# Copyright (c) 2012, 2013, 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 abc
|
|
import argparse
|
|
import os
|
|
|
|
|
|
class AppendReplaceAction(argparse._AppendAction):
|
|
"""Allows setting of a default value which is overriden by the first use
|
|
of the option, while subsequent calls will then append.
|
|
"""
|
|
def __init__(self, *args, **kwargs):
|
|
super(AppendReplaceAction, self).__init__(*args, **kwargs)
|
|
self._reset_default = False
|
|
self.default = list(self.default)
|
|
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
if not self._reset_default:
|
|
setattr(namespace, self.dest, [])
|
|
self._reset_default = True
|
|
super(AppendReplaceAction, self).__call__(parser, namespace, values,
|
|
option_string)
|
|
|
|
|
|
class GitUpstreamCommand(object):
|
|
"""Base command class
|
|
|
|
To create commands simply subclass and implement the necessary abstract
|
|
methods.
|
|
"""
|
|
|
|
__metaclass__ = abc.ABCMeta
|
|
|
|
def __init__(self, parser):
|
|
self.parser = parser
|
|
|
|
def validate(self, args):
|
|
"""Verify the arguments passed for this command"""
|
|
return
|
|
|
|
@abc.abstractmethod
|
|
def run(self, args):
|
|
"""Execute this command"""
|
|
return
|
|
|
|
|
|
def get_subcommands(parser):
|
|
|
|
subparsers = parser.add_subparsers(title="commands", metavar='<command>',
|
|
dest='subcommand')
|
|
|
|
subcommands = _find_actions(subparsers, os.path.dirname(__file__))
|
|
|
|
parser.set_defaults(subcommands=subcommands)
|
|
|
|
return subcommands
|
|
|
|
|
|
# partially taken from python-keystoneclient
|
|
def _find_actions(subparsers, module_path):
|
|
subcommands = {}
|
|
for mod in (p[:-len('.py')] for p in os.listdir(module_path) if
|
|
p.endswith('.py')):
|
|
__import__(__name__ + '.' + mod)
|
|
|
|
for cmd_class in GitUpstreamCommand.__subclasses__():
|
|
command = cmd_class.name
|
|
desc = cmd_class.__doc__ or None
|
|
help = desc.strip().split('\n')[0]
|
|
|
|
subparser = subparsers.add_parser(
|
|
command,
|
|
help=help,
|
|
description=desc)
|
|
subparser.register('action', 'append_replace', AppendReplaceAction)
|
|
subparser.set_defaults(cmd=cmd_class(subparser))
|
|
subcommands[command] = subparser
|
|
|
|
return subcommands
|