fix: Import application and add file format verification (#1928)

This commit is contained in:
shaohuzhang1 2024-12-27 11:13:44 +08:00 committed by GitHub
parent 441fb1a0b7
commit 415b2e1967
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -684,7 +684,10 @@ class ApplicationSerializer(serializers.Serializer):
self.is_valid()
user_id = self.data.get('user_id')
mk_instance_bytes = self.data.get('file').read()
mk_instance = pickle.loads(mk_instance_bytes)
try:
mk_instance = pickle.loads(mk_instance_bytes)
except Exception as e:
raise AppApiException(1001, "不支持的文件格式")
application = mk_instance.application
function_lib_list = mk_instance.function_lib_list
if len(function_lib_list) > 0: