from rest_framework import serializers
from .models import User
from divebases.models import DiveBase

class DiveBaseSerializer(serializers.ModelSerializer):
    class Meta:
        model = DiveBase
        fields = ['id', 'name', 'location']

class UserSerializer(serializers.ModelSerializer):
    dive_base = DiveBaseSerializer(read_only=True)
    dive_base_id = serializers.PrimaryKeyRelatedField(
        source='dive_base', queryset=DiveBase.objects.all(), write_only=True, required=False
    )

    class Meta:
        model = User
        fields = ['id', 'username', 'email', 'role', 'dive_certificate', 'dive_base', 'dive_base_id']
        read_only_fields = ['id']