ok
Mini Shell
# -*- coding: utf-8 -*-
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
from __future__ import absolute_import
import argparse
from typing import Tuple, List, Dict, Any
class CustomFormatter(argparse.ArgumentDefaultsHelpFormatter, argparse.HelpFormatter):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs, width=200, max_help_position=30)
class ArgumentParser:
"""Helper to create parsers."""
def __init__(self, *args, **kwargs):
self._parser = argparse.ArgumentParser(*args, **kwargs)
self._subparser = self._parser.add_subparsers(dest='command', required=True, metavar='{command}')
self._subparsers = {}
# to allow global arguments addition
self.add_argument = self._parser.add_argument
def command(self, allow_abbrev=False, **kwargs):
def decorator(method):
self._subparsers[method.__name__] = self._subparser.add_parser(
method.__name__.replace("_", "-"),
allow_abbrev=allow_abbrev,
**kwargs
)
return method
return decorator
def argument(self, *args, **kwargs):
def decorator(method):
self._subparsers[method.__name__].add_argument(*args, **kwargs)
return method
return decorator
def mutual_exclusive_group(
self,
arguments_params: List[Tuple[List[Any], Dict[str, Any]]],
**kwargs,
):
"""
Create mutually exclusive group with arguments
`argument_params` is a list of tuples with *args and **kwargs to create arguments in the group
"""
def decorator(method):
group = self._subparsers[method.__name__].add_mutually_exclusive_group(**kwargs)
for argument in arguments_params:
group.add_argument(*argument[0], **argument[1])
return method
return decorator
def parse_args(self, args):
return self._parser.parse_args(args)
Zerion Mini Shell 1.0