workflow.py

import supervisely as sly


def check_compatibility(func):
    def wrapper(self, *args, **kwargs):
        if self.is_compatible is None:
            self.is_compatible = self.check_instance_ver_compatibility()
        if not self.is_compatible:
            return
        return func(self, *args, **kwargs)

    return wrapper


class Workflow:
    def __init__(self, api: sly.Api, min_instance_version: str = None):
        self.is_compatible = None
        self.api = api
        self._min_instance_version = (
            "6.9.31" if min_instance_version is None else min_instance_version
        )

    def check_instance_ver_compatibility(self):
        if not self.api.is_version_supported(self._min_instance_version):
            sly.logger.info(
                f"Supervisely instance version {self.api.instance_version} does not support workflow features."
            )
            if not sly.is_community():
                sly.logger.info(
                    f"To use them, please update your instance to version {self._min_instance_version} or higher."
                )
            return False
        return True

    @check_compatibility
    def add_input(self, project_id: int = None, session_id: int = None):
        if session_id is not None:
            sly.logger.debug(f"Workflow Input: Session ID - {session_id}")
            self.api.app.workflow.add_input_task(int(session_id))
        if project_id is not None:
            sly.logger.debug(f"Workflow Input: Project ID - {project_id}")
            self.api.app.workflow.add_input_project(project_id)

    @check_compatibility
    def add_output(self, project_id: int):
        sly.logger.debug(f"Workflow Output: Project ID - {project_id}")
        self.api.app.workflow.add_output_project(project_id)