from rest_framework import viewsets
from .models import Element, Partner
from .serializers import PartnerSerializer, ElementSerializer


class PartnerViewSet(viewsets.ModelViewSet):
    """
    API endpoint that allows partners to be viewed or edited.
    """

    queryset = Partner.objects.all()
    serializer_class = PartnerSerializer
    
class ElementViewSet(viewsets.ModelViewSet):
    """
    API endpoint that allows elements to be viewed or edited.
    """

    queryset = Element.objects.all()
    serializer_class = ElementSerializer