mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2024-09-21 16:33:22 +00:00
126 lines
2.9 KiB
Python
126 lines
2.9 KiB
Python
# coding=utf-8
|
|
import sys
|
|
from client import JudgeClient
|
|
from language import languages
|
|
from compiler import compile_
|
|
from result import result
|
|
|
|
#
|
|
# c_src = r"""
|
|
# #include <stdio.h>
|
|
# #include </dev/random>
|
|
# int main()
|
|
# {
|
|
# FILE *fp;
|
|
# fp = NULL;
|
|
# fprintf(fp, "This is testing for fprintf...\n");
|
|
# fputs("This is testing for fputs...\n", fp);
|
|
# fclose(fp);
|
|
# printf("111111");
|
|
# return 0;
|
|
# }
|
|
# """
|
|
#
|
|
# cpp_src = r"""
|
|
# #include <iostream>
|
|
#
|
|
# using namespace std;
|
|
#
|
|
# int main()
|
|
# {
|
|
# int a,b;
|
|
# cin >> a >> b;
|
|
# cout << a+b;
|
|
# return 0;
|
|
# }
|
|
# """
|
|
#
|
|
# java_src = r"""
|
|
# import java.io.*;
|
|
# import java.util.*;
|
|
# 11
|
|
# public class Main
|
|
# {
|
|
# public static void main(String[] args)
|
|
# {
|
|
# Scanner in = new Scanner(System.in);
|
|
# PrintWriter out = new PrintWriter(System.out);
|
|
#
|
|
# int a = in.nextInt();
|
|
# int b = in.nextInt();
|
|
# out.print(a + b);
|
|
# throw new EmptyStackException();
|
|
#
|
|
# }
|
|
# }
|
|
# """
|
|
# def judge(language_code, source_string):
|
|
# language = languages[str(language_code)]
|
|
# src_path = judger_workspace + language["src_name"]
|
|
# f = open(src_path, "w")
|
|
# f.write(source_string)
|
|
# f.close()
|
|
#
|
|
# try:
|
|
# exe_path = compile_(languages[str(language_code)], src_path, judger_workspace)
|
|
# except Exception as e:
|
|
# print e
|
|
# return [{"result": result["compile_error"]}]
|
|
#
|
|
# client = JudgeClient(language_code=language_code,
|
|
# exe_path=exe_path,
|
|
# max_cpu_time=1000000,
|
|
# max_real_time=200000,
|
|
# max_memory=1000,
|
|
# test_case_dir="/var/test_cases/1/")
|
|
# return client.run()
|
|
#
|
|
# print judge(1, c_src)
|
|
# print judge(2, cpp_src)
|
|
# print judge(3, java_src)
|
|
|
|
judger_workspace = "/var/judger/"
|
|
# 简单的解析命令行参数
|
|
# 参数有 -solution_id -max_cpu_time -max_memory -test_case_id
|
|
# 获取到的值是['xxx.py', '-solution_id', '1111', '-max_cpu_time', '1000', '-max_memory', '100', '-test_case_id', 'aaaa']
|
|
args = sys.argv
|
|
solution_id = args[2]
|
|
max_cpu_time = args[4]
|
|
max_memory = args[6]
|
|
test_case_id = args[8]
|
|
|
|
# todo 去数据库查一下
|
|
language_code = 1
|
|
source_string = """
|
|
#include <stdio.h>
|
|
int main()
|
|
{
|
|
int a, b;
|
|
scanf("%d %d", &a, &b);
|
|
printf("%d", a + b);
|
|
return 0;
|
|
}
|
|
"""
|
|
language = languages[str(language_code)]
|
|
src_path = judger_workspace + language["src_name"]
|
|
f = open(src_path, "w")
|
|
f.write(source_string)
|
|
f.close()
|
|
|
|
try:
|
|
exe_path = compile_(languages[str(language_code)], src_path, judger_workspace)
|
|
except Exception as e:
|
|
print e
|
|
print [{"result": result["compile_error"]}]
|
|
exit()
|
|
|
|
client = JudgeClient(language_code=language_code,
|
|
exe_path=exe_path,
|
|
max_cpu_time=1000000,
|
|
max_real_time=200000,
|
|
max_memory=1000,
|
|
test_case_dir="/var/test_cases/1/")
|
|
print client.run()
|
|
|
|
|