68 lines
1.3 KiB
Python
68 lines
1.3 KiB
Python
from typing import Literal, Union
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, RootModel
|
|
from typing_extensions import Annotated
|
|
|
|
|
|
|
|
class default_scene(BaseModel):
|
|
"""Visit the default scene."""
|
|
|
|
type: Literal["default_scene"] = "default_scene"
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
|
|
class scene_by_index(BaseModel):
|
|
"""Visit the indexed scene."""
|
|
|
|
index: int
|
|
|
|
type: Literal["scene_by_index"] = "scene_by_index"
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
|
|
class scene_by_name(BaseModel):
|
|
"""Visit the first scene with the given name."""
|
|
|
|
name: str
|
|
|
|
type: Literal["scene_by_name"] = "scene_by_name"
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
|
|
class mesh_by_index(BaseModel):
|
|
"""Visit the indexed mesh."""
|
|
|
|
index: int
|
|
|
|
type: Literal["mesh_by_index"] = "mesh_by_index"
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
|
|
class mesh_by_name(BaseModel):
|
|
"""Visit the first mesh with the given name."""
|
|
|
|
name: str
|
|
|
|
type: Literal["mesh_by_name"] = "mesh_by_name"
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
|
|
Selection = RootModel[
|
|
Annotated[
|
|
Union[
|
|
default_scene,
|
|
scene_by_index,
|
|
scene_by_name,
|
|
mesh_by_index,
|
|
mesh_by_name,
|
|
],
|
|
Field(discriminator="type"),
|
|
]
|
|
]
|