from rest_framework import serializers

from .models import Sale, SaleLine


class SaleLineSerializer(serializers.ModelSerializer):
    class Meta:
        model = SaleLine
        fields = ['id', 'sale', 'product', 'unit_price', 'quantity']


class SaleSerializer(serializers.ModelSerializer):
    lines = SaleLineSerializer(many=True)

    class Meta:
        model = Sale
        fields = ['id', 'customer', 'date', 'lines']