如何在Python中伪造/代理一个类

时间:2012-03-30 12:01:41

标签: python

我写了一些包装器,它将另一个对象作为属性。此包装器将__getattr____setattr__的所有属性请求代理(转发)到作为属性存储的对象。我还需要为我的代理提供什么,以便在通常情况下包装器看起来像包装类?

我想我需要修复继承这样的事情,也许是__repr__,...... 我还需要处理什么以及如何修复继承以使instanceof()有效?

编辑:我试图制作一个功能代理,但由于我不完全理解配方,它失败了:(

setattr_=object.__setattr__
getattr_=object.__getattribute__

class Proxy(object):
    __slots__=["_func", "_params", "_kwargs", "_obj", "_loaded", "__weakref__"]
    def __init__(self, func, *params, **kwargs):
        setattr_(self, "_func", func)
        setattr_(self, "_params", params)
        setattr_(self, "_kwargs", kwargs)

        setattr_(self, "_obj", None)
        setattr_(self, "_loaded", False)

    def _get_obj(self):
        if getattr_(self, "_loaded")==False:
            print("Loading")
            setattr_(self, "_obj", getattr_(self, "_func")(*getattr_(self, "_params"), **getattr_(self, "_kwargs")))
            setattr_(self, "_loaded", True)

        return getattr_(self, "_obj")
    #
    # proxying (special cases)
    #
    def __getattribute__(self, name):
        return getattr(getattr_(self, "_get_obj")(), name)
    def __delattr__(self, name):
        delattr(getattr_(self, "_get_obj")(), name)
    def __setattr__(self, name, value):
        setattr(getattr_(self, "_get_obj")(), name, value)

    def __nonzero__(self):
        return bool(getattr_(self, "_get_obj")())
    def __str__(self):
        return str(getattr_(self, "_get_obj")())
    def __repr__(self):
        return repr(getattr_(self, "_get_obj")())

    #
    # factories
    #
    _special_names=[
        '__abs__', '__add__', '__and__', '__call__', '__cmp__', '__coerce__',
        '__contains__', '__delitem__', '__delslice__', '__div__', '__divmod__',
        '__eq__', '__float__', '__floordiv__', '__ge__', '__getitem__',
        '__getslice__', '__gt__', '__hash__', '__hex__', '__iadd__', '__iand__',
        '__idiv__', '__idivmod__', '__ifloordiv__', '__ilshift__', '__imod__',
        '__imul__', '__int__', '__invert__', '__ior__', '__ipow__', '__irshift__',
        '__isub__', '__iter__', '__itruediv__', '__ixor__', '__le__', '__len__',
        '__long__', '__lshift__', '__lt__', '__mod__', '__mul__', '__ne__',
        '__neg__', '__oct__', '__or__', '__pos__', '__pow__', '__radd__',
        '__rand__', '__rdiv__', '__rdivmod__', '__reduce__', '__reduce_ex__',
        '__repr__', '__reversed__', '__rfloorfiv__', '__rlshift__', '__rmod__',
        '__rmul__', '__ror__', '__rpow__', '__rrshift__', '__rshift__', '__rsub__',
        '__rtruediv__', '__rxor__', '__setitem__', '__setslice__', '__sub__',
        '__truediv__', '__xor__', 'next',
    ]

    @classmethod
    def _create_class_proxy(cls, theclass):
        """creates a proxy for the given class"""

        def make_method(name):
            def method(self, *args, **kw):
                return getattr(getattr_(self, "_get_obj")(), name)(*args, **kw)
            return method

        namespace={}
        for name in cls._special_names:
            if hasattr(theclass, name):
                namespace[name]=make_method(name)
        return type("%s(%s)"%(cls.__name__, theclass.__name__), (cls,), namespace)

    def __new__(cls, obj, *args, **kwargs):
        """
        creates an proxy instance referencing `obj`. (obj, *args, **kwargs) are
        passed to this class' __init__, so deriving classes can define an 
        __init__ method of their own.
        note: _class_proxy_cache is unique per deriving class (each deriving
        class must hold its own cache)
        """
        try:
            cache=cls.__dict__["_class_proxy_cache"]
        except KeyError:
            cls._class_proxy_cache=cache={}
        try:
            theclass=cache[obj.__class__]
        except KeyError:
            cache[obj.__class__]=theclass=cls._create_class_proxy(obj.__class__)
        ins=object.__new__(theclass)
        theclass.__init__(ins, obj, *args, **kwargs)
        return ins

if __name__=='__main__':
    def t(x, y):
        print("Running t")
        return x+y

    a=Proxy(t, "a", "b")
    print("Go")
    print(a+"c")

3 个答案:

答案 0 :(得分:24)

这个问题可以很好地解决这个问题:

  

Object Proxying (Python recipe)

您必须遵循的一般想法是,类上的大多数方法都是通过类本身或其元类的__getattr____getattribute__的某种组合来访问的,但这不适用于python特殊方法,以双下划线开头和结尾的方法,对于那些要找到的方法,它们必须是实际类的实际方法,不能进行属性代理。

您必须提供哪些方法显然取决于代理类本身提供哪些方法。您需要为isinstance()提供的特殊方法是__instancecheck__ and __subclasscheck__方法。要使repr()生效,您还必须在代理类本身上定义适当的__repr__()

答案 1 :(得分:7)

通常,您可以使用wrapt库(pypi),这对您来说非常重要:

  

包装模块非常注重正确性。因此,它超越了现有机制,如functools.wraps(),以确保装饰器保持内省,签名,类型检查能力等[...]

     

为了确保开销尽可能小,C扩展模块用于性能关键组件

supports creating custom wrapper classes。为了添加自己的属性,您需要以wrapt不会尝试将它们传递给包装实例的方式声明它们。你可以:

  • 使用_self_添加属性并添加访问属性
  • 除了在__init__
  • 之外,在类级别声明属性
  • 如果适合您的课程(在文档中未提及),请使用插槽,如下所示:

    class ExtendedMesh(ObjectProxy):
        __slots__ = ('foo')
    
        def __init__(self, subject):
            super().__init__(subject)
            self.foo = "bar"
    

它也supports function wrappers,可能适合您的目的。

答案 2 :(得分:0)

如果其他人正在寻找更完整的代理实现

尽管有几种与OP相似的python代理解决方案,但我找不到能同时代理和任意 class对象的解决方案作为 proxy 的自动功能,它们将返回值自变量。这就是我所需要的。

我已经为此目的编写了一些代码,作为完整的代理/日志记录python执行的一部分,将来我可能会将其放入一个单独的库中。如果有人感兴趣,您可以在pull request中找到代码的核心。如果您希望将其作为独立库,请给我下一行。

我的代码将自动返回任何代理对象的属性,函数和类的包装器/代理对象。目的是记录和重播一些代码,因此我具有等效的“重播”代码和一些逻辑,用于将所有代理对象存储到json文件中。