cgiacofei/TandoorRecipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

catch IntegrityError caused by mis-sorted tree

Browse Source
This commit is contained in:
smilerz 2021-09-30 16:17:08 -05:00
parent b7b0b9c690
commit 9bbbd72285
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cookbook/models.py
View File

@ -11,7 +11,7 @@ from django.contrib.postgres.indexes import GinIndex
from django.contrib.postgres.search import SearchVectorField from django.contrib.postgres.search import SearchVectorField
from django.core.files.uploadedfile import UploadedFile, InMemoryUploadedFile from django.core.files.uploadedfile import UploadedFile, InMemoryUploadedFile
from django.core.validators import MinLengthValidator from django.core.validators import MinLengthValidator
from django.db import models from django.db import models, IntegrityError
from django.db.models import Index, ProtectedError from django.db.models import Index, ProtectedError
from django.utils import timezone from django.utils import timezone
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
@ -43,7 +43,13 @@ class TreeManager(MP_NodeManager):
try: try:
return self.get(name__exact=kwargs['name'], space=kwargs['space']), False return self.get(name__exact=kwargs['name'], space=kwargs['space']), False
except self.model.DoesNotExist: except self.model.DoesNotExist:
return self.model.add_root(**kwargs), True with scopes_disabled():
try:
return self.model.add_root(**kwargs), True
except IntegrityError as e:
if 'Key (path)' in e.args[0]:
self.model.fix_tree(fix_paths=True)
return self.model.add_root(**kwargs), True
class TreeModel(MP_Node): class TreeModel(MP_Node):