python equivalent of functools 'partial' for a class / constructor

前端 未结 3 912
花落未央
花落未央 2020-12-09 15:57

I want to create a class that behaves like collections.defaultdict, without having the usage code specify the factory. EG: instead of

class Config(collectio         


        
3条回答
  •  盖世英雄少女心
    2020-12-09 16:32

    I had a similar problem but also required instances of my partially applied class to be pickle-able. I thought I would share what I ended up with.

    I adapted fjarri's answer by peeking at Python's own collections.namedtuple. The below function creates a named subclass that can be pickled.

    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
    

提交回复
热议问题