【OJ】测评机实现-django/python

发布时间 2023-04-20 17:07:42作者: iuk11

tasks.py文件(celery)

通过views.py中的submit函数中的evaluate_submission.delay(sub.id)方法,对一个子进程程序进行:编译、产生.out文件、对比testcase文件、返回测评结果。
sub内包含html中通过表单(form)的post请求提交的内容:问题id,提交者,语言,代码。

补充编译指令解释

g++:这是调用 g++ 编译器的命令。
{codefile}:这是你要编译的 C++ 源代码文件名。你需要将 {codefile} 替换为你实际的源代码文件名。
-O2:这是一个优化标志,告诉编译器对代码进行二级优化。这可以使代码运行更快,但也可能增加编译时间。
-Wall:这个标志启用编译过程中的所有警告信息,可以帮助识别代码中的潜在问题。
-lm:这个标志告诉连接器链接数学库(libm),提供了一些可能在程序中使用的与数学有关的函数。
--static:这个选项告诉连接器创建一个静态链接的可执行文件,这意味着所有必需的库将被包含在可执行文件中。这会使可执行文件变得更大,但可以避免共享库的依赖问题。
-DONLINE_JUDGE:这个选项定义了一个名为 ONLINE_JUDGE 的预处理宏,可以根据它是否为在线评测系统进行有条件地编译代码。
-o main:这个选项指定编译器生成的输出文件的名称。在这种情况下,输出文件将命名为 main

from celery import shared_task
import os
# 产生子进程的模块
import subprocess
# 对应的models,声明的数据库表
from machine.models.coder.coder import Coder
from machine.models.problem.problem import Problem
from machine.models.problem.testcase import TestCase
from machine.models.problem.submission import Submission
# 声明judge文件下的函数
from machine.judge import compiles

# 装饰器 通过name属性找到该函数并调用
@shared_task(name = "evaluate_submission")
def evaluate_submission(sub_id):
    try:
        # 读取到提交元素
        submission = Submission.objects.get(pk=sub_id)
    except:
        return 
    # 提交者
    username = submission.submitter.user.username
    # 当前题目,题目提交数+1
    cur_problem = submission.problem
    cur_problem.num_submissions += 1
    # 时间限制
    tl = cur_problem.time_limit
    timeout_lst = ["timeout", str(tl)]
    # 获取语言
    langu = submission.lang
    # 定义codefile(代码文件名)
    codefile = {
        "C"    :"{}_{}.c",
        "CPP"  :"{}_{}.cpp",
        "JAVA" :"{}_{}.java",
        "PYTH3":"{}_{}.py",
    }
    filename = ''
    compiler = ''
    filemir = "{}_{}".format(username,sub_id)
    filename = codefile[langu].format(username,sub_id)
    # 保存代码到文件
    user_code = open("submissions/{}".format(filename),"w+")
    print(submission.code,file = user_code)
    user_code.close()
    # 定义编译命令
    build_cmd = {
        "C"    :"gcc {} -o main -Wall -lm -O2 -std=c99 --static -DONLINE_JUDGE",
        "CPP"  :"g++ {} -O2 -Wall -lm --static -DONLINE_JUDGE -o",
        "JAVA" :"javac {}",
        "PYTH3":"python3 -m py_compile {}",
    }
    # 开启子进程编译代码->可执行文件
    p = subprocess.Popen(build_cmd[langu].format(filename),shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    stdout, stderr = p.communicate()
    # 获取当前问题所有的测试用例
    testcases = TestCase.objects.filter(problem=cur_problem)
    # 执行测试用例
    for tc in testcases:
        input_filename = tc.input_file.name
        output_filename = tc.output_file.name
        langu = submission.lang