Carte importé fonction à une variable

Question

J'ai quelques fonctions importées from package import fizz, buzz et je veux les mapper à une variable func à partir d'un argument chaîne.

class Whatever(object): 
    def __init__(self, func_name='fizz'): 
     try: 
      self.func = vars()[func_name] 
     except KeyError: 
      print "Bad func_name" 
      raise 

Mais vars() capture que l'espace de noms local, et je ne veux pas permettre l'accès à globals(). Des idées? Ou devrais-je juste aller avec la logique d'if-else?

Answer 1

Vous pouvez utiliser une liste blanche ... ou dans ce cas, un « dict blanc »

class Whatever(object): 

    allowed = { 'fizz':fizz, 'buzz':buz } 

    def __init__(self, func_name='fizz'): 
     try: 
      self.func = self.allowed[func_name] 
     except KeyError: 
      print "Bad func_name" 
      raise