mirror of https://github.com/jumpserver/jumpserver
116 lines
4.1 KiB
Python
116 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
#
|
||
|
||
from django.core.exceptions import ObjectDoesNotExist
|
||
from rest_framework.utils import html
|
||
from rest_framework.settings import api_settings
|
||
from rest_framework.exceptions import ValidationError
|
||
from rest_framework.fields import SkipField, empty
|
||
|
||
__all__ = ['BulkSerializerMixin', 'BulkListSerializerMixin']
|
||
|
||
|
||
class BulkSerializerMixin(object):
|
||
"""
|
||
Become rest_framework_bulk not support uuid as a primary key
|
||
so rewrite it. https://github.com/miki725/django-rest-framework-bulk/issues/66
|
||
"""
|
||
def to_internal_value(self, data):
|
||
from rest_framework_bulk import BulkListSerializer
|
||
ret = super(BulkSerializerMixin, self).to_internal_value(data)
|
||
|
||
id_attr = getattr(self.Meta, 'update_lookup_field', 'id')
|
||
if self.context.get('view'):
|
||
request_method = getattr(getattr(self.context.get('view'), 'request'), 'method', '')
|
||
# add update_lookup_field field back to validated data
|
||
# since super by default strips out read-only fields
|
||
# hence id will no longer be present in validated_data
|
||
if all((isinstance(self.root, BulkListSerializer),
|
||
id_attr,
|
||
request_method in ('PUT', 'PATCH'))):
|
||
id_field = self.fields.get("id") or self.fields.get('pk')
|
||
if data.get("id"):
|
||
id_value = id_field.to_internal_value(data.get("id"))
|
||
else:
|
||
id_value = id_field.to_internal_value(data.get("pk"))
|
||
ret[id_attr] = id_value
|
||
return ret
|
||
|
||
def run_validation(self, data=empty):
|
||
"""
|
||
批量创建时，获取到的self.initial_data是list，
|
||
所以想用一个属性来存放当前操作的数据集，在validate_field中使用
|
||
:param data:
|
||
:return:
|
||
"""
|
||
# 只有批量创建的时候，才需要重写 initial_data
|
||
if self.parent:
|
||
self.initial_data = data
|
||
return super().run_validation(data)
|
||
|
||
|
||
class BulkListSerializerMixin(object):
|
||
"""
|
||
Become rest_framework_bulk doing bulk update raise Exception:
|
||
'QuerySet' object has no attribute 'pk' when doing bulk update
|
||
so rewrite it .
|
||
https://github.com/miki725/django-rest-framework-bulk/issues/68
|
||
"""
|
||
|
||
def to_internal_value(self, data):
|
||
"""
|
||
List of dicts of native values <- List of dicts of primitive datatypes.
|
||
"""
|
||
if not self.instance:
|
||
return super().to_internal_value(data)
|
||
|
||
if html.is_html_input(data):
|
||
data = html.parse_html_list(data)
|
||
|
||
if not isinstance(data, list):
|
||
message = self.error_messages['not_a_list'].format(
|
||
input_type=type(data).__name__
|
||
)
|
||
raise ValidationError({
|
||
api_settings.NON_FIELD_ERRORS_KEY: [message]
|
||
}, code='not_a_list')
|
||
|
||
if not self.allow_empty and len(data) == 0:
|
||
if self.parent and self.partial:
|
||
raise SkipField()
|
||
|
||
message = self.error_messages['empty']
|
||
raise ValidationError({
|
||
api_settings.NON_FIELD_ERRORS_KEY: [message]
|
||
}, code='empty')
|
||
|
||
ret = []
|
||
errors = []
|
||
|
||
for item in data:
|
||
try:
|
||
# prepare child serializer to only handle one instance
|
||
if 'id' in item:
|
||
pk = item["id"]
|
||
elif 'pk' in item:
|
||
pk = item["pk"]
|
||
else:
|
||
raise ValidationError("id or pk not in data")
|
||
child = self.instance.get(id=pk) if self.instance else None
|
||
self.child.instance = child
|
||
self.child.initial_data = item
|
||
# raw
|
||
validated = self.child.run_validation(item)
|
||
except ValidationError as exc:
|
||
errors.append(exc.detail)
|
||
except ObjectDoesNotExist as e:
|
||
errors.append(e)
|
||
else:
|
||
ret.append(validated)
|
||
errors.append({})
|
||
|
||
if any(errors):
|
||
raise ValidationError(errors)
|
||
|
||
return ret
|