用户上传测试用例文件过大时引发memoryERROR

This commit is contained in:
sxw@401
2015-09-05 16:47:59 +08:00
parent c689888bbd
commit 7cf1df266e

View File

@@ -185,8 +185,12 @@ class TestCaseUploadAPIView(APIView):
os.mkdir(test_case_dir) os.mkdir(test_case_dir)
for name in l: for name in l:
f = open(test_case_dir + name, "wb") f = open(test_case_dir + name, "wb")
f.write(test_case_file.read(name).replace("\r\n", "\n")) try:
f.close() f.write(test_case_file.read(name).replace("\r\n", "\n"))
except MemoryError:
return error_response(u"单个测试数据体积过大!")
finally:
f.close()
l.sort() l.sort()
file_info = {"test_case_number": len(l) / 2, "test_cases": {}} file_info = {"test_case_number": len(l) / 2, "test_cases": {}}