from rest_framework import viewsets
from django_filters.rest_framework import DjangoFilterBackend
from .models import ClickableArea, Element3D
from .serializers import ClickableAreaSerializer, Element3DSerializer

class ClickableAreaViewSet(viewsets.ModelViewSet):
    queryset = ClickableArea.objects.all()
    serializer_class = ClickableAreaSerializer
    
    
class Element3DViewSet(viewsets.ModelViewSet):
    queryset = Element3D.objects.all()
    serializer_class = Element3DSerializer
    
    filterset_fields = ('parent', )
    filter_backends = [DjangoFilterBackend]