__init__.py

"""
Sample Inspector Plugin - Python operators
"""

import fiftyone.operators as foo
import fiftyone.operators.types as types


class GetMaskTargets(foo.Operator):
    @property
    def config(self):
        return foo.OperatorConfig(
            name="get_mask_targets",
            label="Get mask targets from dataset",
            unlisted=True,
        )

    def resolve_input(self, ctx):
        inputs = types.Object()
        return types.Property(inputs)

    def execute(self, ctx):
        dataset = ctx.dataset

        mask_targets = {}
        default_mask_targets = {}

        if hasattr(dataset, 'mask_targets') and dataset.mask_targets:
            mask_targets = dataset.mask_targets

        if hasattr(dataset, 'default_mask_targets') and dataset.default_mask_targets:
            default_mask_targets = dataset.default_mask_targets

        return {
            "mask_targets": mask_targets,
            "default_mask_targets": default_mask_targets,
        }


def register(plugin):
    plugin.register(GetMaskTargets)