将"全局导入"注入Python函数

Question

简短但完整的摘要

我希望允许我的函数(类工厂)的用户在使用我的函数时注入/覆盖全局导入(下面对基本原理的更长解释).但是有大约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问题

这个堆栈溢出帖子:在Python中修改本地化,似乎特别相关,最初我想通过存储到本地字典来覆盖本地,但(1)它似乎不起作用,(2)它似乎是一个坏的理念.所以,我想知道是否有另一种方法可以做到这一点.

这看起来很有希望(在python中将对象添加到另一个模块的全局变量),但我不确定如何以与模块相同的方式访问当前文件的全局变量.(而这个问题 - python:变异`globals`来动态地将事物放在范围内 - 并不真正适用,因为我(最终)使用它来定义类).

我想我可以将所有东西都包装在一个exec语句中(比如这篇文章 - 在python exec()中的全局变量和本地变量),但这既繁琐又意味着它更难以进行错误检查/ linting/etc.

因此,这里就是我想要的是什么喜欢做的事.(注意:我会使用from pkg1 import averageAND,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 mixin),但我希望允许我的类的用户传入替代构造函数,因此他们可以使用SQLAlchemy适配器等.

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

Answer 1

您可以使用带有默认值的参数来传递函数。这实际上就是您正在做的事情，但更干净。我将其用作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))