Вызов различных функций в 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 

Imagine три других функций в подобном формате, так что есть одна функции, которая соответствует каждой единственной метрике. Сейчас здесь находится блок кода в скрипте, который принимает список метрик:

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 

Answer 2

Вы можете создать класс со статическими методами и использовать getattr, чтобы получить правильный метод. Это похоже на то, что mgilson предполагает, но вы, по сути получить создание Dict бесплатно:

class Calculators: 

    @staticmethod 
    def calcChurn(): 
     print("called calcChurn") 

    @staticmethod 
    def calcEngagement(): 
     print("called calcEngagement") 

    @staticmethod 
    def calcPenetration(): 
     print("called calcPenetration") 

stats = ["churn", "engagement", "penetration", "churn", "churn", "engagement", "undefined"] 

def capitalise(str): 
    return str[0].upper() + str[1:] 

for stat in stats: 
    try: 
     getattr(Calculators, "calc" + capitalise(stat))() 
    except AttributeError as e: 
     print("Unknown statistic: " + stat) 

называется calcChurn
называется calcEngagement
называется calcPenetration
называется calcChurn
называется calcChurn
называется calcEngagement
Неизвестный статистика: undefined

Answer 3

Возможно, имеет смысл инкапсулировать требуемые вызовы внутри объекта.

Если у вас есть смысл для объекта быть объектом, и особенно если многие клиенты называют один и тот же набор функций для получения показателей, то вы можете создать набор подклассов Client, которые вызывают предопределенный набор функций для получения метрики.

Это немного тяжелее, чем отображение dict.

''' Stand alone functions for sql commands. 
    These definitions however dont really do anything. 
''' 


def calc_churn(clientId): 
    return 'result for calc_churn' 


def calc_engagement(clientId): 
    return 'result for calc_engagement' 


''' Client base object ''' 


class Client(object): 
    ''' Base object allows list of functions 
    to be stored in client subclasses''' 

    def __init__(self, id): 
     self.id = id 
     self.metrics = [] 
     self.args = [] 

    def add_metrics(self, metrics, *args): 
     self.metrics.extend(metrics) 
     self.args = args 

    def execute_metrics(self): 
     return {m.__name__: m(*self.args) for m in self.metrics} 


''' Specific sub classes ''' 


class Client1(Client): 

    def __init__(self, id): 
     ''' define which methods are called for this class''' 

     super(Client1, self).__init__(id) 
     self.add_metrics([calc_churn], id) 


class Client2(Client): 

    def __init__(self, id): 
     ''' define which methods are called for this class''' 

     super(Client2, self).__init__(id) 
     self.add_metrics([calc_churn, calc_engagement], id) 


''' create client objects and ''' 

c1 = Client1(1) 
c2 = Client2(2) 

for client in [c1, c2]: 
    print client.execute_metrics() 

В результате вы получите от execute_metrics является dict отображение имени функции его результатов для этого клиента.