from rest_framework import generics, permissions, request


from .serial import *



class UserViewSet(generics.ListCreateAPIView):
    serializer_class = UserSerializer
    ordering = ('id',)

    queryset = User.objects.all()


class CompaniesViewSet(generics.ListCreateAPIView):
    serializer_class = CompanySerializer
    ordering = ('id',)

    queryset = Company.objects.all()


class CompanyViewSet(generics.RetrieveUpdateDestroyAPIView):
    serializer_class = CompanySerializer
    ordering = ('id',)

    queryset = Company.objects.all()


class CategoriesViewSet(generics.ListCreateAPIView):
    serializer_class = CategorySerializer
    ordering = ('id',)

    queryset = Category.objects.all()


class CategoryViewSet(generics.RetrieveUpdateDestroyAPIView):
    serializer_class = CategorySerializer
    ordering = ('id',)
    lookup_field = 'slug'

    queryset = Category.objects.all()