from django.shortcuts import render
from rest_framework.views import APIView
from rest_framework.response import Response

from crm.models import Consumer
from examtest.models import ExamTest
from question.models import Question

import pickle
# Create your views here.

class BasicCount(APIView):
    '''
    获取基本统计数据
    '''
    def get(self, request, format=None):
        ret={}
        ret['consumer1_count'] = Consumer.objects.filter(is_delete=False).exclude(username=None).count()
        ret['consumer2_count'] = Consumer.objects.filter(is_delete=False).exclude(subjects=None).count()
        ret['test_count'] = ExamTest.objects.filter(is_delete=False).count()
        ret['question_count'] = Question.objects.filter(is_delete=False).count()
        return Response(ret)
    
class Quota(APIView):
    '''
    获取考试名额
    '''
    def get(self, request, format=None):
        with open('quota.dat','rb') as f:
            data = pickle.load(f)
            return Response(data)