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

allow city main image changing

Browse Source
This commit is contained in:
Kuroshini 2019-12-26 15:00:05 +03:00
parent 668bc92b97
commit 7648e478f5
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from utils.permissions import IsCountryAdmin
from utils.views import CreateDestroyGalleryViewMixin from utils.views import CreateDestroyGalleryViewMixin
from rest_framework.permissions import IsAuthenticatedOrReadOnly from rest_framework.permissions import IsAuthenticatedOrReadOnly
from django.shortcuts import get_object_or_404 from django.shortcuts import get_object_or_404
from django.db import IntegrityError
from utils.serializers import ImageBaseSerializer from utils.serializers import ImageBaseSerializer
from location.filters import RegionFilter from location.filters import RegionFilter
@ -81,6 +82,15 @@ class CityGalleryCreateDestroyView(common.CityViewMixin,
return gallery return gallery
def create(self, request, *args, **kwargs):
try:
return super(CityGalleryCreateDestroyView, self).create(request, *args, **kwargs)
except IntegrityError as e:
if not 'unique constraint' in e.args[0]:
raise e
models.CityGallery.objects.filter(city=kwargs['pk'], is_main=request.data['is_main']).delete()
return super(CityGalleryCreateDestroyView, self).create(request, *args, **kwargs)
class CityGalleryListView(common.CityViewMixin, class CityGalleryListView(common.CityViewMixin,
generics.ListAPIView): generics.ListAPIView):