Source code for intranet.apps.announcements.api

import logging

from rest_framework import generics, permissions

from .models import Announcement
from .serializers import AnnouncementSerializer

logger = logging.getLogger(__name__)




class IsAnnouncementAdminOrReadOnly(permissions.BasePermission):
    def has_permission(self, request, view):
        return (
            (request.user and request.user.is_authenticated and not request.user.is_restricted and request.user.oauth_and_api_access) or request.auth
        ) and (request.method in permissions.SAFE_METHODS or (request.user and request.user.is_announcements_admin))





class ListCreateAnnouncement(generics.ListCreateAPIView):
    serializer_class = AnnouncementSerializer
    permission_classes = (IsAnnouncementAdminOrReadOnly,)

    def get_queryset(self):
        user = self.request.user
        if not user and self.request.auth:
            return Announcement.objects.filter(groups__isnull=True)
        return Announcement.objects.visible_to_user(user).prefetch_related("groups")





class RetrieveUpdateDestroyAnnouncement(generics.RetrieveUpdateDestroyAPIView):
    serializer_class = AnnouncementSerializer
    permission_classes = (IsAnnouncementAdminOrReadOnly,)

    def get_queryset(self):
        user = self.request.user
        return Announcement.objects.visible_to_user(user).prefetch_related("groups")