尝试一下判题

judge/controller/tasks.py

@@ -1,8 +1,11 @@
 # coding=utf-8
 from __future__ import absolute_import
 from judge.controller.celery import app
+import subprocess32 as subprocess
+
+subprocess.call("ping baidu.com", timeout=5)
 
 
 @app.task
 def judge(source_code, language, test_case_id):
-    print source_code, language, test_case_id
+    pass

judge/judger/client.py

@@ -170,79 +170,4 @@ class JudgeClient(object):
         # http://stackoverflow.com/questions/25382455/python-notimplementederror-pool-objects-cannot-be-passed-between-processes
         self_dict = self.__dict__.copy()
         del self_dict['_pool']
-        return self_dict
-
-
-
-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(languege_code, source_string):
-    language = languages[str(languege_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(languege_code)], src_path, judger_workspace)
-    except Exception as e:
-        print e
-        return [{"result": result["compile_error"]}]
-
-    client = JudgeClient(language_code=languege_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)
+        return self_dict

judge/judger/run.py

@@ -0,0 +1,125 @@
+# 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()
+
+

requirements.txt

@@ -6,4 +6,5 @@ djangorestframework
 django-rest-swagger
 celery
 gunicorn
-coverage
+coverage
+subprocess32