from rest_framework.serializers import ModelSerializer

from .models import Location


class NestedLocationSerializer(ModelSerializer):
    class Meta:
        model = Location
        
        fields = (
            'id',
            'name',
            'children',
            'parent',
            'type',
            'lft',
            'rght',
            'level',
            'tree_id',
        )


class LocationSerializer(ModelSerializer):
    children = NestedLocationSerializer(many=True)
    parent = NestedLocationSerializer()
    
    class Meta:
        model = Location

        fields = (
            'id',
            'name',
            'children',
            'parent',
            'type',
            'lft',
            'rght',
            'level',
            'tree_id',
        )