Question

我的代码中实现了很多类。现在我意识到，对于为所有这些类调用的每个方法，我需要添加一行：

with service as object:

所以我正在尝试使用Proxy模式自动完成工作，这是我的示例代码

class A(object):
    def __init__(self, name):
        self.name = name
    def hello(self):
        print 'hello %s!' % (self.name)
    def __enter__(self):
        print 'Enter the function'
        return self
    def __exit__(self, exc_type, exc_value, traceback):
        print 'Exit the function'
class Proxy(object):
    def __init__(self, object_a):
#        object.__setattr__(self, '_object_a', object_a)
        self._object_a = object_a

    def __getattribute__(self, name):
        service = object.__getattribute__(self, '_object_a')
        with service as service:
            result = getattr(service, name)
        return result    

if __name__=='__main__':
    a1 = A('A1')
    b = Proxy(a1)
    b.hello()
    a2 = A('A2')
    b = Proxy(a2)
    b.hello()

一切正常，我有输出：

Enter the function A1
Exit the function A1
hello A1!
Enter the function A2
Exit the function A2
hello A2!

但这并不是我所需要的，因为我需要的是相当于：

with a1 as a1:
    a1.hello()

我必须有输出：

Enter the function A1
hello A1!
Exit the function A1
Enter the function A2
hello A2!
Exit the function A2

我需要得到什么结果？感谢

Answer 1

我会使用装饰器：

class Proxy(object):
    def __init__(self, object_a):
        self._object_a = object_a

    def decorateEnterExit(self, obj, f):
        def inner(*args, **kwargs):
            with obj as _:
                return f(*args, **kwargs)
        return inner

    def __getattribute__(self, name):
        obj = object.__getattribute__(self, '_object_a')
        dee = object.__getattribute__(self, 'decorateEnterExit')
        return dee(obj, getattr(obj, name))

这样，只有在执行函数时才会计算with。你这样做的方式是首先评估with（包括退出），然后返回函数进行调用。现在我们返回一个函数，它本身将进入with并在里面调用函数。

>>> Proxy(A('Ax')).hello()
Enter the function
hello Ax!
Exit the function

请注意，您需要使用self __enter__方法返回A：

def __enter__(self):
    print 'Enter the function'
    return self

Answer 2

您在__enter__方法中缺少return语句，该语句应返回您的类的实例（self）。

关于这个问题，在您的Proxy中，可能会在关闭上下文后评估结果。试试看你实际返回的内容 - 是一种绑定方法：

def __getattribute__(self, name):
    service = object.__getattribute__(self, '_object_a')
    with service as service:
        result = getattr(service, name)
        print result
    return result

这将显示

Enter the function
<bound method A.hello of <__main__.A object at 0x022A2950>>
Exit the function

只有在__getattribute__返回后才调用方法。

Python中的代理模式

2 个答案: