Заказал ManyToManyField, который можно использовать в наборах полей - PullRequest
8 голосов
/ 07 июля 2010

Я работал через упорядоченный виджет ManyToManyField, и у него хорошо работает внешний интерфейс:

альтернативный текст http://i45.tinypic.com/33e79c8.png

К сожалению, у меня большие проблемы с работой бэкэнда. Очевидный способ подключить бэкэнд - это использовать таблицу through, связанную с моделью с ForeignKey s по обе стороны отношения, и перезаписать метод сохранения. Это будет прекрасно работать, за исключением того, что из-за особенностей содержимого, абсолютно необходимо, чтобы этот виджет был помещен в набор полей (используя свойство ModelAdmin fieldsets), что , по-видимому, невозможно .

У меня нет идей. Есть предложения?

Спасибо!

1 Ответ

8 голосов
/ 13 июля 2010

Что касается настройки моделей, вы правы в том, что сквозная таблица со столбцом «заказ» является идеальным способом ее представления. Вы также правы в том, что Django не позволит вам ссылаться на эти отношения в наборе полей. Хитрость для решения этой проблемы состоит в том, чтобы помнить, что имена полей, которые вы указываете в «fieldsets» или «fields» ModelAdmin, на самом деле относятся не к полям Model, а к полям ModelForm, который мы можем переопределить к радости нашего сердца. Со многими многими полями это сложно, но потерпите меня:

Допустим, вы пытаетесь представлять конкурсы и конкурентов, которые соревнуются в них, с упорядоченным множеством между конкурсами и конкурентами, где порядок представляет рейтинг конкурентов в этом конкурсе. Ваш models.py будет выглядеть так:

from django.db import models

class Contest(models.Model):
    name = models.CharField(max_length=50)
    # More fields here, if you like.
    contestants = models.ManyToManyField('Contestant', through='ContestResults')

class Contestant(models.Model):
    name = models.CharField(max_length=50)

class ContestResults(models.Model):
    contest = models.ForeignKey(Contest)
    contestant = models.ForeignKey(Contestant)
    rank = models.IntegerField()

Надеюсь, это похоже на то, с чем вы имеете дело. Теперь для админа. Я написал пример admin.py с множеством комментариев, чтобы объяснить, что происходит, но вот краткое изложение, чтобы помочь вам:

Поскольку у меня нет кода для заказанного вами виджета m2m, я использовал фиктивный виджет-заполнитель, который просто наследуется от TextInput. Вход содержит разделенный запятыми список (без пробелов) идентификаторов участников, а порядок их появления в строке определяет значение их столбца «rank» в модели ContestResults.

Что происходит, так это то, что мы переопределяем значение ModelForm по умолчанию для Contest своим собственным, а затем определяем поле "результаты" внутри него (мы не можем назвать поле "участники", так как будет конфликт имен с поле m2m в модели). Затем мы переопределяем __init__(), который вызывается, когда форма отображается в администраторе, поэтому мы можем получить любые результаты ContestResults, которые, возможно, уже были определены для конкурса, и использовать их для заполнения виджета. Мы также переопределяем save(), чтобы мы могли, в свою очередь, получить данные из виджета и создать необходимые ContestResults.

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

Я должен также добавить, что я действительно запустил этот код и убедился, что он работает.

from django import forms
from django.contrib import admin
from models import Contest, Contestant, ContestResults

# Generates a function that sequentially calls the two functions that were
# passed to it
def func_concat(old_func, new_func):
    def function():
        old_func()
        new_func()
    return function

# A dummy widget to be replaced with your own.
class OrderedManyToManyWidget(forms.widgets.TextInput):
    pass

# A simple CharField that shows a comma-separated list of contestant IDs.
class ResultsField(forms.CharField):
    widget = OrderedManyToManyWidget()

class ContestAdminForm(forms.models.ModelForm):
    # Any fields declared here can be referred to in the "fieldsets" or
    # "fields" of the ModelAdmin. It is crucial that our custom field does not
    # use the same name as the m2m field field in the model ("contestants" in
    # our example).
    results = ResultsField()

    # Be sure to specify your model here.
    class Meta:
        model = Contest

    # Override init so we can populate the form field with the existing data.
    def __init__(self, *args, **kwargs):
        instance = kwargs.get('instance', None)
        # See if we are editing an existing Contest. If not, there is nothing
        # to be done.
        if instance and instance.pk:
            # Get a list of all the IDs of the contestants already specified
            # for this contest.
            contestants = ContestResults.objects.filter(contest=instance).order_by('rank').values_list('contestant_id', flat=True)
            # Make them into a comma-separated string, and put them in our
            # custom field.
            self.base_fields['results'].initial = ','.join(map(str, contestants))
            # Depending on how you've written your widget, you can pass things
            # like a list of available contestants to it here, if necessary.
        super(ContestAdminForm, self).__init__(*args, **kwargs)

    def save(self, *args, **kwargs):
        # This "commit" business complicates things somewhat. When true, it 
        # means that the model instance will actually be saved and all is
        # good. When false, save() returns an unsaved instance of the model.
        # When save() calls are made by the Django admin, commit is pretty
        # much invariably false, though I'm not sure why. This is a problem
        # because when creating a new Contest instance, it needs to have been
        # saved in the DB and have a PK, before we can create ContestResults.
        # Fortunately, all models have a built-in method called save_m2m()
        # which will always be executed after save(), and we can append our
        # ContestResults-creating code to the existing same_m2m() method.
        commit = kwargs.get('commit', True)
        # Save the Contest and get an instance of the saved model
        instance = super(ContestAdminForm, self).save(*args, **kwargs)
        # This is known as a lexical closure, which means that if we store
        # this function and execute it later on, it will execute in the same
        # context (i.e. it will have access to the current instance and self).
        def save_m2m():
            # This is really naive code and should be improved upon,
            # especially in terms of validation, but the basic gist is to make
            # the needed ContestResults. For now, we'll just delete any
            # existing ContestResults for this Contest and create them anew.
            ContestResults.objects.filter(contest=instance).delete()
            # Make a list of (rank, contestant ID) tuples from the comma-
            # -separated list of contestant IDs we get from the results field.
            formdata = enumerate(map(int, self.cleaned_data['results'].split(',')), 1)
            for rank, contestant in formdata:
                ContestResults.objects.create(contest=instance, contestant_id=contestant, rank=rank)
        if commit:
            # If we're committing (fat chance), simply run the closure.
            save_m2m()
        else:
            # Using a function concatenator, ensure our save_m2m closure is
            # called after the existing save_m2m function (which will be
            # called later on if commit is False).
            self.save_m2m = func_concat(self.save_m2m, save_m2m)
        # Return the instance like a good save() method.
        return instance

class ContestAdmin(admin.ModelAdmin):
    # The precious fieldsets.
    fieldsets = (
        ('Basic Info', {
            'fields': ('name', 'results',)
        }),)
    # Here's where we override our form
    form = ContestAdminForm

admin.site.register(Contest, ContestAdmin)

Если вам интересно, я сам столкнулся с этой проблемой в проекте, над которым я работал, поэтому большая часть этого кода взята из этого проекта. Я надеюсь, что вы найдете это полезным.

...