dify/api/fields/member_fields.py

from __future__ import annotations

from datetime import datetime

from flask_restx import fields
from pydantic import BaseModel, ConfigDict, computed_field, field_validator

from core.file import helpers as file_helpers

simple_account_fields = {
    "id": fields.String,
    "name": fields.String,
    "email": fields.String,
}


def _to_timestamp(value: datetime | int | None) -> int | None:
    if isinstance(value, datetime):
        return int(value.timestamp())
    return value


def _build_avatar_url(avatar: str | None) -> str | None:
    if avatar is None:
        return None
    if avatar.startswith(("http://", "https://")):
        return avatar
    return file_helpers.get_signed_file_url(avatar)


class ResponseModel(BaseModel):
    model_config = ConfigDict(
        from_attributes=True,
        extra="ignore",
        populate_by_name=True,
        serialize_by_alias=True,
        protected_namespaces=(),
    )


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


class _AccountAvatar(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 Account(_AccountAvatar):
    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

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


class AccountWithRole(_AccountAvatar):
    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
    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 AccountWithRoleList(ResponseModel):
    accounts: list[AccountWithRole]