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

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

View File

@@ -185,7 +185,11 @@ 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")
try:
f.write(test_case_file.read(name).replace("\r\n", "\n")) f.write(test_case_file.read(name).replace("\r\n", "\n"))
except MemoryError:
return error_response(u"单个测试数据体积过大!")
finally:
f.close() f.close()
l.sort() l.sort()