74 lines
3.0 KiB
Python
74 lines
3.0 KiB
Python
from rest_framework import serializers
|
|
from .models import Company, Consumer, PaySubject, ConsumerPerm, ConsumerRole
|
|
from question.models import Question, Questioncat
|
|
from .permission import get_consumerperm_list
|
|
from rbac.serializers.user_serializer import UserListSerializer
|
|
|
|
|
|
class ConsumerPermSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = ConsumerPerm
|
|
fields = '__all__'
|
|
|
|
class ConsumerRoleSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = ConsumerRole
|
|
fields = '__all__'
|
|
|
|
class CompanySerializer(serializers.ModelSerializer):
|
|
"""
|
|
公司序列化
|
|
"""
|
|
create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
|
|
update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
|
|
create_admin_name = serializers.StringRelatedField(source='create_admin', read_only=True)
|
|
class Meta:
|
|
model = Company
|
|
fields = '__all__'
|
|
|
|
|
|
class ConsumerSerializer(serializers.ModelSerializer):
|
|
"""
|
|
客户序列化
|
|
"""
|
|
create_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
|
|
update_time = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
|
|
company_name = serializers.StringRelatedField(source='company', read_only=True)
|
|
company_ = CompanySerializer(source='company', read_only=True)
|
|
subjects_name = serializers.StringRelatedField(source='subjects', many=True, read_only=True)
|
|
workscope_name = serializers.StringRelatedField(source='workscope', read_only=True)
|
|
role_name = serializers.StringRelatedField(source='role', read_only=True)
|
|
create_admin_ = UserListSerializer(source='create_admin', read_only=True)
|
|
class Meta:
|
|
model = Consumer
|
|
exclude = ('avatar','collects', 'process')
|
|
|
|
@staticmethod
|
|
def setup_eager_loading(queryset):
|
|
""" Perform necessary eager loading of data. """
|
|
queryset = queryset.select_related('company','workscope', 'role', 'create_admin')
|
|
queryset = queryset.prefetch_related('subjects',)
|
|
return queryset
|
|
|
|
class ConsumerDetailSerializer(serializers.ModelSerializer):
|
|
"""
|
|
客户详情序列化
|
|
"""
|
|
company_name = serializers.StringRelatedField(source='company', read_only=True)
|
|
subjects_name = serializers.StringRelatedField(source='subjects', many=True, read_only=True)
|
|
workscope_name = serializers.StringRelatedField(source='workscope', read_only=True)
|
|
role_name = serializers.StringRelatedField(source='role', read_only=True)
|
|
perms = serializers.SerializerMethodField(read_only=True)
|
|
class Meta:
|
|
model = Consumer
|
|
exclude = ('avatar','collects')
|
|
|
|
@staticmethod
|
|
def setup_eager_loading(queryset):
|
|
""" Perform necessary eager loading of data. """
|
|
queryset = queryset.select_related('company','workscope', 'role')
|
|
queryset = queryset.prefetch_related('subjects',)
|
|
return queryset
|
|
|
|
def get_perms(self,obj):
|
|
return get_consumerperm_list(obj) |