Django's ModelForm validazione unique_together

Question

Ho un modello Django simile a questo.

class Solution(models.Model): 
    ''' 
    Represents a solution to a specific problem. 
    ''' 
    name = models.CharField(max_length=50) 
    problem = models.ForeignKey(Problem) 
    description = models.TextField(blank=True) 
    date = models.DateTimeField(auto_now_add=True) 

    class Meta: 
     unique_together = ("name", "problem") 

Io uso una forma per l'aggiunta di modelli che assomiglia a questo:

class SolutionForm(forms.ModelForm): 
    class Meta: 
     model = Solution 
     exclude = ['problem'] 

mio problema è che il SolutionForm non convalida Solution s' unique_together vincolo e, quindi, restituisce un IntegrityError quando si cerca di salva il modulo. So che potrei usare validate_unique per verificare manualmente questo problema, ma mi chiedevo se c'è un modo per capirlo nella convalida del modulo e restituire automaticamente un errore del modulo.

Grazie.

Answer 1

sono riuscito a risolvere il problema senza modificare la vista con l'aggiunta di un metodo pulito per il mio modulo:

class SolutionForm(forms.ModelForm): 
    class Meta: 
     model = Solution 
     exclude = ['problem'] 

    def clean(self): 
     cleaned_data = self.cleaned_data 

     try: 
      Solution.objects.get(name=cleaned_data['name'], problem=self.problem) 
     except Solution.DoesNotExist: 
      pass 
     else: 
      raise ValidationError('Solution with this Name already exists for this problem') 

     # Always return cleaned_data 
     return cleaned_data 

L'unica cosa che mi serve fare ora nella vista è aggiungere una proprietà problema al modulo prima di eseguire is_valid.

Answer 2

Come dice Felix, i ModelForms devono controllare il vincolo unique_together nella loro convalida.

Tuttavia, nel tuo caso in realtà stai escludendo un elemento di tale vincolo dal tuo modulo. Immagino che questo sia il tuo problema: in che modo il modulo controllerà il vincolo, se la metà non è nemmeno nella forma?

Answer 3

Avrete bisogno di fare qualcosa di simile:

def your_view(request): 
    if request.method == 'GET': 
     form = SolutionForm() 
    elif request.method == 'POST': 
     problem = ... # logic to find the problem instance 
     solution = Solution(problem=problem) # or solution.problem = problem 
     form = SolutionForm(request.POST, instance=solution) 
     # the form will validate because the problem has been provided on solution instance 
     if form.is_valid(): 
      solution = form.save() 
      # redirect or return other response 
    # show the form 

Answer 4

Ho risolto questo stesso problema sostituendo il metodo del ModelForm validate_unique():

def validate_unique(self): 
    exclude = self._get_validation_exclusions() 
    exclude.remove('problem') # allow checking against the missing attribute 

    try: 
     self.instance.validate_unique(exclude=exclude) 
    except ValidationError, e: 
     self._update_errors(e.message_dict) 

Ora ho solo sempre assicurarsi che l'attributo non previsto sul modulo è ancora disponibile, ad esempio instance=Solution(problem=some_problem) sull'inizializzatore.

Answer 5

Con l'aiuto di risposta di Jarmo, il seguente sembra funzionare bene per me (in Django 1.3), ma è possibile che io ho rotto alcuni casi angolo (ci sono un sacco di biglietti circostanti _get_validation_exclusions):

class SolutionForm(forms.ModelForm): 
    class Meta: 
     model = Solution 
     exclude = ['problem'] 

    def _get_validation_exclusions(self): 
     exclude = super(SolutionForm, self)._get_validation_exclusions() 
     exclude.remove('problem') 
     return exclude 

Non sono sicuro, ma questo mi sembra un bug di Django ... ma dovrei dare un'occhiata ai problemi precedentemente segnalati.

---

Modifica: ho parlato troppo presto. Forse quello che ho scritto sopra funzionerà in alcune situazioni, ma non nelle mie; Ho finito per usare direttamente la risposta di Jarmo.

Answer 6

la soluzione di @sttwister ha ragione ma può essere semplificata.

class SolutionForm(forms.ModelForm): 

    class Meta: 
     model = Solution 
     exclude = ['problem'] 

    def clean(self): 
     cleaned_data = self.cleaned_data 
     if Solution.objects.filter(name=cleaned_data['name'],   
            problem=self.problem).exists(): 
      raise ValidationError(
        'Solution with this Name already exists for this problem') 

     # Always return cleaned_data 
     return cleaned_data 

Come bonus voi non retreive l'oggetto in caso di duplicato, ma verificare solo se esiste nel database risparmiando un po 'di prestazioni.

Answer 7

Se si desidera che il messaggio di errore da un associato al campo name (e apparire accanto ad essa):

def clean(self): 
    cleaned_data = super().clean() 
    name_field = 'name' 
    name = cleaned_data.get(name_field) 

    if name: 
     if Solution.objects.filter(name=name, problem=self.problem).exists(): 
      cleaned_data.pop(name_field) # is also done by add_error 
      self.add_error(name_field, _('There is already a solution with this name.')) 

    return cleaned_data