使用函数属性来存储延迟（潜在）处理的结果

Question

我正在做一些碰撞检测，我非常想在两个不同的上下文中使用相同的函数。在一种情况下，我希望它是这样的

def detect_collisions(item, others):
    return any(collides(item, other) for other in others)

，而在另一种情况下，我希望它是

def get_collisions(item, others):
    return [other for other in others if collides(item, other)]

我真的很讨厌在这里编写两个函数的想法。仅仅保持他们的名字简单是一个问题，而使碰撞检测的接口复杂化则是另一个问题。所以我在想：

def peek(gen):
    try:
        first = next(gen)
    except StopIteration:
        return False
    else:
        return it.chain((first,), gen)

def get_collisions(item, others):
    get_collisions.all = peek(other for other in others if collides(item, other))
    return get_collisions.all

现在，当我只想进行检查时，我可以说：

if get_collisions(item, others):
    # aw snap

或者

if not get_collisions(item, others):
    # w00t

在我真正想检查它们的其他上下文中，我可以这样做：

if get_collisions(item, others):
    for collision in get_collisions.all:
        # fix it

并且在这两种情况下，我不再进行任何处理比我需要的。

我认识到这比前两个函数的代码更多，但它还有以下优点：

1. 将碰撞检测接口保留为树，节点位于顶层而不是中层。这看起来更简单。

2. 使用方便的查看功能来连接自己。如果我再使用它一次，那么我实际上编写的代码就会更少。 （回应 YAGNI，如果我有的话，我会的）

所以。如果你是众所周知的杀人狂，知道我住在哪里，那么如果我写了上面的代码，我会期待你来访吗？如果是这样，您会如何处理这种情况？

Answer 1

只需让 get_collisions 返回一个生成器：

def get_collisions(item, others):
    return (other for other in others if collides(item, other))

然后，如果您想进行检查：

for collision in get_collisions(item, others):
    print 'Collision!'
    break
else:
    print 'No collisions!'

Answer 2

这与我们在

我从来没有真正关心过您接受多个关键字并具有不同命名函数的建议，具体取决于您是否希望通过 any() 或 all() 处理所有结果 - 但是喜欢使用关键字参数的想法，它允许单个函数同时在两个或多个位置使用。这就是我最终得到的结果：

# can be called with a single unnamed value or a single named value
def cache(*args, **kwargs):
    if len(args)+len(kwargs) == 1:
        if args:
            name, value = 'value', args[0]  # default attr name 'value'
        else:
            name, value = kwargs.items()[0]
    else:
        raise NotImplementedError('"cache" calls require either a single value argument '
                                  'or a name=value argument identifying an attribute.')
    setattr(cache, name, value)
    return value

# add a sub-function to clear the cache attributes (optional and a little weird)
cache.clear = lambda: cache.func_dict.clear()

# you could then use it either of these two ways
if get_collisions(item, others):
    # no cached value

if cache(collisions=get_collisions(item, others)):
    for collision in cache.collisions:
        # fix them

通过将所有丑陋的细节放在一个单独的函数中，它不会以某种方式影响 get_collisions() 中的代码，并且也可以在其他地方使用。