Переключатель python по имени класса?

Question

В настоящее время я делаю это, чтобы делать разные вещи в зависимости от типа объекта:

actions = { 
     SomeClass: lambda: obj.name 
     AnotherClass: lambda: self.normalize(obj.identifier) 
     ...[5 more of these]... 
    } 

    for a in actions.keys(): 
     if isinstance(obj, a): 
      return actions[a]() 

Можно ли вырезать для цикла, и сделать что-то вроде этого?

actions[something to do with obj]() 

Answer 1

Что о

actions[obj.__class__]() 

Answer 2

actions[type(obj)]() 

Answer 3

actions[obj.__class__]() 

работает, если obj действительно является экземпляром (скажем) SomeClass и не подкласса - так, если это может быть дело, результат будет отличаться от вашего текущего способа его обработки. Также обратите внимание, что это может привести к возникновению KeyError, если класс не имеет соответствующего действия. Для того, чтобы обрабатывать этот случай так же, как вы делаете сейчас (то есть ничего не делать), вы можете использовать

иметь значение по умолчанию возвращается.

О, и послушайте комментарий С.Лотта по вашему вопросу. Во многих случаях есть лучшие способы добиться чего-то подобного. Например, у вас есть все классы, которые определяют do_whatever(self) и просто звоните obj.do_whatever().

Answer 4

results = [func() for cls, func in actions.iteritems() if isinstance(obj, cls)] 

Там будет ноль или больше результатов, если ваш объект isinstance нуля или более классов-ключей.

Использование type(obj) как ключ будет работать только в том случае, если ваш объект имеет такой тип. Если он будет дальше по дереву наследования, вы пропустите его.

Answer 5

class SomeClass(object): 
.... 
    def action(self): 
     return self.name 

class AnotherClass(object): 
.... 
    def action(self): 
     return self.normalize(self.identifier) 

[5 more classes like the above two] 

a.action() 

Упрощенный. Отчетливое. Более растяжимая. Меньше магии. Нет словаря. Нет петли.

Answer 6

Я предполагаю, что у вас есть родительский класс для всех этих или, по крайней мере, для микширования. Поместите функцию возврата по умолчанию в родительский или mixin, а затем переопределите ее в тех, которые отличаются друг от друга ... Это единственный правильный способ сделать это.

Уверенный, он делает дополнительный код, но по крайней мере он инкапсулирован и масштабируется. Скажите, что вы хотите добавить поддержку еще пяти классов. Вместо того, чтобы изменять этот код, просто добавьте правильный код в новые классы. По внешнему виду это две строки для каждого класса (определение функции и обратная линия). Это не плохо, не так ли?

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

class MyMixin: 
    def my_return(self, *args): 
    return self.name 
    ... possibly other things... 

class SomeClass(MyMixin): 
    ... no alteration to the default ... 

class AnotherClass(MyParent, MyMixin): 
    def my_return(self, *args): 
    return args[0].normalize(self.identifier) 
    ... blabla 


# now, this is in the caller object... 
try: 
    rval = obj.my_return(self) # this is the caller object 'self', not the 'self' in the 'obj' 
    #dosomething with rval 
except Exception: 
    pass #no rval for this object type, skipping it...