Динамическое создание класса - Python

Question

Вот сценарий:

У меня есть два класса:

class A: 
    pass: 

class B: 
    pass 

Теперь я хочу, чтобы создать клиента, в том, что мне нужно иметь небольшой служебный метод, который должен верните шаблон/объект класса, например: class A, класс B, поскольку я передаю имя класса этой утилите, например get_obj(classA).

Теперь, это возможно? Если тогда, пожалуйста, предложите подход, так как я не получаю никакого правильного ответа на данный момент в Интернете.

Надеюсь, что у меня есть смысл.

Answer 1

Вот возможная реализация. Весь код содержится в одном «.py» файла

class A:                        
    pass                        


class B:                        
    pass                        

# map class name to class                          
_classes = {                       
     A.__name__: A,                    
     B.__name__: B,                    
}                         

def get_obj(cname):                      
    return _classes[cname]()                    

# test the function                          
if __name__ == '__main__':                   
    print get_obj('A') 

Он будет производить следующий вывод

<__main__.A instance at 0x1026ea950> 

Answer 2

Стандартная функция библиотеки namedtuple создает и возвращает класс. Внутри он использует exec. Это может быть вдохновение для того, что вам нужно.

код Источник: https://github.com/python/cpython/blob/master/Lib/collections/init.py#L356

Answer 3

globals() возвращает словарь, содержащий все символы, определенные в глобальной области видимости модуля (включая классы A и B):

a_and_b_module.py

class A: pass 
class B: pass 

def get_cls(cls_name): 
    return globals()[cls_name] 

Если вы ищете для простоты

Если код, который будет вызывать эту функцию внутри модуля, то вы можете исключить функцию вообще и использовать globals()[cls_name] непосредственно.

Если код, который будет вызывать эту функцию вне модуля, то вы можете использовать функцию getattr:

a_and_b_module.py

class A: pass 
class B: pass 

another_file.py

import a_and_b_module 

cls_name = 'A' 
chosen_cls = getattr(a_and_b_module, cls_name) 

Если вы ищете для полного контроля

Проблема с выше подхода заключается в том, что он может вернуть что-либо определенное в a_and_b_module.py, не ограничивая себя A и B. Если вы хотите, чтобы убедиться, что только А и В могут быть возвращены:

class A: pass 
class B: pass 

allowed_classes = ('A', 'B') 

def get_cls(cls_name): 
    assert cls_name in allowed_classes 
    return globals()[cls_name] 

Примечание: вы также можете быть заинтересованы в концепции factory.