Добавление функциональности в Django FlatPages без изменения оригинального приложения Django

Я хотел бы добавить поле в модель базы данных Django FlatPage, но я действительно не знаю, как это расширить без редактирования исходного приложения.

Что я хочу сделать, так это добавить в модель следующее поле:


from django.db import models
from django.contrib.flatpages.models import FlatPage as FlatPageOld

class FlatPage(FlatPageOld):
    order = models.PositiveIntegerField(unique=True)

Как мне добавить это в модель FlatPage?

Заранее спасибо

Ответы

Ответ 1

Твой подход в порядке - вы просто не видите результат, потому что старая модель платформы зарегистрирована в админе, а новая - нет. Вот что вы могли бы сделать в своем новом приложении admin.py(используя менее двусмысленное название, чем то, что у вас было выше):

from django.contrib import admin
from django.contrib.flatpages.admin import FlatPageAdmin
from django.contrib.flatpages.forms import FlatpageForm
from django.contrib.flatpages.models import FlatPage

from models import ExtendedFlatPage

class ExtendedFlatPageForm(FlatpageForm):
    class Meta:
        model = ExtendedFlatPage

class ExtendedFlatPageAdmin(FlatPageAdmin):
    form = ExtendedFlatPageForm
    fieldsets = (
        (None, {'fields': ('url', 'title', 'content', 'sites', 'order')}),
    )     

admin.site.unregister(FlatPage)
admin.site.register(ExtendedFlatPage, ExtendedFlatPageAdmin)

Очевидно, здесь есть несколько вещей, но, самое главное, модель FlatPage не регистрируется и модель ExtendedFlatPage регистрируется на своем месте.

Ответ 2

И метод в вашем сообщении не работает, потому что...?

Если по какой-то причине вам действительно нужно возиться со встроенным классом FlatPage и редактировать его динамически, вы можете подключиться к сигналу class_prepared:

http://docs.djangoproject.com/en/dev/ref/signals/#class-prepared

Edit

Вот как вы это сделаете с помощью class_prepared:

from django.db.models.signals import class_prepared
from django.db import models

def alter_flatpages(sender, **kwargs):
    if sender.__module__ == 'django.contrib.flatpages.models' and sender.__name__ == 'FlatPage':
        order = models.IntegerField()
        order.contribute_to_class(sender, 'order')

class_prepared.connect(alter_flatpages)

Поместите это, скажем, "сигналы .py" в тот же каталог, что и ваши settings.py, и добавьте "сигналы" в верхний (это важно, чтобы убедиться, что обработчик сигнала устанавливается вовремя) списка INSTALLED_APPS.

Однако это все равно не получит поле, отображаемое в Admin, потому что для FlatPages есть специальный класс ModelAdmin, который явно отображает поля. Поэтому после регистрации в приложении flatpages вам нужно отменить регистрацию где-нибудь (admin.site.unregister) и зарегистрировать собственный ModelAdmin.