from rest_framework import serializers
from .models import CustomUser, OTP


class OTPSerializer(serializers.ModelSerializer):
    class Meta:
        model = OTP
        fields = "__all__"  # Include all fields from the OTP model


class CustomUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = CustomUser
        fields = (
            "id",
            "mobile_number",
            "first_name",
            "last_name",
            "email",
            "age",
            "gender",
            "employer",
            "license_number",
            "is_verified",
            "profile_pic"
        )