拆开一个类

fin*_*ian 3 python argument-unpacking

我想创建一个类来像字典一样解压它的对象。

例如,使用字典,您可以这样做

foo = {
  "a" : 1
  "b" : 2
}

def bar(a,b):
   return a + b

bar(**foo)
Run Code Online (Sandbox Code Playgroud)

输出3

我希望能够做到这一点

class FooClass:
    def __init__(self):
        self.a = a
        self.b = b

f = FooClass()
bar(**f)
Run Code Online (Sandbox Code Playgroud)

并让它输出3

是我能找到的最相关的问题,但它没有解决这个问题,所以我认为这可能是不可能的。


目前我的解决方案是这样的:

class FooClass:
    def __init__(self):
        self.a = a
        self.b = b

    def to_dict(self):
        return {
          "a" : self.a,
          "b" : self.b
        }
Run Code Online (Sandbox Code Playgroud)
f = FooClass()
bar(**f.to_dict())
Run Code Online (Sandbox Code Playgroud)

Sil*_*olo 8

正如评论中指出的,编写collections.abc.Mapping抽象类的一致子类是正确的方法。要(具体地)子类化此类,您需要实现__getitem____len__和 ,__iter__使其像字典一样表现一致。这意味着__getitem__需要一个字符串,__iter__返回一个可迭代的字符串,等等。

举一个简单的例子,我们将简单地将所有这些委托给self.__dict__,但在实际代码中,您可能想要做一些更精致的事情。

from collections.abc import Mapping

class FooClass(Mapping):

    def __init__(self, a, b):
        self.a = a
        self.b = b

    def __getitem__(self, x):
        return self.__dict__[x]

    def __iter__(self):
        return iter(self.__dict__)

    def __len__(self):
        return len(self.__dict__)

def bar(a, b):
    return a + b

foo = FooClass(40, 2)
print(bar(**foo))
Run Code Online (Sandbox Code Playgroud)

  • 实际上,在这种情况下只需要`__iter__` (3认同)