101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
|
"""
|
|||
|
|
StarSystem Pydantic Models
|
|||
|
|
恒星系统数据模型（用于API）
|
|||
|
|
"""
|
|||
|
|
from typing import Optional, List
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemBase(BaseModel):
|
|||
|
|
"""恒星系统基础模型"""
|
|||
|
|
name: str = Field(..., description="恒星系名称")
|
|||
|
|
name_zh: Optional[str] = Field(None, description="中文名称")
|
|||
|
|
host_star_name: str = Field(..., description="主恒星名称")
|
|||
|
|
|
|||
|
|
# 位置信息
|
|||
|
|
distance_pc: Optional[float] = Field(None, description="距离地球（秒差距）")
|
|||
|
|
distance_ly: Optional[float] = Field(None, description="距离地球（光年）")
|
|||
|
|
ra: Optional[float] = Field(None, description="赤经（度）")
|
|||
|
|
dec: Optional[float] = Field(None, description="赤纬（度）")
|
|||
|
|
position_x: Optional[float] = Field(None, description="笛卡尔坐标 X（pc）")
|
|||
|
|
position_y: Optional[float] = Field(None, description="笛卡尔坐标 Y（pc）")
|
|||
|
|
position_z: Optional[float] = Field(None, description="笛卡尔坐标 Z（pc）")
|
|||
|
|
|
|||
|
|
# 恒星参数
|
|||
|
|
spectral_type: Optional[str] = Field(None, description="光谱类型")
|
|||
|
|
radius_solar: Optional[float] = Field(None, description="恒星半径（太阳半径）")
|
|||
|
|
mass_solar: Optional[float] = Field(None, description="恒星质量（太阳质量）")
|
|||
|
|
temperature_k: Optional[float] = Field(None, description="表面温度（K）")
|
|||
|
|
magnitude: Optional[float] = Field(None, description="视星等")
|
|||
|
|
luminosity_solar: Optional[float] = Field(None, description="光度（太阳光度）")
|
|||
|
|
|
|||
|
|
# 显示属性
|
|||
|
|
color: Optional[str] = Field(None, description="显示颜色（HEX）")
|
|||
|
|
|
|||
|
|
# 描述
|
|||
|
|
description: Optional[str] = Field(None, description="描述")
|
|||
|
|
details: Optional[str] = Field(None, description="详细信息（Markdown）")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemCreate(StarSystemBase):
|
|||
|
|
"""创建恒星系统"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemUpdate(BaseModel):
|
|||
|
|
"""更新恒星系统（所有字段可选）"""
|
|||
|
|
name: Optional[str] = None
|
|||
|
|
name_zh: Optional[str] = None
|
|||
|
|
host_star_name: Optional[str] = None
|
|||
|
|
distance_pc: Optional[float] = None
|
|||
|
|
distance_ly: Optional[float] = None
|
|||
|
|
ra: Optional[float] = None
|
|||
|
|
dec: Optional[float] = None
|
|||
|
|
position_x: Optional[float] = None
|
|||
|
|
position_y: Optional[float] = None
|
|||
|
|
position_z: Optional[float] = None
|
|||
|
|
spectral_type: Optional[str] = None
|
|||
|
|
radius_solar: Optional[float] = None
|
|||
|
|
mass_solar: Optional[float] = None
|
|||
|
|
temperature_k: Optional[float] = None
|
|||
|
|
magnitude: Optional[float] = None
|
|||
|
|
luminosity_solar: Optional[float] = None
|
|||
|
|
color: Optional[str] = None
|
|||
|
|
description: Optional[str] = None
|
|||
|
|
details: Optional[str] = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemResponse(StarSystemBase):
|
|||
|
|
"""恒星系统响应模型"""
|
|||
|
|
id: int
|
|||
|
|
planet_count: int = Field(default=0, description="已知行星数量")
|
|||
|
|
star_count: int = Field(default=1, description="恒星数量（包括主星和伴星）")
|
|||
|
|
created_at: datetime
|
|||
|
|
updated_at: datetime
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
from_attributes = True
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemWithBodies(StarSystemResponse):
|
|||
|
|
"""包含天体的恒星系统"""
|
|||
|
|
bodies: List[dict] = Field(default_factory=list, description="关联的天体列表")
|
|||
|
|
body_count: int = Field(default=0, description="天体数量")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemListResponse(BaseModel):
|
|||
|
|
"""恒星系统列表响应"""
|
|||
|
|
total: int
|
|||
|
|
systems: List[StarSystemResponse]
|
|||
|
|
|
|||
|
|
|
|||
|
|
class StarSystemStatistics(BaseModel):
|
|||
|
|
"""恒星系统统计信息"""
|
|||
|
|
total_systems: int = Field(..., description="总恒星系统数")
|
|||
|
|
exo_systems: int = Field(..., description="系外恒星系统数")
|
|||
|
|
total_planets: int = Field(..., description="总行星数")
|
|||
|
|
exo_planets: int = Field(..., description="系外行星数")
|
|||
|
|
solar_system_planets: int = Field(..., description="太阳系行星数")
|
|||
|
|
nearest_systems: List[dict] = Field(default_factory=list, description="最近的10个恒星系统")
|