from django.shortcuts import render
from apps.opm.models import Operation, OptCate
from apps.opm.serializers import OperationCreateUpdateSerializer, OptCateCreateUpdateSerializer, OptCateSerializer
from apps.utils.viewsets import CustomModelViewSet


# Create your views here.
class OptCateViewSet(CustomModelViewSet):
    queryset = OptCate.objects.all()
    create_serializer_class = OptCateCreateUpdateSerializer
    update_serializer_class = OptCateCreateUpdateSerializer
    serializer_class = OptCateSerializer


class OperationViewSet(CustomModelViewSet):
    queryset = Operation.objects.all()
    create_serializer_class = OperationCreateUpdateSerializer
    update_serializer_class = OperationCreateUpdateSerializer