декоратора, который создает классы

Question

У меня есть класс

class Test: 
    name = "Test" 

Я хочу создать декоратор mc_me, который после того, как:

@mc_me 
class Test: 
    name = "Test" 

предоставит мне три класса так же, как если бы я написал:

class Test: 
    name = "Test" 

class TestSuper: 
    name = "TestSuper" 

class TestExtra: 
    name = "TestExtra" 

Таким образом, по сути, первая сложная часть - это глубокое копирование класса, а второе назначение трех переменных в области вызова. Кажется, что это должно быть возможно ...

Answer 1

import inspect 

def mc_me(cls): 
    main_locals = inspect.currentframe().f_back.f_locals 
    for suffix in ['Super', 'Extra']: 
     new_name = cls.__name__ + suffix 
     main_locals[new_name] = type(new_name,(), {'name': new_name}) 
    return cls 

Будьте осторожны, это не самый лучший способ кодирования. С большой властью приходит большая ответственность. Я рекомендую вам проверить все возможные недостатки подхода inspect (Google - ваш друг).

Answer 2

Невозможно. Делать 3 класса легко. Однако присвоение результатов 3 переменным? Это нарушает правила области Python.

@mc_me 
class Test: 
    name = "Test" 

эквивалентно

class Test: 
    name = "Test" 
Test = mc_me(Test) 

Там нет никакого способа для декоратора, чтобы сказать, что он хочет создать еще несколько переменных и назначить их тоже. Даже если вам каким-то образом удалось заставить его работать (возможно, только с областью модуля или класса с inspect безумием), Python не мог бы знать, что ссылки на TestSuper или TestExtra должны относиться к переменным, созданным декоратором при компиляции байт-кода.

Answer 3

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

# Library code 
def mc_me(base): 
    class Sp(base): 
     name = base.name + "Super" 

    class Ex(base): 
     name = base.name + "Extra" 

    return (Sp, Ex) 

# Application code 
class Test: 
    name = "Test" 

(TestSuper, TestExtra) = mc_me(Test)