OnlineJudge/monitor/views.py

# coding=utf-8
import redis
import datetime
from rest_framework.views import APIView
from judge.result import result
from django.conf import settings
from utils.shortcuts import success_response
from submission.models import Submission


class QueueLengthMonitorAPIView(APIView):
    def get(self, request):
        r = redis.Redis(host=settings.redis_config["host"], port=settings.redis_config["port"], db=settings.redis_config["db"])
        waiting_number = r.get("judge_queue_length")
        if waiting_number is None:
            waiting_number = 0
        now = datetime.datetime.now()
        return success_response({"time": ":".join([str(now.hour), str(now.minute), str(now.second)]),
                                 "count": waiting_number})