Рассмотрите следующие модели и форму:
class Pizza(models.Model):
name = models.CharField(max_length=50)
class Topping(models.Model):
name = models.CharField(max_length=50)
ison = models.ManyToManyField(Pizza, blank=True)
class ToppingForm(forms.ModelForm):
class Meta:
model = Topping
При просмотре ToppingForm, он позволяет Вам выбрать, на какие пиццы начинки идут, и все - просто денди.
Мои вопросы: Как я определяю ModelForm для Пиццы, которая позволяет мне использовать в своих интересах отношения Many-Many между Пиццей и Положением во главе и позволяет мне выбрать, какие Начинки идут на Пиццу?
Я полагаю, вам нужно здесь добавить новое ModelMultipleChoiceField
в вашу PizzaForm
и вручную связать это поле формы с полем модели, поскольку Django не будет делать это автоматически для ты.
Следующий фрагмент может быть полезным:
class PizzaForm(forms.ModelForm):
class Meta:
model = Pizza
# Representing the many to many related field in Pizza
toppings = forms.ModelMultipleChoiceField(queryset=Topping.objects.all())
# Overriding __init__ here allows us to provide initial
# data for 'toppings' field
def __init__(self, *args, **kwargs):
# Only in case we build the form from an instance
# (otherwise, 'toppings' list should be empty)
if kwargs.get('instance'):
# We get the 'initial' keyword argument or initialize it
# as a dict if it didn't exist.
initial = kwargs.setdefault('initial', {})
# The widget for a ModelMultipleChoiceField expects
# a list of primary key for the selected data.
initial['toppings'] = [t.pk for t in kwargs['instance'].topping_set.all()]
forms.ModelForm.__init__(self, *args, **kwargs)
# Overriding save allows us to process the value of 'toppings' field
def save(self, commit=True):
# Get the unsave Pizza instance
instance = forms.ModelForm.save(self, False)
# Prepare a 'save_m2m' method for the form,
old_save_m2m = self.save_m2m
def save_m2m():
old_save_m2m()
# This is where we actually link the pizza with toppings
instance.topping_set.clear()
instance.topping_set.add(*self.cleaned_data['toppings'])
self.save_m2m = save_m2m
# Do we need to save all changes now?
if commit:
instance.save()
self.save_m2m()
return instance
Этот PizzaForm
затем можно использовать везде, даже в админке:
# yourapp/admin.py
from django.contrib.admin import site, ModelAdmin
from yourapp.models import Pizza
from yourapp.forms import PizzaForm
class PizzaAdmin(ModelAdmin):
form = PizzaForm
site.register(Pizza, PizzaAdmin)
Метод save ()
может быть слишком многословным, но вы можете упростить его, если вам не нужно поддерживать ситуацию commit = False
, тогда это будет примерно так:
def save(self):
instance = forms.ModelForm.save(self)
instance.topping_set.clear()
instance.topping_set.add(*self.cleaned_data['toppings'])
return instance
Честно говоря, я бы поместил отношение «многие ко многим» в модель Pizza
. Думаю, это ближе к реальности. Представьте себе человека, который заказывает несколько пицц.Он не сказал бы: «Я бы хотел сыр на пиццу один и второй и помидоры на пиццу один и третий», но, вероятно, «Одна пицца с сыром, одна пицца с сыром и помидорами, ...».
Конечно, можно заставить форму работать по-своему, но я бы выбрал:
class Pizza(models.Model):
name = models.CharField(max_length=50)
toppings = models.ManyToManyField(Topping)
У нас была аналогичная проблема в нашем приложении, которое использовало администратор django. Между пользователями и группами существует связь "многие ко многим", и добавить пользователей в группу нелегко. Я создал патч для django, который делает это, но не уделяет ему особого внимания ;-) Вы можете прочитать его и попробовать применить подобное решение к вашей проблеме с пиццей / топпингом. Таким образом, находясь внутри топпинга, вы можете легко добавить похожие пиццы или наоборот.
Я не уверен, что это то, что вы ищете, но знаете ли вы, что Pizza имеет атрибут topping_set
? Используя этот атрибут, вы можете легко добавить новую вершину в свой ModelForm.
new_pizza.topping_set.add(new_topping)
Я не уверен, что отвечу на вопрос на 100%, поэтому я буду работать с таким предположением:
Каждая пицца
может иметь много Topping
s. Каждый Топпинг
может иметь много пиццы
. Но если Топпинг
добавлен к Пицце
, этот Топпинг
, тогда автоматически будет Пицца
, и наоборот.
В этом случае лучше всего использовать таблицу отношений, которую Django поддерживает достаточно хорошо. Это могло бы выглядеть так:
models.py
class PizzaTopping(models.Model):
topping = models.ForeignKey('Topping')
pizza = models.ForeignKey('Pizza')
class Pizza(models.Model):
name = models.CharField(max_length=50)
topped_by = models.ManyToManyField('Topping', through=PizzaTopping)
def __str__(self):
return self.name
def __unicode__(self):
return self.name
class Topping(models.Model):
name=models.CharField(max_length=50)
is_on = models.ManyToManyField('Pizza', through=PizzaTopping)
def __str__(self):
return self.name
def __unicode__(self):
return self.name
forms.py
class PizzaForm(forms.ModelForm):
class Meta:
model = Pizza
class ToppingForm(forms.ModelForm):
class Meta:
model = Topping
Пример:
>>> p1 = Pizza(name="Monday")
>>> p1.save()
>>> p2 = Pizza(name="Tuesday")
>>> p2.save()
>>> t1 = Topping(name="Pepperoni")
>>> t1.save()
>>> t2 = Topping(name="Bacon")
>>> t2.save()
>>> PizzaTopping(pizza=p1, topping=t1).save() # Monday + Pepperoni
>>> PizzaTopping(pizza=p2, topping=t1).save() # Tuesday + Pepperoni
>>> PizzaTopping(pizza=p2, topping=t2).save() # Tuesday + Bacon
>>> tform = ToppingForm(instance=t2) # Bacon
>>> tform.as_table() # Should be on only Tuesday.
u'<tr><th><label for="id_name">Name:</label></th><td><input id="id_name" type="text" name="name" value="Bacon" maxlength="50" /></td></tr>\n<tr><th><label for="id_is_on">Is on:</label></th><td><select multiple="multiple" name="is_on" id="id_is_on">\n<option value="1">Monday</option>\n<option value="2" selected="selected">Tuesday</option>\n</select><br /> Hold down "Control", or "Command" on a Mac, to select more than one.</td></tr>'
>>> pform = PizzaForm(instance=p1) # Monday
>>> pform.as_table() # Should have only Pepperoni
u'<tr><th><label for="id_name">Name:</label></th><td><input id="id_name" type="text" name="name" value="Monday" maxlength="50" /></td></tr>\n<tr><th><label for="id_topped_by">Topped by:</label></th><td><select multiple="multiple" name="topped_by" id="id_topped_by">\n<option value="1" selected="selected">Pepperoni</option>\n<option value="2">Bacon</option>\n</select><br /> Hold down "Control", or "Command" on a Mac, to select more than one.</td></tr>'
>>> pform2 = PizzaForm(instance=p2) # Tuesday
>>> pform2.as_table() # Both Pepperoni and Bacon
u'<tr><th><label for="id_name">Name:</label></th><td><input id="id_name" type="text" name="name" value="Tuesday" maxlength="50" /></td></tr>\n<tr><th><label for="id_topped_by">Topped by:</label></th><td><select multiple="multiple" name="topped_by" id="id_topped_by">\n<option value="1" selected="selected">Pepperoni</option>\n<option value="2" selected="selected">Bacon</option>\n</select><br /> Hold down "Control", or "Command" on a Mac, to select more than one.</td></tr>'
Я сделал нечто подобное на основе кода Клемана с формой администратора пользователя:
# models.py
class Clinica(models.Model):
...
users = models.ManyToManyField(User, null=True, blank=True, related_name='clinicas')
# admin.py
class CustomUserChangeForm(UserChangeForm):
clinicas = forms.ModelMultipleChoiceField(queryset=Clinica.objects.all())
def __init__(self,*args,**kwargs):
if 'instance' in kwargs:
initial = kwargs.setdefault('initial',{})
initial['clinicas'] = kwargs['instance'].clinicas.values_list('pk',flat=True)
super(CustomUserChangeForm,self).__init__(*args,**kwargs)
def save(self,*args,**kwargs):
instance = super(CustomUserChangeForm,self).save(*args,**kwargs)
instance.clinicas = self.cleaned_data['clinicas']
return instance
class Meta:
model = User
admin.site.unregister(User)
UserAdmin.fieldsets += ( (u'Clinicas', {'fields': ('clinicas',)}), )
UserAdmin.form = CustomUserChangeForm
admin.site.register(User,UserAdmin)