infrasynth-backend-kit/infrasynth/billing/serializers.py

from rest_framework import serializers

from .models import (
    BillingPlan,
    Invoice,
    PaymentGateway,
    PaymentTransaction,
    Subscription,
)


class PaymentGatewaySerializer(serializers.ModelSerializer):
    class Meta:
        model = PaymentGateway
        fields = [
            "slug",
            "display_name",
            "gateway_class",
            "config",
            "is_active",
            "supported_currencies",
            "webhook_secret",
        ]
        read_only_fields = ["slug"]


class BillingPlanSerializer(serializers.ModelSerializer):
    gateway_info = serializers.SerializerMethodField()

    class Meta:
        model = BillingPlan
        fields = [
            "id",
            "slug",
            "name",
            "description",
            "price_amount",
            "price_currency",
            "interval",
            "trial_days",
            "features",
            "is_active",
            "gateway",
            "gateway_info",
            "external_id",
        ]
        read_only_fields = ["id"]

    def get_gateway_info(self, obj):
        if obj.gateway_id:
            return {"slug": obj.gateway_id, "display_name": obj.gateway.display_name}
        return None


class SubscriptionSerializer(serializers.ModelSerializer):
    plan_info = serializers.SerializerMethodField()
    gateway_info = serializers.SerializerMethodField()
    user_info = serializers.SerializerMethodField()

    class Meta:
        model = Subscription
        fields = [
            "id",
            "user",
            "user_info",
            "plan",
            "plan_info",
            "gateway",
            "gateway_info",
            "external_id",
            "status",
            "current_period_start",
            "current_period_end",
            "cancel_at_period_end",
            "cancelled_at",
            "trial_end",
            "metadata",
        ]
        read_only_fields = ["id"]

    def get_plan_info(self, obj):
        if obj.plan_id:
            return {"id": obj.plan_id, "name": obj.plan.name, "slug": obj.plan.slug}
        return None

    def get_gateway_info(self, obj):
        if obj.gateway_id:
            return {"slug": obj.gateway_id, "display_name": obj.gateway.display_name}
        return None

    def get_user_info(self, obj):
        return {"id": obj.user_id, "display": str(obj.user)}


class InvoiceSerializer(serializers.ModelSerializer):
    subscription_info = serializers.SerializerMethodField()
    gateway_info = serializers.SerializerMethodField()
    user_info = serializers.SerializerMethodField()
    pdf_file_info = serializers.SerializerMethodField()

    class Meta:
        model = Invoice
        fields = [
            "id",
            "subscription",
            "subscription_info",
            "user",
            "user_info",
            "gateway",
            "gateway_info",
            "external_id",
            "invoice_number",
            "amount",
            "currency",
            "tax_amount",
            "tax_name",
            "status",
            "due_date",
            "paid_at",
            "line_items",
            "pdf_file",
            "pdf_file_info",
            "metadata",
        ]
        read_only_fields = ["id"]

    def get_subscription_info(self, obj):
        if obj.subscription_id:
            return {"id": obj.subscription_id}
        return None

    def get_gateway_info(self, obj):
        if obj.gateway_id:
            return {"slug": obj.gateway_id, "display_name": obj.gateway.display_name}
        return None

    def get_user_info(self, obj):
        return {"id": obj.user_id, "display": str(obj.user)}

    def get_pdf_file_info(self, obj):
        if obj.pdf_file_id:
            return {"id": obj.pdf_file_id, "filename": obj.pdf_file.original_filename}
        return None


class PaymentTransactionSerializer(serializers.ModelSerializer):
    invoice_info = serializers.SerializerMethodField()
    gateway_info = serializers.SerializerMethodField()

    class Meta:
        model = PaymentTransaction
        fields = [
            "id",
            "invoice",
            "invoice_info",
            "gateway",
            "gateway_info",
            "external_id",
            "amount",
            "currency",
            "status",
            "payment_method",
            "metadata",
            "created_at",
        ]
        read_only_fields = ["id", "created_at"]

    def get_invoice_info(self, obj):
        if obj.invoice_id:
            return {
                "id": obj.invoice_id,
                "invoice_number": obj.invoice.invoice_number,
            }
        return None

    def get_gateway_info(self, obj):
        if obj.gateway_id:
            return {"slug": obj.gateway_id, "display_name": obj.gateway.display_name}
        return None


class SubscribeSerializer(serializers.Serializer):
    plan_slug = serializers.SlugField()
    success_url = serializers.URLField(required=False)
    cancel_url = serializers.URLField(required=False)