from rest_framework import serializers
from .models import *


class ConsumableGoodsInboundSerializer(serializers.ModelSerializer):
    category = serializers.PrimaryKeyRelatedField(queryset=ConsumableGoodsCategory.objects.all())
    category_name = serializers.SerializerMethodField()
    inventory = serializers.PrimaryKeyRelatedField(queryset=ConsumableGoodsInventory.objects.all())

    class Meta:
        model = ConsumableGoodsInboundManagement
        fields = '__all__'

    def get_category_name(self, obj):
        return obj.category.category_name


class ConsumableGoodsOutboundSerializer(serializers.ModelSerializer):
    category = serializers.PrimaryKeyRelatedField(queryset=ConsumableGoodsCategory.objects.all())
    category_name = serializers.SerializerMethodField()
    inventory = serializers.PrimaryKeyRelatedField(queryset=ConsumableGoodsInventory.objects.all())

    class Meta:
        model = ConsumableGoodsOutboundRecords
        fields = '__all__'

    def get_category_name(self, obj):
        return obj.category.category_name


class StoredValueCardUseRecordSerializer(serializers.ModelSerializer):
    class Meta:
        model = StoredValueCardUseRecord
        fields = '__all__'


class MembershipModificationRecordSerializer(serializers.ModelSerializer):
    class Meta:
        model = MembershipModificationRecord
        fields = '__all__'