Coverage for src/accounts/views/user_photo.py: 80%

1from rest_framework.generics import RetrieveUpdateDestroyAPIView 

2from rest_framework.parsers import MultiPartParser, FormParser 

3 

4from src.accounts.models.user_photo import UserPhoto 

5from src.accounts.serializers.user_photo import PhotoSerializer 

6 

7 

8class PhotoUploadView(RetrieveUpdateDestroyAPIView): 

9 serializer_class = PhotoSerializer 

10 # These parsers allow the API to accept file uploads (e.g., images) 

11 parser_classes = (MultiPartParser, FormParser) 

12 

13 def get_object(self): 

14 photo, _ = UserPhoto.objects.get_or_create(user=self.request.user) 

15 

16 return photo