python - 如何干净地将常量与函数关联起来？

Question

我有一系列函数，应用于数据集中的每条记录，以生成存储在字典中的新字段(记录(“文档”)使用 MongoDB 存储)。我将它们全部分解，因为它们基本上不相关，然后通过将它们作为列表传递给一个函数来将它们重新连接在一起，该函数迭代每个记录的每个操作并添加结果。

令我恼火的是我的处理方式似乎相当不优雅；除其他外，半重复的名称。

def _midline_length(blob):
    '''Generate a midline sequence for *blob*'''
    return 42
midline_length = {
        'func':  _midline_length,
        'key': 'calc_seq_midlen'} #: Midline sequence key/function pair.

很多这样的...

do_calcs = [midline_length, ] # all the functions ...

然后调用如下:

for record in mongo_collection.find():
    for calc in do_calcs:
        record[calc['key']] = calc['func'](record) # add new data to record
    # update record in DB

像这样拆分键可以更轻松地删除数据库中的所有计算字段(设置完所有内容后毫无意义，但在开发代码和方法时很方便)。

我想过也许可以使用类，但它看起来更像是一个ab使用:

class midline_length(object):
    key = 'calc_seq_midlen'
    @staticmethod
    def __call__(blob):
        return 42

然后我可以创建一个实例列表 (do_calcs = [midline_length(), ...]) 并运行该列表，调用每个事物或提取它的key成员。或者，我似乎可以任意向函数添加成员，def myfunc(): 然后myfunc.key = 'mykey'...这看起来更糟糕。更好的想法？

Answer 1

您可能想使用装饰器来实现此目的。

import collections
RecordFunc = collections.namedtuple('RecordFunc', 'key func')

def record(key):
    def wrapped(func):
        return RecordFunc(key, func)
    return wrapped

@record('midline_length')
def midline_length(blob):
    return 42

现在，midline_length 实际上并不是一个函数，而是一个 RecordFunc 对象。

>>> midline_length
RecordFunc(key='midline_length', func=<function midline_length at 0x24b92f8>)

它有一个 func 属性(原始函数)和一个 key 属性。

如果它们被添加到同一个字典中，您可以在装饰器中执行此操作:

RECORD_PARSERS = {}

def record(key):
    def wrapped(func):
        RECORD_PARSERS[key] = func
        return func
    return wrapped

@record('midline_length')
def midline_length(blob):
    return 42