mirror of
https://github.com/kaosat-dev/Blender_bevy_components_workflow.git
synced 2024-11-23 04:10:53 +00:00
85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
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'} |