import uuid
from typing import Any, List

from sqlalchemy import JSON
from sqlmodel import SQLModel, Field

from crud.config.db import engine


def get_uuid():
    return uuid.uuid1().hex

class PointScene(SQLModel, table=True):
    id: str = Field(default_factory=get_uuid, max_length=32, primary_key=True)
    scene_name: str = Field(max_length=256, description="事件名称", unique=True)
    calib_json: List[dict] = Field(sa_type=JSON, nullable=True, description="calib是从点云到图像的校准矩阵。它是可选的，但如果提供，该框将投影在图像上以帮助注释。")
    box_type: str = Field(default="psr", max_length=256, description="标注框类型")
    camera_ext: str = Field(default=".jpg", max_length=256, description="标注的图片类型（后缀）")
    lidar_ext: str = Field(default=".pcd", max_length=256)
    radar_ext: str = Field(default=".pcd", max_length=256)
    aux_lidar_ext: str = Field(default=".pcd", max_length=256)

    desc: List[dict] = Field(sa_type=JSON, nullable=True, description="calib是从点云到图像的校准矩阵。它是可选的，但如果提供，该框将投影在图像上以帮助注释。")

class PointSceneWorldItem(SQLModel, table=True):
    __tablename__ = "scene_world_item"
    id: str = Field(default_factory=get_uuid, max_length=32, primary_key=True)
    scene_id: str = Field(max_length=256, description="事件id")
    scene_name: str = Field(max_length=256, description="事件名")
    frame: str = Field(max_length=256, description="帧序号")
    label_json: List[dict] = Field(sa_type=JSON, nullable=True, description="标注信息 == annotation")
    ego_pose_json: List[dict] = Field(sa_type=JSON, nullable=True, description="")

    front_img_path: str = Field(max_length=512, description="minio对应文件的路径")
    left_img_path: str = Field(max_length=512, description="minio对应文件的路径")
    right_img_path: str = Field(max_length=512, description="minio对应文件的路径")
    lidar_pcd_path: str = Field(max_length=512, description="minio对应的pcd文件的路径")
# 初始化数据库表（异步执行）
SQLModel.metadata.create_all(engine)