actionHandler.py

Go to the documentation of this file.

#!/usr/bin/env python
# PYTHON_ARGCOMPLETE_OK
# -*- coding: utf-8 -*-
 
# Copyright 2019-2020 CERN and copyright holders of ALICE O2.
# See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
# All rights not expressly granted are reserved.
#
# This software is distributed under the terms of the GNU General Public
# License v3 (GPL Version 3), copied verbatim in the file "COPYING".
#
# In applying this license CERN does not waive the privileges and immunities
# granted to it by virtue of its status as an Intergovernmental Organization
# or submit itself to any jurisdiction.
 
import argparse
 
 
class NoAction(argparse.Action):
    
    """
    NoAction class adds dummy positional arguments to an argument,
    so sub helper messages can be created
 
    Args:
        argparse (Class): Input as args
    """
    
    def __init__(self, **kwargs):
        kwargs.setdefault("default", argparse.SUPPRESS)
        kwargs.setdefault("nargs", 0)
        super(NoAction, self).__init__(**kwargs)
    
    def __call__(self, parser, namespace, values, option_string = None):
        pass
 
 
class ChoicesAction(argparse._StoreAction):
    
    """
    ChoicesAction class is used to add extra choices
    to a parseargs choices list
 
    Args:
        argparse (Class): Input as args
    """
    
    def add_choice(self, choice, help = ""):
        if self.choices is None:
            self.choices = []
        self.choices.append(choice)
        self.container.add_argument(choice, help = help, action = "none")