/home/report/liman_adapter/Moshtarian/ Current File: /home/report/liman_adapter/Moshtarian/api.py

from rest_framework import viewsets
from rest_framework.decorators import api_view
from rest_framework.response import Response
from django.http import JsonResponse
from Core.table_engine.src.table_engine import TableEngine
from .models import Moshtarian
from .serializer import MoshtarianSerializer


class MoshtarianApi(viewsets.ModelViewSet):
    queryset = Moshtarian.objects.all()
    serializer_class = MoshtarianSerializer
    http_method_names = ['get', 'post', 'patch', 'put', 'delete']

    @staticmethod
    @api_view(['POST'])
    def table(request):
        try:
            table_engine = TableEngine(Moshtarian.objects, request.dict).init_cols(
                ["ID", "name", "code"]).execute()
            return JsonResponse(table_engine)
        except Exception as e:
            return Response(str(e), 422)