浏览代码

Update app.py

haha 3 年之前
父节点
当前提交
3f52d3396a
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      web/app.py

+ 4 - 0
web/app.py

@@ -1,6 +1,8 @@
 # encoding: utf8
 # encoding: utf8
 import os
 import os
 import shutil
 import shutil
+from datetime import time
+
 from flask import Flask, request, json
 from flask import Flask, request, json
 import judgement
 import judgement
 
 
@@ -23,6 +25,7 @@ inputOutputMapping: hashmap 输入输出对
 
 
 @app.route('/api/judge', methods=['POST'])
 @app.route('/api/judge', methods=['POST'])
 def judge():
 def judge():
+    start = time()
     data = json.loads(request.get_data(as_text=True))
     data = json.loads(request.get_data(as_text=True))
     code = data['code']
     code = data['code']
     language = data['language']
     language = data['language']
@@ -39,6 +42,7 @@ def judge():
     rst = judgement.judge(src_file, 'testcases', td_total, time_limit, memory_limit, language)
     rst = judgement.judge(src_file, 'testcases', td_total, time_limit, memory_limit, language)
     os.remove(src_file)
     os.remove(src_file)
     shutil.rmtree('testcases')
     shutil.rmtree('testcases')
+    print(time()-start)
     return rst
     return rst