create_workfile

class CreateWorkfile(AutoCreator):
    """Workfile auto-creator."""
    settings_category = "hiero"

    identifier = "io.ayon.creators.hiero.workfile"
    label = "Workfile"
    product_base_type = "workfile"
    product_type = product_base_type
    icon = "fa5.file"

    default_variant = "Main"

    @classmethod
    def dump_instance_data(cls, data):
        """ Dump instance data into AyonData project tag.

        Args:
            data (dict): The data to push to the project tag.
        """
        project_tag = tags.get_or_create_workfile_tag(
            constants.AYON_WORKFILE_TAG_NAME,
            create=True
        )

        tag_data = {
            "metadata": data,
            "note": "AYON workfile data"
        }
        tags.update_tag(project_tag, tag_data)

    def load_instance_data(cls):
        """ Returns the data stored in AyonData project tag if any.

        Returns:
            dict. The project data.
        """
        project_tag = tags.get_or_create_workfile_tag(
            constants.AYON_WORKFILE_TAG_NAME
        )
        if project_tag is None:
            return {}

        instance_data = tags.get_tag_data(project_tag)
        return instance_data

    def _create_new_instance(self):
        """Create a new workfile instance.

        Returns:
            dict. The data of the instance to be created.
        """

        project_entity = self.create_context.get_current_project_entity()
        folder_entity = self.create_context.get_current_folder_entity()
        task_entity = self.create_context.get_current_task_entity()

        project_name = project_entity["name"]
        folder_path = folder_entity["path"]
        task_name = task_entity["name"]
        host_name = self.create_context.host_name
        variant = self.default_variant

        product_name = self.get_product_name(
            project_name=project_name,
            project_entity=project_entity,
            folder_entity=folder_entity,
            task_entity=task_entity,
            variant=variant,
            host_name=host_name,
            product_type=self.product_type,
        )

        instance_data = {
            "folderPath": folder_path,
            "task": task_name,
            "variant": variant,
            "productName": product_name,
        }

        return instance_data

    def create(self):
        """Auto-create an instance by default."""
        instance_data = self.load_instance_data()
        if instance_data:
            return

        self.log.info("Auto-creating workfile instance...")
        data = self._create_new_instance()
        current_instance = CreatedInstance(
            product_base_type=self.product_base_type,
            product_type=self.product_type,
            product_name=data["productName"],
            data=data,
            creator=self,
        )
        self._add_instance_to_context(current_instance)

    def collect_instances(self):
        """Collect from timeline marker or create a new one."""
        data = self.load_instance_data()
        if not data:
            return

        product_type = data.get("productType")
        instance = CreatedInstance(
            product_base_type=self.product_base_type,
            product_type=product_type or self.product_base_type,
            product_name=data["productName"],
            data=data,
            creator=self,
        )
        self._add_instance_to_context(instance)

    def update_instances(self, update_list):
        """Store changes in project metadata so they can be recollected.

        Args:
            update_list(List[UpdateData]): Gets list of tuples. Each item
                contain changed instance and its changes.
        """
        for created_inst, _ in update_list:
            data = created_inst.data_to_store()
            self.dump_instance_data(data)