Skip to content

Commit eb7724c

Browse files
committed
fix: Import application and add file format verification
1 parent 441fb1a commit eb7724c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

apps/application/serializers/application_serializers.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -684,7 +684,10 @@ def import_(self, with_valid=True):
684684
self.is_valid()
685685
user_id = self.data.get('user_id')
686686
mk_instance_bytes = self.data.get('file').read()
687-
mk_instance = pickle.loads(mk_instance_bytes)
687+
try:
688+
mk_instance = pickle.loads(mk_instance_bytes)
689+
except Exception as e:
690+
raise AppApiException(1001, "不支持的文件格式")
688691
application = mk_instance.application
689692
function_lib_list = mk_instance.function_lib_list
690693
if len(function_lib_list) > 0:

0 commit comments

Comments
 (0)