from rest_framework import routers, serializers, viewsets
from .models import Product, Floorplan


class ProductSerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = Product
        fields = [
            "id",
            "title",
            "description",
            "model3d",
            "image1",
            "image2",
            "image3",
        ]

    
class FloorplanSerializer(serializers.Serializer):
    title = serializers.CharField(max_length=200)
    np_field = serializers.CharField()
    
    def create(self,validated_data):
        return Floorplan.objects.create(**validated_data)