45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import strawberry
|
|
from typing import List, Optional
|
|
from datetime import datetime
|
|
from strawberry.types import Info
|
|
|
|
from app.utils.format import get_image_url
|
|
from app.graphql.context import CustomContext
|
|
|
|
@strawberry.type
|
|
class UserType:
|
|
id: strawberry.ID
|
|
email: str
|
|
name: Optional[str]
|
|
handle: Optional[str]
|
|
avatar: Optional[str]
|
|
following_count: int
|
|
follower_count: int
|
|
created_at: datetime
|
|
|
|
# 动态解析字段：例如自动处理头像的绝对路径
|
|
@strawberry.field
|
|
def avatar_url(self) -> Optional[str]:
|
|
return get_image_url(self.avatar) if self.avatar else None
|
|
|
|
@strawberry.type
|
|
class WorkspaceMemberType:
|
|
id: strawberry.ID
|
|
workspace_id: str
|
|
user_id: Optional[str]
|
|
role: str
|
|
status: str
|
|
|
|
@strawberry.type
|
|
class WorkspaceType:
|
|
id: strawberry.ID
|
|
name: str
|
|
slug: str
|
|
logo: Optional[str]
|
|
description: Optional[str]
|
|
storage_used: int
|
|
|
|
# 支持前端按需加载关联的成员
|
|
@strawberry.field
|
|
def members(self, info: Info) -> List[WorkspaceMemberType]:
|
|
return info.context.members_loader.load(self.id) |