from rest_framework import serializers

from finance.models import Bill, Invoice


class BillSerializer(serializers.ModelSerializer):
    opener = serializers.SerializerMethodField()
    user = serializers.SerializerMethodField()

    class Meta:
        model = Bill
        fields = '__all__'

    @staticmethod
    def get_user(self):
        return self.user.email

    @staticmethod
    def get_opener(self):
        return self.opener.email


class InvoiceSerializer(serializers.ModelSerializer):

    class Meta:
        model = Invoice
        exclude = ('bill', )