47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from apps.utils.serializers import CustomModelSerializer
|
|
from apps.mtm.models import Shift, Material, Mgroup, Team, Goal
|
|
from apps.utils.constants import EXCLUDE_FIELDS
|
|
from rest_framework import serializers
|
|
from apps.system.models import Dept
|
|
|
|
class ShiftSerializer(CustomModelSerializer):
|
|
class Meta:
|
|
model = Shift
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
class MaterialSerializer(CustomModelSerializer):
|
|
class Meta:
|
|
model = Material
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
class MgroupSerializer(CustomModelSerializer):
|
|
belong_dept = serializers.PrimaryKeyRelatedField(label="所属部门", queryset=Dept.objects.all(), required=True)
|
|
belong_dept_name = serializers.CharField(source='belong_dept.name', read_only=True)
|
|
class Meta:
|
|
model = Mgroup
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
class TeamSerializer(CustomModelSerializer):
|
|
belong_dept = serializers.PrimaryKeyRelatedField(label="所属部门", queryset=Dept.objects.all(), required=True)
|
|
leader_name = serializers.CharField(source='leader.name', read_only=True)
|
|
belong_dept_name = serializers.CharField(source='belong_dept.name', read_only=True)
|
|
class Meta:
|
|
model = Team
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
|
|
class GoalSerializer(CustomModelSerializer):
|
|
goal_cate_name = serializers.CharField(source='goal_cate.name', read_only=True)
|
|
mgroup_name = serializers.CharField(source='mgroup.name', read_only=True)
|
|
class Meta:
|
|
model = Goal
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS
|
|
|
|
class MgroupGoalYearSerializer(serializers.Serializer):
|
|
mgroup = serializers.CharField(label='ID或名称')
|
|
year = serializers.IntegerField(label='年') |