mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2025-07-16 02:02:53 -07:00
8d43b8ea39
Also updates: - importlib-metadata-0.7 - jaraco-windows - jaraco.classes-1.5 - jaraco.collections-1.6.0 - jaraco.functools-1.20 - jaraco.structures-1.1.2 - jaraco.text-1.10.1 - jaraco.ui-1.6 - more-itertools-4.3.0 - path.py-11.5.0 - six-1.12.0
77 lines
1.9 KiB
Python
77 lines
1.9 KiB
Python
import argparse
|
|
|
|
import six
|
|
from jaraco.classes import meta
|
|
from jaraco import text
|
|
|
|
|
|
@six.add_metaclass(meta.LeafClassesMeta)
|
|
class Command(object):
|
|
"""
|
|
A general-purpose base class for creating commands for a command-line
|
|
program using argparse. Each subclass of Command represents a separate
|
|
sub-command of a program.
|
|
|
|
For example, one might use Command subclasses to implement the Mercurial
|
|
command set::
|
|
|
|
class Commit(Command):
|
|
@staticmethod
|
|
def add_arguments(cls, parser):
|
|
parser.add_argument('-m', '--message')
|
|
|
|
@classmethod
|
|
def run(cls, args):
|
|
"Run the 'commit' command with args (parsed)"
|
|
|
|
class Merge(Command): pass
|
|
class Pull(Command): pass
|
|
...
|
|
|
|
Then one could create an entry point for Mercurial like so::
|
|
|
|
def hg_command():
|
|
Command.invoke()
|
|
"""
|
|
|
|
@classmethod
|
|
def add_subparsers(cls, parser):
|
|
subparsers = parser.add_subparsers()
|
|
[cmd_class.add_parser(subparsers) for cmd_class in cls._leaf_classes]
|
|
|
|
@classmethod
|
|
def add_parser(cls, subparsers):
|
|
cmd_string = text.words(cls.__name__).lowered().dash_separated()
|
|
parser = subparsers.add_parser(cmd_string)
|
|
parser.set_defaults(action=cls)
|
|
cls.add_arguments(parser)
|
|
return parser
|
|
|
|
@classmethod
|
|
def add_arguments(cls, parser):
|
|
pass
|
|
|
|
@classmethod
|
|
def invoke(cls):
|
|
"""
|
|
Invoke the command using ArgumentParser
|
|
"""
|
|
parser = argparse.ArgumentParser()
|
|
cls.add_subparsers(parser)
|
|
args = parser.parse_args()
|
|
args.action.run(args)
|
|
|
|
|
|
class Extend(argparse.Action):
|
|
"""
|
|
Argparse action to take an nargs=* argument
|
|
and add any values to the existing value.
|
|
|
|
>>> parser = argparse.ArgumentParser()
|
|
>>> _ = parser.add_argument('--foo', nargs='*', default=[], action=Extend)
|
|
>>> args = parser.parse_args(['--foo', 'a=1', '--foo', 'b=2', 'c=3'])
|
|
>>> args.foo
|
|
['a=1', 'b=2', 'c=3']
|
|
"""
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
getattr(namespace, self.dest).extend(values)
|