You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.6 KiB
44 lines
1.6 KiB
10 months ago
|
from django.shortcuts import get_object_or_404
|
||
|
|
from rest_framework.generics import ListAPIView, CreateAPIView, DestroyAPIView
|
||
|
|
from rest_framework.pagination import PageNumberPagination
|
||
|
|
|
||
|
|
from core.permissions import IsAuthorOrReadOnly
|
||
|
|
from organizations.repositories import OrganizationRepository
|
||
|
|
from .repositories import PhotoRepository, FavouriteOrganizationRepository
|
||
|
|
from .serializers import PhotoSerializer, ListPhotoSerializer
|
||
|
|
|
||
|
|
# Create your views here.
|
||
|
|
|
||
|
|
|
||
|
|
class PhotoListCreateAPIView(ListAPIView, CreateAPIView):
|
||
|
|
queryset = PhotoRepository.all()
|
||
|
|
pagination_class = PageNumberPagination
|
||
|
|
permission_classes = [IsAuthorOrReadOnly]
|
||
|
|
|
||
|
|
def get_serializer_class(self):
|
||
|
|
if self.request.POST:
|
||
|
|
return PhotoSerializer
|
||
|
|
else:
|
||
|
|
return ListPhotoSerializer
|
||
|
|
|
||
|
|
def perform_create(self, serializer):
|
||
|
|
serializer.save(user=self.request.user)
|
||
|
|
|
||
|
|
|
||
|
|
class FavouriteOrganizationAPIView(ListAPIView, CreateAPIView, DestroyAPIView):
|
||
|
|
|
||
|
|
def get_queryset(self):
|
||
|
|
return FavouriteOrganizationRepository.get_filtered(user=self.request.user)
|
||
|
|
|
||
|
|
def get_object(self):
|
||
|
|
queryset = self.get_queryset()
|
||
|
|
obj = get_object_or_404(queryset, user=self.request.user, organization__pk=self.kwargs.get('id'))
|
||
|
|
return obj
|
||
|
|
|
||
|
|
def perform_create(self, serializer):
|
||
|
|
if FavouriteOrganizationRepository.get_filtered(user=self.request.user, organization=OrganizationRepository.get(
|
||
|
|
self.kwargs.get('id'))):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
serializer.save(user=self.request.user, organization=self.kwargs.get('id'))
|