2020-01-02 18:32:15 +00:00
|
|
|
def register_model(verbose_name):
|
2020-01-05 12:37:51 +00:00
|
|
|
'''Decorator to register a section subclass'''
|
2020-01-02 18:32:15 +00:00
|
|
|
def wrapper(model):
|
2020-01-05 12:37:51 +00:00
|
|
|
parent_model = model.__bases__[-1]
|
|
|
|
parent_model.TYPES.append((model.__name__.lower(), verbose_name))
|
2020-01-02 18:32:15 +00:00
|
|
|
return model
|
|
|
|
return wrapper
|
|
|
|
|
|
|
|
def register_view(section_class):
|
2020-01-05 12:37:51 +00:00
|
|
|
'''Decorator to connect a section model to a view class'''
|
2020-01-02 18:32:15 +00:00
|
|
|
def wrapper(model):
|
|
|
|
section_class.view_class = model
|
|
|
|
return model
|
|
|
|
return wrapper
|