from rest_framework import serializers

from core.serializers.base import BaseModelSerializer
from apps.workspaces.models import Workspace, WorkspaceMembership


class WorkspaceSerializer(BaseModelSerializer):
    class Meta:
        model = Workspace
        fields = BaseModelSerializer.Meta.fields + (
            "name",
            "description",
        )


class WorkspaceMembershipSerializer(BaseModelSerializer):
    class Meta:
        model = WorkspaceMembership
        fields = BaseModelSerializer.Meta.fields + (
            "workspace",
            "user",
            "role",
            "is_active",
        )