class 中的单独函数中的 argparse 并从 init 调用 args

Question

我有一个关于将参数传递给 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 代码放在 class 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 来使参数起作用，但在这种情况下，我调用了 self.id = self.parse_args.id?

parse_args 函数 return 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() 时都运行是吗？您只能设置一组值。

我认为在创建 class.

之后，将 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'}

class 中的单独函数中的 argparse 并从 init 调用 args

argparse in separate function inside a class and calling args from init

python

class

function

argparse