如何在 Python 中创建对象集合代理？

Question

我正在尝试创建一个对象集合代理，它可以执行以下操作：

class A:
    def do_something():
        # ...

class B:
    def get_a():
        return A()

class Proxy:
    ?

collection = [B(), B()]
proxy = Proxy(collection)

proxy.get_a().do_something()
# ^ for each B in collection get_a() and do_something()

实现这一目标的最佳架构/策略是什么？

我猜关键问题是，如何缓存 get_a() 的结果，以便我可以代理 do_something()

注意我不指望proxy.get_a().do_something() 会返回任何明智的东西，它只是应该做的事情。

Answer 1

足够简单...您可能需要对其进行调整以进行更多检查

class A(object):
    def do_something(self):
        print id(self), "called"

class B(object):
    def get_a(self):
        return A()

class Proxy(object):
    def __init__(self, objs):
        self._objs = objs

    def __getattr__(self, name):
        def func(*args, **kwargs):
            return Proxy([getattr(o, name)(*args, **kwargs) for o in self._objs])
        return func

collection = [B(), B()]

proxy = Proxy(collection)
proxy.get_a().do_something()

结果：

4455571152 called
4455571216 called

Answer 2

最pythonic的方式可能是list comprehension：

results = [b.get_a().do_something() for b in collection]

如果你想缓存对B.get_a()的调用，你可以使用memoization。自己进行记忆的一种简单方法如下所示：

cache = None

# ...

class B:
    def get_a(self):
        global cache
        if cache is None:
            cache = A()
        return cache

如果你想在多个地方使用缓存，你需要根据键缓存结果以区分它们，为了方便起见，写一个decorator，你可以简单地包装你想要结果的函数缓存。

Python 算法：掌握 Python 语言中的基本算法（参见 this question）就是一个很好的例子。针对您的情况进行了修改，不使用函数参数而是使用函数名作为缓存键，它看起来像这样：

from functools import wraps

def memoize(func):
    cache = {}
    key = func.__name__
    @ wraps(func)
    def wrap(*args):
        if key not in cache:
            cache[key] = func(*args)
        return cache[key]
    return wrap

class A:
    def do_something(self):
        return 1

class B:
    @memoize
    def get_a(self):
        print "B.get_a() was called"
        return A()


collection = [B(), B()]

results = [b.get_a().do_something() for b in collection]
print results

输出：

B.get_a() was called
[1, 1]