Как динамически генерировать подкласс в функции?

Question

Я пытаюсь написать функцию, которая создает новый подкласс с именем, который передается в качестве аргумента. Я не знаю, какие инструменты были бы лучше для этого, но я сделал это в коде ниже, и мне удалось создать подкласс «x», а не «MySubClass», как предполагалось. Как я могу правильно написать эту функцию?

class MySuperClass: 
    def __init__(self,attribute1): 
     self.attribute1 = attribute1 

def makeNewClass(x): 
    class x(MySuperClass): 
     def __init__(self,attribute1,attribute2): 
      self.attribute2 = attribute2 

x = "MySubClass" 
makeNewClass(x) 
myInstance = MySubClass(1,2) 

Answer 1

Самый безопасный и простой способ сделать это было бы использовать встроенную функцию type. Это принимает необязательный второй аргумент (кортеж базовых классов) и третий аргумент (dict of functions). Моя рекомендация будет следующее:

def makeNewClass(x): 
    def init(self,attribute1,attribute2): 
     # make sure you call the base class constructor here 
     self.attribute2 = attribute2 

    # make a new type and return it 
    return type(x, (MySuperClass,), {'__init__': init}) 

x = "MySubClass" 
MySubClass = makeNewClass(x) 

Вам нужно будет заполнить Dict третьего аргумента все, что вы хотите, чтобы новый класс иметь. Очень вероятно, что вы генерируете классы и захотите отбросить их обратно в список, где имена фактически не будут иметь значения. Однако я не знаю вашего случая использования.

---

В качестве альтернативы вы можете получить доступ к globals и поставить новый класс в том, что вместо этого. Это действительно странно динамический способ генерации классов, но это лучший способ, который я могу представить, чтобы получить именно то, что вам кажется нужным.

def makeNewClass(x): 
    def init(self,attribute1,attribute2): 
     # make sure you call the base class constructor here 
     self.attribute2 = attribute2 

    globals()[x] = type(x, (MySuperClass,), {'__init__': init}) 

Answer 2

ответ Райана является полным, но я думаю, что стоит отметить, что есть по крайней мере один другой гнусный способ сделать это, помимо использования встроенной type и exec/eval или что-то:

class X: 
    attr1 = 'some attribute' 

    def __init__(self): 
     print 'within constructor' 

    def another_method(self): 
     print 'hey, im another method' 

# black magics 
X.__name__ = 'Y' 
locals()['Y'] = X 
del X 

# using our class 
y = locals()['Y']() 
print y.attr1 
y.another_method() 

Обратите внимание, что я использовал строки только при создании класса Y и при инициализации экземпляра Y, поэтому этот метод является полностью динамическим.