python - 将 "global imports"注入(inject) Python 函数

Question

简短但完整的总结

我想允许我的函数(类工厂)的用户在使用我的函数时注入(inject)/覆盖全局导入(下面对基本原理的详细解释)。但是大约有 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 字典来覆盖 locals 但是(1)它似乎不起作用，并且(2)这似乎是个坏主意。所以，我想知道是否有另一种方法来做到这一点。

这个看起来很有前途(Adding an object to another module's globals in python)，但我不太确定如何以与模块相同的方式访问当前文件的全局变量。 (这个问题 - python: mutating `globals` to dynamically put things in scope - 并不真正适用，因为我(最终)使用它来定义类)。

我想我可以将所有内容都包装在一个 exec 语句中(就像这篇文章 - globals and locals in python exec() )，但这既繁琐又意味着更难进行错误检查/检查等。

所以这就是我喜欢做的事情。 (注意:我会使用 from pkg1 import average 和 from 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 混入)，但我想允许我的类的用户传入备用构造函数，以便他们可以使用 SQLAlchemy 适配器等.

例如，Flask-SQLAlchemy 提供了与 SQLAlchemy 相同的接口(interface)，但提供了一个自定义对象/类(db)来包装所有 SQLAlchemy 对象以提供更多特点。

Answer 1

您可以将参数与 default values 一起使用传递函数。这实际上是你正在做的，但更干净。我使用 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))