86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
from datetime import datetime
|
||
from enum import StrEnum
|
||
from typing import TYPE_CHECKING, Optional
|
||
from uuid import UUID
|
||
from sqlmodel import Field, Relationship, String
|
||
from pydantic_extra_types.semantic_version import SemanticVersion
|
||
|
||
from .base import SQLModelBase, UUIDTableBase
|
||
from .firmware import ChipTypeEnum
|
||
|
||
if TYPE_CHECKING:
|
||
from .user import User
|
||
|
||
class ItemTypeEnum(StrEnum):
|
||
normal = 'normal'
|
||
car = 'car'
|
||
esp32 = 'esp32'
|
||
|
||
class ItemStatusEnum(StrEnum):
|
||
ok = 'ok'
|
||
lost = 'lost'
|
||
|
||
class ItemBase(SQLModelBase):
|
||
type: ItemTypeEnum = ItemTypeEnum.normal
|
||
"""物品的类型"""
|
||
|
||
name: str
|
||
"""物品名称"""
|
||
|
||
icon: str | None = None
|
||
"""物品图标"""
|
||
|
||
status: ItemStatusEnum = ItemStatusEnum.ok
|
||
"""物品状态"""
|
||
|
||
phone: str | None = None
|
||
"""联系电话"""
|
||
|
||
description: str | None = None
|
||
"""物品描述"""
|
||
|
||
version: SemanticVersion = Field(sa_type=String(64))
|
||
"""版本号"""
|
||
|
||
chip_type: ChipTypeEnum | None = Field(default=None, index=True)
|
||
"""ESP设备芯片类型，仅当type=esp32时有值"""
|
||
|
||
class Item(ItemBase, UUIDTableBase, table=True):
|
||
expires_at: datetime | None = None
|
||
"""物品过期时间"""
|
||
|
||
lost_at: datetime | None = None
|
||
"""物品丢失的时间"""
|
||
|
||
find_ip: str | None = None
|
||
"""最后一次发现的IP地址"""
|
||
|
||
user_id: UUID = Field(foreign_key='user.id', ondelete='CASCADE')
|
||
"""所属用户ID"""
|
||
|
||
user: 'User' = Relationship(back_populates='items')
|
||
|
||
parent_item_id: UUID | None = Field(foreign_key='item.id', ondelete='RESTRICT', default=None)
|
||
parent_item: Optional['Item'] = Relationship(back_populates='sub_items', sa_relationship_kwargs={'remote_side': 'Item.id'})
|
||
sub_items: list['Item'] = Relationship(back_populates='parent_item', passive_deletes='all')
|
||
|
||
class ItemDataUpdateRequest(ItemBase):
|
||
pass
|
||
|
||
class ItemDataResponse(ItemBase):
|
||
expires_at: datetime | None = None
|
||
"""物品过期时间"""
|
||
|
||
lost_at: datetime | None = None
|
||
"""物品丢失的时间"""
|
||
|
||
class ItemDataResponseAdmin(ItemBase):
|
||
expires_at: datetime | None = None
|
||
"""物品过期时间"""
|
||
|
||
lost_at: datetime | None = None
|
||
"""物品丢失的时间"""
|
||
|
||
user_id: UUID = Field(foreign_key='user.id')
|
||
"""所属用户ID"""
|