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

python argparse在一个类中的单独函数中，并从init

1 个答案: