from django.shortcuts import render
from rest_framework.views import APIView
from rest_framework.permissions import IsAdminUser
from rest_framework.response import Response
from apps.inm.models import FIFO
from apps.mtm.models import Material
from apps.sam.models import Order
# Create your views here.

class CleanDataView(APIView):
    permission_classes = [IsAdminUser]

    def post(self, request, format=None):
        """
        清空数据库
        """
        Order.objects.delete()
        FIFO.objects.delete()
        Material.objects.filter(type__in=[Material.MA_TYPE_GOOD, Material.MA_TYPE_HALFGOOD]).delete()
        return Response()