Python相当于类/构造函数的functools的"部分"



我想创建一个行为类似collections.defaultdict的类,而不需要使用代码指定工厂。例如:而不是

class Config(collections.defaultdict):
    pass

:

Config = functools.partial(collections.defaultdict, list)

这几乎可以工作,但是

isinstance(Config(), Config)

失败。我敢打赌,这条线索意味着,在更深层次上,还存在着更狡猾的问题。那么,有没有一种方法可以真正做到这一点呢?

我也试过:

class Config(Object):
    __init__ = functools.partial(collections.defaultdict, list)

我不认为有一个标准的方法来做到这一点,但如果你经常需要它,你可以把你自己的小函数:

import functools
import collections

def partialclass(cls, *args, **kwds):
    class NewCls(cls):
        __init__ = functools.partialmethod(cls.__init__, *args, **kwds)
    return NewCls

if __name__ == '__main__':
    Config = partialclass(collections.defaultdict, list)
    assert isinstance(Config(), Config)

至少在Python 3.8.5中,它只适用于functools.partial:

import functools
class Test:
    def __init__(self, foo):
        self.foo = foo
    
PartialClass = functools.partial(Test, 1)
instance = PartialClass()
instance.foo

我也遇到过类似的问题,但我也要求部分应用的类的实例能够pickle。我想我应该分享一下我最后的结果。

我通过窥视Python自己的collections.namedtuple来改编fjarri的答案。下面的函数创建了一个可以pickle的命名子类。

from functools import partialmethod
import sys
def partialclass(name, cls, *args, **kwds):
    new_cls = type(name, (cls,), {
        '__init__': partialmethod(cls.__init__, *args, **kwds)
    })
    # The following is copied nearly ad verbatim from `namedtuple's` source.
    """
    # For pickling to work, the __module__ variable needs to be set to the frame
    # where the named tuple is created.  Bypass this step in enviroments where
    # sys._getframe is not defined (Jython for example) or sys._getframe is not
    # defined for arguments greater than 0 (IronPython).
    """
    try:
        new_cls.__module__ = sys._getframe(1).f_globals.get('__name__', '__main__')
    except (AttributeError, ValueError):
        pass
    return new_cls

如果您确实需要通过isinstance进行显式类型检查,您可以简单地创建一个不太平凡的子类:

class Config(collections.defaultdict):
    def __init__(self): # no arguments here
        # call the defaultdict init with the list factory
        super(Config, self).__init__(list)

列表工厂和

将有无参数构造
isinstance(Config(), Config)

可以使用*args**kwargs:

class Foo:
    def __init__(self, a, b):
        self.a = a
        self.b = b
    def printy(self):
        print("a:", self.a, ", b:", self.b)
class Bar(Foo):
    def __init__(self, *args, **kwargs):
        return super().__init__(*args, b=123, **kwargs)
if __name__=="__main__":
    bar = Bar(1)
    bar.printy()  # Prints: "a: 1 , b: 123"

相关内容

  • 没有找到相关文章

最新更新