在argparse

Question

更換的說法我有一個基類，我不能修改（代碼可能有其他錯誤，但請忽略這些）

class BaseClass(object): 
    def __init__(self): 
     self.parser = argparse.ArgumentParser() 
     self.parser.add_argument("arg1", choices=("a", "b")) 

我想要的是重寫ARG1如下

class DerivedClass(BaseClass): 
    def __init__(self): 
     BaseClass.__init__(self) 
     self.parser.add_argument("arg1", choices=("a", "c", "d")) 

Answer 1

如果您不能修改基類的實現，你可以達到進入parser的屬性和修改您所關心的具體行動。下面是代碼演示只是一個草圖：

from base import BaseClass 


def _modify_choices(parser, dest, choices): 
    for action in parser._actions: 
     if action.dest == dest: 
      action.choices = choices 
      return 
    else: 
     raise AssertionError('argument {} not found'.format(dest)) 


class MySubClass(BaseClass): 
    def __init__(self): 
     super(MySubClass, self).__init__() 
     _modify_choices(self.parser, 'arg1', ('a', 'b', 'c')) 


def main(): 
    inst = MySubClass() 
    inst.parser.parse_args() 


if __name__ == '__main__': 
    exit(main()) 

與用法：

$ python subclass.py d 
usage: subclass.py [-h] {a,b,c} 
subclass.py: error: argument arg1: invalid choice: 'd' (choose from 'a', 'b', 'c') 

注意，這伸進私人實現細節（特別是ArgumentParser._actions），但以其他方式使用公共接口。

Answer 2

add_argument創建Action對象，該對象編碼add_argument調用中給出的參數。這些可以在創建後讀取和/或修改。

class BaseClass(object) 
    def __init__(self): 
     self.parser = argparse.ArgumentParser() 
     self.arg = self.parser.add_argument("arg1", choices=("a", "b")) 
     # self.arg is now the Action object defined by this add_argument method 

class DerivedClass(BaseClass): 
    def __init__(self): 
     BaseClass.__init__(self) 
     # modify the choices attribute of self.arg 
     self.arg.choices = ("a","b","c") 

的行動目標，也可在parser._actions列表中找到，但我更喜歡保存自己的代碼的參考。

（此代碼沒有經過測試，所以可能有錯誤或兩個）。

---

class BaseClass(object): 
    def __init__(self): 
     self.parser = argparse.ArgumentParser(prog="BASE") 
     self.parser.add_argument("arg1", choices=("a", "b")) 
     self.parser.add_argument('-f','--foo') 

class DerivedClass(BaseClass): 
    def __init__(self): 
     BaseClass.__init__(self) 
     self.parser.prog = "DERIVED" 
     print(self.parser._actions) 

顯示了當DerivedClass創建這個列表：

[_HelpAction(option_strings=['-h', '--help'], dest='help', nargs=0, const=None, default='==SUPPRESS==', type=None, choices=None, help='show this help message and exit', metavar=None), 
_StoreAction(option_strings=[], dest='arg1', nargs=None, const=None, default=None, type=None, choices=('a', 'b'), help=None, metavar=None), 
_StoreAction(option_strings=['-f', '--foo'], dest='foo', nargs=None, const=None, default=None, type=None, choices=None, help=None, metavar=None)] 

因此增加：

idx = [a.dest for a in self.parser._actions].index('arg1') 
    self.parser._actions[idx].choices = ("a","b","c") 

p1 = BaseClass() 
p1.parser.print_help() 

p2 = DerivedClass() 
p2.parser.print_help() 

產生2個usag ES：

usage: BASE [-h] [-f FOO] {a,b} 

positional arguments: 
    {a,b} 

optional arguments: 
    -h, --help   show this help message and exit 
    -f FOO, --foo FOO 

和

usage: DERIVED [-h] [-f FOO] {a,b,c} 

positional arguments: 
    {a,b,c} 

optional arguments: 
    -h, --help   show this help message and exit 
    -f FOO, --foo FOO