Django:IntegrityError podczas add()

Czy błąd można odtworzyć?

Tak, użyjmy słynnej Publication i Article modele z dokumentacja Django . Następnie utwórzmy kilka wątków.

import threading
import random

def populate():

    for i in range(100):
        Article.objects.create(headline = 'headline{0}'.format(i))
        Publication.objects.create(title = 'title{0}'.format(i))

    print 'created objects'


class MyThread(threading.Thread):

    def run(self):
        for q in range(1,100):
            for i in range(1,5):
                pub = Publication.objects.all()[random.randint(1,2)]
                for j in range(1,5):
                    article = Article.objects.all()[random.randint(1,15)]
                    pub.article_set.add(article)

            print self.name


Article.objects.all().delete()
Publication.objects.all().delete()
populate()
thrd1 = MyThread()
thrd2 = MyThread()
thrd3 = MyThread()

thrd1.start()
thrd2.start()
thrd3.start()

Na pewno zauważysz unikalne naruszenia ograniczeń klucza typu zgłoszonego w raporcie o błędzie . Jeśli ich nie widzisz, spróbuj zwiększyć liczbę wątków lub iteracji.

Czy jest jakieś rozwiązanie?

TAk. Użyj through modele i get_or_create . Oto plik models.py zaadaptowany z przykładu w dokumentacji django.

class Publication(models.Model):
    title = models.CharField(max_length=30)

    def __str__(self):              # __unicode__ on Python 2
        return self.title

    class Meta:
        ordering = ('title',)

class Article(models.Model):
    headline = models.CharField(max_length=100)
    publications = models.ManyToManyField(Publication, through='ArticlePublication')

    def __str__(self):              # __unicode__ on Python 2
        return self.headline

    class Meta:
        ordering = ('headline',)

class ArticlePublication(models.Model):
    article = models.ForeignKey('Article', on_delete=models.CASCADE)
    publication = models.ForeignKey('Publication', on_delete=models.CASCADE)
    class Meta:
        unique_together = ('article','publication')

Oto nowa klasa wątków, która jest modyfikacją powyższej.

class MyThread2(threading.Thread):

    def run(self):
        for q in range(1,100):
            for i in range(1,5):
                pub = Publication.objects.all()[random.randint(1,2)]
                for j in range(1,5):
                    article = Article.objects.all()[random.randint(1,15)]
                    ap , c = ArticlePublication.objects.get_or_create(article=article, publication=pub)
            print 'Get  or create', self.name

Przekonasz się, że wyjątek już się nie pojawia. Zapraszam do zwiększenia liczby iteracji. Doszedłem tylko do 1000 z get_or_create nie rzucił wyjątku. Jednak add() zwykle rzucał wyjątek w 20 iteracjach.

Dlaczego to działa?

Ponieważ get_or_create jest atomowy.

Aktualizacja: Dzięki @louis za wskazanie, że model przelotowy można w rzeczywistości wyeliminować. Stąd get_or_create w MyThread2 można zmienić jako.

ap , c = article.publications.through.objects.get_or_create(
            article=article, publication=pub)