gitrusprusandCursor
4eb4c145b6
Users can register, log in, and manage profile/password in a personal account page; server data is scoped by owner across contacts, maps, tags, and import. Co-authored-by: Cursor <cursoragent@cursor.com>
87 lines
3.4 KiB
Python
87 lines
3.4 KiB
Python
from rest_framework import viewsets
|
|
from rest_framework.exceptions import PermissionDenied, ValidationError
|
|
|
|
from core.access import use_jwt_auth
|
|
from core.drf_mixins import JwtAuthMixin
|
|
from .mixins import OwnerScopedMixin
|
|
from .models import Contact, Relation, NetworkMap, NetworkMapMembership, NetworkMapType
|
|
from .serializers import (
|
|
ContactSerializer,
|
|
RelationSerializer,
|
|
NetworkMapSerializer,
|
|
NetworkMapMembershipSerializer,
|
|
NetworkMapTypeSerializer,
|
|
)
|
|
|
|
|
|
class ContactViewSet(JwtAuthMixin, OwnerScopedMixin, viewsets.ModelViewSet):
|
|
queryset = Contact.objects.all()
|
|
serializer_class = ContactSerializer
|
|
|
|
def get_queryset(self):
|
|
qs = super().get_queryset()
|
|
q = self.request.query_params.get('search', '')
|
|
if q:
|
|
qs = qs.filter(name__icontains=q)
|
|
return qs
|
|
|
|
|
|
class RelationViewSet(JwtAuthMixin, OwnerScopedMixin, viewsets.ModelViewSet):
|
|
queryset = Relation.objects.select_related('source', 'target').all()
|
|
serializer_class = RelationSerializer
|
|
|
|
|
|
class NetworkMapViewSet(JwtAuthMixin, OwnerScopedMixin, viewsets.ModelViewSet):
|
|
queryset = NetworkMap.objects.select_related('map_type').all()
|
|
serializer_class = NetworkMapSerializer
|
|
|
|
|
|
class NetworkMapTypeViewSet(JwtAuthMixin, OwnerScopedMixin, viewsets.ModelViewSet):
|
|
queryset = NetworkMapType.objects.all()
|
|
serializer_class = NetworkMapTypeSerializer
|
|
|
|
def perform_destroy(self, instance):
|
|
if instance.is_default:
|
|
raise ValidationError({'detail': 'Нельзя удалить тип карты по умолчанию.'})
|
|
if instance.maps.exists():
|
|
raise ValidationError({'detail': 'Тип используется картами сети. Сначала смените тип у карт.'})
|
|
instance.delete()
|
|
|
|
|
|
class NetworkMapMembershipViewSet(JwtAuthMixin, viewsets.ModelViewSet):
|
|
serializer_class = NetworkMapMembershipSerializer
|
|
|
|
def get_queryset(self):
|
|
map_id = self.kwargs.get('map_pk')
|
|
qs = NetworkMapMembership.objects.filter(
|
|
map_id=map_id
|
|
).select_related('contact', 'map')
|
|
if use_jwt_auth():
|
|
user = self.request.user
|
|
if user and user.is_authenticated:
|
|
return qs.filter(map__owner=user, contact__owner=user)
|
|
return qs.none()
|
|
return qs
|
|
|
|
def perform_create(self, serializer):
|
|
map_id = self.kwargs.get('map_pk')
|
|
if use_jwt_auth():
|
|
user = self.request.user
|
|
if not user or not user.is_authenticated:
|
|
raise PermissionDenied()
|
|
network_map = NetworkMap.objects.filter(pk=map_id, owner=user).first()
|
|
if not network_map:
|
|
raise PermissionDenied()
|
|
contact = serializer.validated_data.get('contact')
|
|
if contact.owner_id != user.id:
|
|
raise ValidationError({'contact': 'Контакт не принадлежит текущему пользователю.'})
|
|
serializer.save(map_id=map_id)
|
|
|
|
def perform_update(self, serializer):
|
|
if use_jwt_auth():
|
|
user = self.request.user
|
|
contact = serializer.validated_data.get('contact', serializer.instance.contact)
|
|
if contact.owner_id != user.id:
|
|
raise ValidationError({'contact': 'Контакт не принадлежит текущему пользователю.'})
|
|
serializer.save()
|