基于列表中的值调用Python中的不同函数

Question

我有一个脚本,它将一个度量列表作为输入,然后从数据库中提取这些度量以使用它们执行各种操作.

我的问题是不同的客户端获得指标的不同子集,但我不希望每次添加新客户端时都编写新的IF块.所以现在,我有一个大的IF块,根据相应的度量是否在列表中调用不同的函数.什么是最优雅或Pythonic处理这个的方式？

设置和功能定义:

clientOne = ['churn','penetration','bounce']
clientTwo = ['engagement','bounce']

def calcChurn(clientId):
    churn = cursor.execute(sql to get churn)
    [...]
    return churn

def calcEngagement(clientId):
    engagement = cursor.execute(sql to get engagement)
    [...]
    return engagement

想象一下类似格式的其他三个函数,因此有一个函数对应于每个唯一度量.现在,脚本中的代码块采用度量列表:

def scriptName(client, clientId):
    if churn in client:
        churn = calcChurn(clientId)
    if engagement in client:
        engagement = calcEngagement(clientId)
    if penetration in client:
    [...]

Answer 1

通常,您要创建名称到函数的映射,并使用它来计算您想要的东西:

client_action_map = {
  'churn': calcChurn,
  'engagement': calcEngagement,
  ...
}

def scriptName(actions, clientId):
    results = {}
    for action in actions:
        results[action] = client_action_map[action](clientId)
    return results