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'}