37 lines
887 B
Python
37 lines
887 B
Python
from rest_framework import serializers
|
|
|
|
from tmc.models import Territory
|
|
from tmc.serializers import TerritorySerializer
|
|
from .models import TgItem, TmcField
|
|
import logging
|
|
|
|
logger = logging.getLogger("root")
|
|
|
|
|
|
class TgItemSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = TgItem
|
|
fields = "__all__"
|
|
depth = 3
|
|
|
|
|
|
class TgStatItemSerializer(serializers.Serializer):
|
|
# location = serializers.CharField()
|
|
location = serializers.SerializerMethodField()
|
|
inv_count = serializers.IntegerField()
|
|
|
|
def get_location(self, obj):
|
|
queryset = Territory.objects.get(id=obj.get('location__parent'))
|
|
serializer = TerritorySerializer(queryset)
|
|
|
|
return serializer.data
|
|
|
|
class Meta:
|
|
depth = 2
|
|
|
|
|
|
class TmcFieldSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = TmcField
|
|
fields = "__all__"
|