zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/application/serializers/application_serializers.py
View File

@ -684,7 +684,10 @@ class ApplicationSerializer(serializers.Serializer):
self.is_valid() self.is_valid()
user_id = self.data.get('user_id') user_id = self.data.get('user_id')
mk_instance_bytes = self.data.get('file').read() 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 application = mk_instance.application
function_lib_list = mk_instance.function_lib_list function_lib_list = mk_instance.function_lib_list
if len(function_lib_list) > 0: if len(function_lib_list) > 0: