72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django.db.models import Q
|
|
from rest_framework import serializers
|
|
|
|
from dvadmin.system.models import Area
|
|
from dvadmin.utils.json_response import SuccessResponse
|
|
from dvadmin.utils.serializers import CustomModelSerializer
|
|
from dvadmin.utils.viewset import CustomModelViewSet
|
|
|
|
|
|
class AreaSerializer(CustomModelSerializer):
|
|
"""
|
|
地区-序列化器
|
|
"""
|
|
pcode_count = serializers.SerializerMethodField(read_only=True)
|
|
hasChild = serializers.SerializerMethodField()
|
|
def get_pcode_count(self, instance: Area):
|
|
return Area.objects.filter(pcode=instance).count()
|
|
def get_hasChild(self, instance):
|
|
hasChild = Area.objects.filter(pcode=instance.code)
|
|
if hasChild:
|
|
return True
|
|
return False
|
|
class Meta:
|
|
model = Area
|
|
fields = "__all__"
|
|
read_only_fields = ["id"]
|
|
|
|
|
|
class AreaCreateUpdateSerializer(CustomModelSerializer):
|
|
"""
|
|
地区管理 创建/更新时的列化器
|
|
"""
|
|
|
|
class Meta:
|
|
model = Area
|
|
fields = '__all__'
|
|
|
|
|
|
class AreaViewSet(CustomModelViewSet):
|
|
"""
|
|
地区管理接口
|
|
list:查询
|
|
create:新增
|
|
update:修改
|
|
retrieve:单例
|
|
destroy:删除
|
|
"""
|
|
queryset = Area.objects.all()
|
|
serializer_class = AreaSerializer
|
|
extra_filter_class = []
|
|
|
|
def get_queryset(self):
|
|
self.request.query_params._mutable = True
|
|
params = self.request.query_params
|
|
pcode = params.get('pcode', None)
|
|
page = params.get('page', None)
|
|
limit = params.get('limit', None)
|
|
if page:
|
|
del params['page']
|
|
if limit:
|
|
del params['limit']
|
|
if params:
|
|
if pcode:
|
|
queryset = self.queryset.filter(enable=True, pcode=pcode)
|
|
else:
|
|
queryset = self.queryset.filter(enable=True)
|
|
else:
|
|
queryset = self.queryset.filter(enable=True, pcode__isnull=True)
|
|
return queryset
|
|
|