mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-22 06:28:41 +02:00
Antoine Nguyen
8980f4c4c4
No more module federation at the moment, it does not seem very stable yet.
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
"""Contacts viewsets."""
|
|
|
|
import django_filters.rest_framework
|
|
from rest_framework.decorators import action
|
|
from rest_framework import filters, response, viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from . import models
|
|
from . import serializers
|
|
from . import tasks
|
|
|
|
|
|
class AddressBookViewSet(viewsets.GenericViewSet):
|
|
"""Address book viewset."""
|
|
|
|
permission_classes = (IsAuthenticated,)
|
|
serializer_class = serializers.AddressBookSerializer
|
|
|
|
@action(methods=["get"], detail=False)
|
|
def default(self, request):
|
|
"""Return default user address book."""
|
|
abook = request.user.addressbook_set.first()
|
|
if not abook:
|
|
return response.Response(status_code=404)
|
|
serializer = serializers.AddressBookSerializer(abook)
|
|
return response.Response(serializer.data)
|
|
|
|
@action(methods=["get"], detail=False)
|
|
def sync_to_cdav(self, request):
|
|
"""Synchronize address book with CardDAV collection."""
|
|
abook = request.user.addressbook_set.first()
|
|
if request.user.parameters.get_value("enable_carddav_sync", app="contacts"):
|
|
tasks.push_addressbook_to_carddav(request, abook)
|
|
return response.Response({})
|
|
|
|
@action(methods=["get"], detail=False)
|
|
def sync_from_cdav(self, request):
|
|
"""Synchronize from CardDAV address book."""
|
|
abook = request.user.addressbook_set.first()
|
|
if not abook.last_sync:
|
|
return response.Response()
|
|
if request.user.parameters.get_value("enable_carddav_sync", app="contacts"):
|
|
tasks.sync_addressbook_from_cdav(request, abook)
|
|
return response.Response({})
|
|
|
|
|
|
class CategoryViewSet(viewsets.ModelViewSet):
|
|
"""Category ViewSet."""
|
|
|
|
permission_classes = [IsAuthenticated]
|
|
serializer_class = serializers.CategorySerializer
|
|
|
|
def get_queryset(self):
|
|
"""Filter based on current user."""
|
|
qset = models.Category.objects.filter(user=self.request.user)
|
|
return qset.select_related("user")
|
|
|
|
|
|
class ContactFilter(django_filters.rest_framework.FilterSet):
|
|
"""Filter for Contact."""
|
|
|
|
category = django_filters.CharFilter(field_name="categories__name")
|
|
|
|
class Meta:
|
|
model = models.Contact
|
|
fields = ["categories"]
|
|
|
|
|
|
class ContactViewSet(viewsets.ModelViewSet):
|
|
"""Contact ViewSet."""
|
|
|
|
filter_backends = [
|
|
filters.SearchFilter,
|
|
django_filters.rest_framework.DjangoFilterBackend,
|
|
]
|
|
filterset_class = ContactFilter
|
|
permission_classes = [IsAuthenticated]
|
|
search_fields = ("^first_name", "^last_name", "^emails__address")
|
|
serializer_class = serializers.ContactSerializer
|
|
|
|
def get_queryset(self):
|
|
"""Filter based on current user."""
|
|
qset = models.Contact.objects.filter(addressbook__user=self.request.user)
|
|
return qset.prefetch_related("categories", "emails", "phone_numbers")
|
|
|
|
def perform_destroy(self, instance):
|
|
"""Also remove cdav contact."""
|
|
if self.request.user.parameters.get_value(
|
|
"enable_carddav_sync", app="contacts"
|
|
):
|
|
tasks.delete_contact_cdav(self.request, instance)
|
|
instance.delete()
|
|
|
|
|
|
class EmailAddressViewSet(viewsets.ReadOnlyModelViewSet):
|
|
"""EmailAddress viewset."""
|
|
|
|
filter_backends = [filters.SearchFilter]
|
|
permission_classes = [IsAuthenticated]
|
|
search_fields = (
|
|
"^address",
|
|
"^contact__display_name",
|
|
"^contact__first_name",
|
|
"^contact__last_name",
|
|
)
|
|
serializer_class = serializers.EmailAddressWithNameSerializer
|
|
|
|
def get_queryset(self):
|
|
"""Filter based on current user."""
|
|
return models.EmailAddress.objects.filter(
|
|
contact__addressbook__user=self.request.user
|
|
)
|