Blender_bevy_components_workflow/tools/blenvy/assets/operators.py

import os
import json
import bpy
from bpy_types import (Operator)
from bpy.props import (StringProperty, EnumProperty)

class OT_add_bevy_asset(Operator):
    """Add asset"""
    bl_idname = "bevyassets.add"
    bl_label = "Add bevy asset"
    bl_options = {"UNDO"}

    asset_name: StringProperty(
        name="asset name",
        description="name of asset to add",
    ) # type: ignore

    asset_type: EnumProperty(
            items=(
                ('MODEL', "Model", ""),
                ('AUDIO', "Audio", ""),
                ('IMAGE', "Image", ""),
                )
        ) # type: ignore

    asset_path: StringProperty(
        name="asset path",
        description="path of asset to add",
        subtype='FILE_PATH'
    ) # type: ignore

    def execute(self, context):
        assets_list = []
        blueprint_assets = False
        if context.collection is not None and context.collection.name == 'Scene Collection':
            assets_list = json.loads(context.scene.get('assets'))
            blueprint_assets = False
        else:
            if 'assets' in context.collection:
                assets_list = json.loads(context.collection.get('assets'))
            blueprint_assets = True

        in_list = [asset for asset in assets_list if (asset["path"] == self.asset_path)]
        in_list = len(in_list) > 0
        if not in_list:
            assets_list.append({"name": self.asset_name, "type": self.asset_type, "path": self.asset_path, "internal": False})

        if blueprint_assets:
            context.collection["assets"] = json.dumps(assets_list)
        else:
            context.scene["assets"] = json.dumps(assets_list)
        #context.window_manager.assets_registry.add_asset(self.asset_name, self.asset_type, self.asset_path, False)
        return {'FINISHED'}
    

class OT_remove_bevy_asset(Operator):
    """Remove asset"""
    bl_idname = "bevyassets.remove"
    bl_label = "remove bevy asset"
    bl_options = {"UNDO"}

    asset_path: StringProperty(
        name="asset path",
        description="path of asset to add",
        subtype='FILE_PATH'
    ) # type: ignore

    def execute(self, context):
        assets_list = []
        blueprint_assets = False
        if context.collection is not None and context.collection.name == 'Scene Collection':
            assets_list = json.loads(context.scene.get('assets'))
            blueprint_assets = False
        else:
            if 'assets' in context.collection:
                assets_list = json.loads(context.collection.get('assets'))
            blueprint_assets = True

        assets_list = [asset for asset in assets_list if (asset["path"] != self.asset_path)]
        if blueprint_assets:
            context.collection["assets"] = json.dumps(assets_list)
        else:
            context.scene["assets"] = json.dumps(assets_list)
        #context.window_manager.assets_registry.remove_asset(self.asset_path)
        return {'FINISHED'}