5

简短但完整的总结

我想允许我的函数（类工厂）的用户在使用我的函数时注入/覆盖全局导入（下面对基本原理的详细说明）。但是可以传入大约 10 个不同的变量，并且它会在代码中添加许多非常重复的行。（当然，也使调用变得更加复杂：P）现在，我正在做类似以下的事情（只是简化所有这些）。为了使其可运行，我使用了一个虚拟类，但在实际脚本中我将使用import pkg1等。认为这比类工厂等更清晰、更短。

class Dummy(object): pass

pkg1, pkg2 = Dummy(), Dummy()
pkg1.average = lambda *args : sum(args) / len(args)
pkg2.get_lengths = lambda *args : map(len, args)


def get_average(*args, **kwargs):
    average = kwargs.get("average") or pkg1.average
    get_lengths = kwargs.get("get_lengths") or pkg2.get_lengths
    return average(*get_lengths(*args))

adjusted_length = lambda *args: map(len, args) + [15]
print get_average([1,2], [10, 4, 5, 6]) == 3 # True
print get_average([1,2], [10, 4, 5, 6], get_lengths=adjusted_length) == 7 # True

相关的 SO 问题

这个堆栈溢出帖子：Modifying locals in Python，似乎特别相关，最初我想通过存储到 locals 字典来覆盖本地，但是（1）它似乎不起作用，并且（2）它看起来很糟糕主意。所以，我想知道是否有另一种方法可以做到这一点。

这个看起来很有希望（在 python 中将对象添加到另一个模块的全局变量），但我不确定如何以与模块相同的方式访问当前文件的全局变量。（和这个问题 - python: mutating `globals` to dynamic put things in scope - 并没有真正适用，因为我（最终）使用它来定义类）。

我想我可以将所有内容包装在一个 exec 语句中（就像这篇文章 - python exec() 中的全局变量和局部变量），但这既繁琐又意味着进行错误检查/linting/等要困难得多。

这就是我想做的。（注意：我会使用from pkg1 import averageANDfrom pkg2 import get_lengths 但我希望示例更清晰（需要复制上面的 pkg1 和 pkg2 来运行它））

average = pkg1.average
get_lengths = pkg2.get_lengths

def get_average(*args, **kwargs):
    localvars = locals()
    for k in ("get_lengths", "average"):
        if kwargs.get(k, None) and kwargs[k] is not None:
            localvars[k] = kwargs[k]
    return average(*get_lengths(*args))

print get_average([1,2], [10, 4, 5, 6]) == 3 #True
print get_average([1,2], [10, 4, 5, 6], get_lengths=adjusted_length) == 7 # False, is 3

我的特定用例的基本原理

现在，我正在尝试编写一个动态生成的类工厂（用作 SQLAlchemy mixin），但我想允许我的类的用户传入备用构造函数，以便他们可以使用 SQLAlchemy 适配器等。

例如，Flask-SQLAlchemy 提供与 SQLAlchemy 相同的接口，但提供了一个自定义对象/类 ( db)，它包含所有 SQLAlchemy 对象以提供更多功能。

4

1 回答 1

2

您可以使用具有默认值的参数来传递函数。这实际上是您正在做的事情，但更简洁。我将lists其用作单个参数，而不是*args因为当您有其他参数时更容易处理。您必须将列表括在一个元组中才能将它们传递给get_average.

内置函数sorted是这样工作的，所以 Python 程序员应该很容易理解。

get_average(lists, average=pkg1.average, get_lengths=pkg2.get_lengths):
    return average(*get_lengths(*lists))

print get_average(([1,2], [10, 4, 5, 6]))
print get_average(([1,2], [10, 4, 5, 6]), get_lengths=adjusted_length)

如果您有许多关键字参数，您可以将它们打包在一个对象中：

class GetAverageContext(object):
    def __init__(self, average=pkg1.average, get_lengths=pkg2.get_lengths):
        self.average = average
        self.get_lengths = get_lengths

DefaultGetAverageContext = GetAverageContext()

def get_average(lists, context=DefaultGetAverageContext):
    return context.average(*context.get_lengths(*lists))
于 2012-07-30T01:48:36.327 回答