phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

catch ForeignKey delete exception on city

Browse Source
This commit is contained in:
Kuroshini 2020-02-10 12:43:38 +03:00
parent dcf88408ae
commit f16adb23f7
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/location/views/back.py
View File

@ -1,5 +1,6 @@
"""Location app views.""" """Location app views."""
from django.contrib.postgres.fields.jsonb import KeyTextTransform from django.contrib.postgres.fields.jsonb import KeyTextTransform
from django.db.models.deletion import ProtectedError
from rest_framework import generics, status from rest_framework import generics, status
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.views import APIView from rest_framework.views import APIView
@ -167,6 +168,13 @@ class CityRUDView(common.CityViewMixin, generics.RetrieveUpdateDestroyAPIView):
IsGuest, IsGuest,
) )
def delete(self, request, *args, **kwargs):
try:
return self.destroy(request, *args, **kwargs)
except ProtectedError:
from rest_framework.serializers import ValidationError
raise ValidationError({'detail': 'City couldn’t be deleted. Some objects has this city'})
# Region # Region
class RegionListCreateView(common.RegionViewMixin, generics.ListCreateAPIView): class RegionListCreateView(common.RegionViewMixin, generics.ListCreateAPIView):