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__"