load_clip

class LoadClip(MochaLoader):
    """Load a clip from a file."""

    label = "Load Clip"
    order = -10
    icon = "code-fork"
    color = "orange"

    product_types: ClassVar[set[str]] = {"*"}
    representations: ClassVar[set[str]] = {"*"}
    extensions: ClassVar[set[str]] = {
        ext.lstrip(".") for ext in IMAGE_EXTENSIONS}

    def load(self,
             context: dict,
             name: Optional[str] = None,
             namespace: Optional[str] = None,
             options: Optional[dict] = None) -> None:
        """Load a clip from a file."""
        host: MochaProHost = registered_host()
        project = host.get_current_project()
        with project.undo_group():
            file_path = self.filepath_from_context(context)

            # Check if the clip with the same name already exists
            # Mocha will load clips with the same name, but it will
            # show in the UI just one of them, and it makes things
            # confusing for the user.
            clips = project.get_clips()
            if name in clips:
                self.log.warning("Clip %s already exists", name)
                idx = 1
                while f"{name}_{idx}" in clips:
                    idx += 1
                name = f"{name}_{idx}"

            clip = Clip(file_path, name)
            project.add_clip(clip, name)
            project.new_output_clip(clip, name)

            container = Container(
                name=name,
                namespace=namespace or "",
                loader=self.__class__.__name__,
                representation=str(context["representation"]["id"]),
                objectName=clip.name,
                timestamp=time.time_ns()
            )
            host.add_container(container)
            update_ui()
            self.log.debug("Loaded clip: %s", clip)

    def switch(self, container: dict, context: dict) -> None:
        """Switch the image sequence on the current camera."""
        self.update(container, context)

    def remove(self, container: dict) -> None:
        """Remove a container."""
        host: MochaProHost = registered_host()
        project = host.get_current_project()

        clips = project.get_clips()
        clip = clips.get(container["objectName"])
        if not clip:
            self.log.warning("Clip %s not found", container["objectName"])
            return
        del clip
        host.remove_container(Container(**container))

    def update(self, container: dict, context: dict) -> None:
        """Update a container.

        Args:
            container (dict): Container to update.
            context (dict): Context to update the container to.

        Raises:
            LoadError: If the image info cannot be retrieved.

        """
        host: MochaProHost = registered_host()

        version_entity = context["version"]
        repre_entity = context["representation"]

        file_path = get_representation_path(repre_entity)
        project = host.get_current_project()
        clips = project.get_clips()

        # get image info using OIIO
        try:
            image_info = get_image_info(Path(file_path))
        except ValueError as e:
            msg = f"Failed to get image info for {file_path}: {e}"
            raise LoadError(msg) from e

        try:
            clips[container["objectName"]].relink(file_path)
            clips[container["objectName"]].frame_size = (
                image_info.get("width", 1920),
                image_info.get("height", 1080),
            )
        except KeyError:
            self.log.warning("Clip %s not found", container["objectName"])
        update_ui()

        container["representation"] = repre_entity["id"]
        container["version"] = str(version_entity["version"])
        host.add_container(Container(**container))