gitee
/
django-vue-admin
mirror of https://gitee.com/liqianglog/django-vue-admin
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

功能变化: 

1）修复数据导入报attname错误的问题
2）数据导入支持存在联合主键的情况
pull/78/head
猿小天 2022-11-03 23:52:31 +08:00
parent 7dd6065a29
commit 994e4d5134
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/dvadmin/utils/import_export_mixin.py
View File

@ -138,17 +138,20 @@ class ImportSerializerMixin:
queryset = self.filter_queryset(self.get_queryset()) queryset = self.filter_queryset(self.get_queryset())
# 获取多对多字段 # 获取多对多字段
m2m_fields = [ m2m_fields = [
ele.attname ele.name
for ele in queryset.model._meta.get_fields() for ele in queryset.model._meta.get_fields()
if hasattr(ele, "many_to_many") and ele.many_to_many == True if hasattr(ele, "many_to_many") and ele.many_to_many == True
] ]
data = import_to_data(request.data.get("url"), self.import_field_dict, m2m_fields) data = import_to_data(request.data.get("url"), self.import_field_dict, m2m_fields)
unique_list = [ unique_list = [
ele.attname for ele in queryset.model._meta.get_fields() if hasattr(ele, "unique") and ele.unique == True ele.name for ele in queryset.model._meta.get_fields() if hasattr(ele, "unique") and ele.unique == True
] ]
for ele in data: for ele in data:
# 获取 unique 字段 # 获取 unique 字段
filter_dic = {i: ele.get(i) for i in list(set(self.import_field_dict.keys()) & set(unique_list))} if queryset.model._meta.unique_together: # 判断是否存在联合主键
filter_dic = {i: ele.get(i) for i in list(queryset.model._meta.unique_together[0])}
else:
filter_dic = {i: ele.get(i) for i in list(set(self.import_field_dict.keys()) & set(unique_list))}
instance = filter_dic and queryset.filter(**filter_dic).first() instance = filter_dic and queryset.filter(**filter_dic).first()
if instance and not updateSupport: if instance and not updateSupport:
continue continue