favorites.py

from .core import *
from . import favorites_data

# -----------------------------------------------------------------------------
# Favorites panel
# -----------------------------------------------------------------------------

# Favorites must be an array of (node.bl_idname, node.bl_label) tuples
favorites = favorites_data.favorites
favorites_file = favorites_data.__file__


class BVTK_PT_Favorites(bpy.types.Panel):
    """BVTK Favorites Panel"""

    bl_label = "Favorites"
    bl_space_type = "NODE_EDITOR"
    bl_region_type = "UI"
    bl_category = "Favorites"

    @classmethod
    def poll(cls, context):
        return context.space_data.tree_type == "BVTK_NodeTreeType"

    def draw(self, context):
        global favorites
        active_node = context.active_node
        layout = self.layout
        # Button to add active node to favorites
        if active_node:
            add = layout.operator(
                "node.bvtk_update_favorites", icon="ZOOM_IN", text=active_node.bl_label
            )
            add.label = active_node.bl_label
            add.type = active_node.bl_idname
            layout.separator()
        # Favorites buttons
        for f in favorites:
            row = layout.row(align=True)
            remove = row.operator(
                "node.bvtk_update_favorites", icon="PANEL_CLOSE", text=""
            )
            remove.label = f[1]
            remove.type = f[0]
            remove.remove = True
            op = row.operator("node.add_node", text=f[1])
            op.type = f[0]
            op.use_transform = True


class BVTK_OT_UpdateFavorites(bpy.types.Operator):
    """Update favorites operator"""

    bl_idname = "node.bvtk_update_favorites"
    bl_label = "add/remove favorites"

    remove: bpy.props.BoolProperty(default=False)
    label: bpy.props.StringProperty()
    type: bpy.props.StringProperty()

    def execute(self, context):
        global favorites
        global favorites_file
        fav = (self.type, self.label)
        if self.remove:
            favorites.remove(fav)
        else:
            if fav in favorites:
                self.report({"INFO"}, "Already in favorites")
                return {"FINISHED"}
            favorites.append(fav)
        open(favorites_file, "w").write(
            "favorites = " + repr(favorites).replace("),", "),\n")
        )
        self.remove = False
        return {"FINISHED"}


add_ui_class(BVTK_PT_Favorites)
add_ui_class(BVTK_OT_UpdateFavorites)