with open() as 语句添加try&except，返回上传错误

2015-09-12 21:25:36 +08:00
parent 196c3ee934
commit a084614af9
2 changed files with 16 additions and 7 deletions
--- a/problem/views.py
+++ b/problem/views.py
@@ -147,9 +147,12 @@ class TestCaseUploadAPIView(APIView):
        f = request.FILES["file"]

        tmp_zip = "/tmp/" + rand_str() + ".zip"
-        with open(tmp_zip, "wb") as test_case_zip:
-            for chunk in f:
-                test_case_zip.write(chunk)
+        try:
+            with open(tmp_zip, "wb") as test_case_zip:
+                for chunk in f:
+                    test_case_zip.write(chunk)
+        except IOError:
+            return error_response(u"上传错误，写入临时目录失败")

        test_case_file = zipfile.ZipFile(tmp_zip, 'r')
        name_list = test_case_file.namelist()