Как перепроверить ModelAdmin и его inline? - PullRequest
Новая
       15

Как перепроверить ModelAdmin и его inline?

5 голосов
/ 27 апреля 2010

У меня есть две модели (ModelParent и ModelChild) с одинаковыми полями m2m в предметной модели. ModelChild имеет внешний ключ для ModelParent, а ModelChild определен как встроенный для ModelParent на странице администратора. 

### models.py ###
  class Subject(Models.Model):
    pass

  class ModelParent(models.Model):
    subjects_parent = ManyToManyField(Subject)

  class ModelChild(models.Model):
    parent = ForeignKey(ModelParent)
    subjects_child = ManyToManyField(Subject)

### admin.py ###
  class ModelChildInline(admin.TabularInline):
      model = ModelChild

  class ModelParentAdmin(admin.ModelAdmin):
    inlines = [ModelChildInline]

  admin.site.register(ModelParent, ModelParentAdmin)

У меня есть одно важное ограничение, хотя поле subject_child в ModelChild не должно ссылаться ни на один предмет, который subject_parent делает со своим subject_parent.

Итак, если я выберу один и тот же предмет (в subject_parent и subject_child) на странице администратора для обеих моделей, как я могу это проверить? Если изменяется только одно поле, вы проверяете его по БД, но что, если оба изменяются (subject_parent и subject_child)? Как я могу проверить обе формы вместе перед сохранением?

Ответы [ 2 ]

5 голосов
/ 30 апреля 2010

Я унаследовал новый класс с именем ModelAdminWithInline от admin.ModelAdmin и изменил методы add_view (...) и change_view (...) для вызова функции is_cross_valid (self, form, formsets), где вы можете проверить все формы все вместе. Обе функции имели: 

#...
if all_valid(formsets) and form_validated:
#...

изменено на: 

#...
formsets_validated = all_valid(formsets)
cross_validated = self.is_cross_valid(form, formsets)
if formsets_validated and form_validated and cross_validated:
#...

Новая функция is_cross_valid (...) определена так: 

def is_cross_valid(self, form, formsets):
  return True

поэтому новый класс должен работать точно так же, как ModelAdmin, если вы не измените функцию is_cross_valid (...).

Теперь мой admin.py выглядит так: 

###admin.py###
class ModelAdminWithInline(admin.ModelAdmin):
  def is_cross_valid(self, form, formsets):
    return True

  def add_view(self, request, form_url='', extra_context=None):
    #modified code

  def change_view(self, request, object_id, extra_context=None):
    #modified code

class ModelChildInline(admin.TabularInline):
  model = ModelChild

class ModelParentAdmin(ModelAdminWithInline):
  inlines = [ModelChildInline]

  def is_cross_valid(self, form, formsets):
    #Do some cross validation on forms
    #For example, here is my particular validation:
    valid = True

    if hasattr(form, 'cleaned_data'):   

      subjects_parent = form.cleaned_data.get("subjects_parent")

      #You can access forms from formsets like this:
      for formset in formsets:
        for formset_form in formset.forms:
          if hasattr(formset_form, 'cleaned_data'):

            subjects_child = formset_form.cleaned_data.get("subjects_child")
            delete_form = formset_form.cleaned_data.get("DELETE")

            if subjects_child and (delete_form == False):
              for subject in subjects_child:
                if subject in subjects_parent:
                  valid = False
                  #From here you can still report errors like in regular forms:
                  if "subjects_child" in formset_form.cleaned_data.keys():
                    formset_form._errors["subjects_child"] = ErrorList([u"Subject %s is already selected in parent ModelParent" % subject])
                    del formset_form.cleaned_data["subjects_child"]
                  else:
                    formset_form._errors["subjects_child"] += ErrorList(u"Subject %s is already selected in parent ModelParent" % subject])

      #return True on success or False otherwise.
      return valid

admin.site.register(ModelParent, ModelParentAdmin)

Решение немного хакерское, но оно работает :). Ошибки отображаются так же, как с обычными классами ModelForm и ModelAdmin. Django 1.2 (который должен быть выпущен в ближайшее время) должен пройти валидацию модели, поэтому я надеюсь, что тогда эта проблема будет решена более красиво.

2 голосов
/ 27 апреля 2010

В административных классах нет метода clean (). Их формы делают. Каждый класс администратора имеет параметр с именем form. Вы просто расширяете форму по умолчанию (это обычная форма ModelAdmin), реализуете метод clean () и добавляете форму в класс администратора. Пример: 

class SomeForm(ModelForm):
  #some code
  def clean(self):
   #some code
class SomeAdminClass(ModelAdmin):
 #some code
 form = SomeForm
 #more code
...