from rest_framework import serializers

from .models import BillingAddress, Country


class BillingAddressSerializer(serializers.ModelSerializer):
    class Meta:
        model = BillingAddress
        fields = [
            "first_name",
            "last_name",
            "street",
            "street_number",
            "postal_code",
            "city",
            "country",
            "company_name",
            "company_street",
            "company_street_number",
            "company_postal_code",
            "company_city",
            "company_country",
        ]


class CountrySerializer(serializers.ModelSerializer):
    class Meta:
        model = Country
        fields = "__all__"
