dify/api/fields/member_fields.py

from __future__ import annotations

from datetime import datetime

from pydantic import Field, computed_field, field_validator

from fields.base import ResponseModel
from libs.helper import build_avatar_url, to_timestamp


class SimpleAccountResponse(ResponseModel):
    id: str
    name: str
    email: str


class _AccountAvatarResponseMixin(ResponseModel):
    avatar: str | None = None

    @computed_field(return_type=str | None)  # type: ignore[prop-decorator]
    @property
    def avatar_url(self) -> str | None:
        return build_avatar_url(self.avatar)


class AccountResponse(_AccountAvatarResponseMixin):
    id: str
    name: str
    email: str
    is_password_set: bool
    interface_language: str | None = None
    interface_theme: str | None = None
    timezone: str | None = None
    last_login_at: int | None = None
    last_login_ip: str | None = None
    created_at: int | None = None
    first_workspace_created_at: int | None = Field(
        default=None,
        description=(
            "First owned workspace creation time or migrated legacy Builder baseline; null if neither is recorded."
        ),
    )

    @field_validator("last_login_at", "created_at", "first_workspace_created_at", mode="before")
    @classmethod
    def _normalize_timestamp(cls, value: datetime | int | None) -> int | None:
        return to_timestamp(value)


class AccountProfileResponse(AccountResponse):
    has_owned_tenant: bool | None = Field(
        default=None,
        description="Whether the Account owns an existing tenant, including archived tenants. Null means unavailable.",
    )
    has_tenant: bool | None = Field(
        default=None,
        description="Whether the Account has a tenant membership, including ownership. Null means unavailable.",
    )


class MemberRoleSummaryResponse(ResponseModel):
    id: str
    name: str


class AccountWithRoleResponse(_AccountAvatarResponseMixin):
    id: str
    name: str
    email: str
    last_login_at: int | None = None
    last_active_at: int | None = None
    created_at: int | None = None
    role: str
    roles: list[MemberRoleSummaryResponse] = Field(default_factory=list)
    status: str

    @field_validator("last_login_at", "last_active_at", "created_at", mode="before")
    @classmethod
    def _normalize_timestamp(cls, value: datetime | int | None) -> int | None:
        return to_timestamp(value)


class AccountWithRoleListResponse(ResponseModel):
    accounts: list[AccountWithRoleResponse]


SimpleAccount = SimpleAccountResponse
Account = AccountResponse
AccountWithRole = AccountWithRoleResponse
AccountWithRoleList = AccountWithRoleListResponse