python - argparse 在类内的单独函数中并从 init 调用 args

Question

我有一个关于将 args 传递给 init 中的变量的问题

这是我的代码的工作版本。

class A:
  def __init__(self):
      self.id = args.id
      self.pw = args.pw
      self.endpoint = args.endpoint

  def B:
      ..do something..

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', '--id', type=str, help = 'username')
    parser.add_argument('-p', '--pw', type=str, help ='password')
    parser.add_argument('-e', '--end_point', type=str , help='end point of api')
    args = parser.parse_args()

上面的代码有效，但我现在正在尝试的是将所有 argparse 代码放入类 A 的函数中，并将 arg 分配给 init。我查看了网络，但找不到好的解决方案。

class A:
  def __init__(self):
      self.id = self.parse_args(id)
      self.pw = self.parse_args(pw)
      self.endpoint = self.parse_args(endpoint)

  def B:
      ..do something..

  def parse_args(self,args):
      parser = argparse.ArgumentParser()
      parser.add_argument('-i', '--id', type=str, help = 'username')
      parser.add_argument('-p', '--pw', type=str, help ='password')
      parser.add_argument('-e', '--end_point', type=str , help='end point of api')
      return parser.parse_args(args)

好吧，确切地说，我不确定我应该如何解决这个问题。

在上面的例子中，我只是调用了 args.variable 来使 args 起作用，但在这种情况下我调用了 self.id = self.parse_args.id?

parse_args 函数返回 args，我也试过 self.id = self.parse_args(id) 这给了我

  TypeError: 'builtin_function_or_method' object is not iterable

我想将 args 分离到一个单独的函数中的部分原因是为了使用 argparse 简化我的单元测试。

Answer 1

所以在第一种情况下，你一定是在做

if __name__ ...
....
args = parser.parse_args()
a = A()

A.__init__ 可以看到 args 因为它是全局的。

我不明白您为什么要将 argparse 代码作为 A 的一部分；您不希望每次使用 A() 时它都运行，对吗？您只能创建一组值。

我认为让 parse_args 代码成为一个方法，可以在创建类后随意运行。

我认为这是一种非常灵活的方法:

import argparse

class A:
  def __init__(self, id=None, pw=None, endpoint=None):
      self.id = id
      self.pw = pw
      self.endpoint = endpoint

  def parse_args(self, argv=None):
      parser = argparse.ArgumentParser()
      parser.add_argument('-i', '--id', type=str, help = 'username')
      parser.add_argument('-p', '--pw', type=str, help ='password')
      parser.add_argument('-e', '--end_point', type=str , help='end point of api')
      args = parser.parse_args(argv)
      self.id = args.id
      self.pw = args.pw
      self.endpoint = args.end_point

  def __str__(self):
      return 'A(%s,...)'%self.id

if __name__ == "__main__":
    a = A()
    print(a)
    a.parse_args()
    print(a)
    b = A(id='you')
    print(b)
    b.parse_args(['--id','me'])
    print(b)

可以在对象创建期间从命令行或自定义 argv 设置值

1610:~/mypy$ python stack39967787.py --id joe
A(None,...)
A(joe,...)
A(you,...)
A(me,...)

==================

我的第一种方法(暂时删除)

class A():
    def __init__(self):
        args = self.parse_args()
        self.a = args.a
        etc
    @static_method
    def parse_args(self):
        parser = ....
        return parser.parse_args()

=====================

你的 class A 可以用作 Namespace，让 parse_args 直接更新属性(它使用 setattr .

import argparse
class A:
  def __init__(self, id=None, pw=None, endpoint=None):
      self.id = id
      self.pw = pw
      self.endpoint = endpoint

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', '--id', type=str, help = 'username')
    parser.add_argument('-p', '--pw', type=str, help ='password')
    parser.add_argument('-e', '--endpoint', type=str , help='end point of api')
    args = parser.parse_args()
    print(args)

    a = A()
    parser.parse_args(namespace=a)
    print(vars(a))

制作:

1719:~/mypy$ python stack39967787_1.py --id joe --pw=xxxx -e1
Namespace(endpoint='1', id='joe', pw='xxxx')
{'endpoint': '1', 'id': 'joe', 'pw': 'xxxx'}