134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
|
|
import inspect
|
|
|
|
import bpy
|
|
from bpy.types import (AddonPreferences, PropertyGroup)
|
|
from bpy.props import (BoolProperty, StringProperty, CollectionProperty,
|
|
EnumProperty, IntProperty, PointerProperty)
|
|
|
|
from .core.bl_utils import get_addon_prefs
|
|
from .core.file_utils import import_module_from_path, norm_str
|
|
from .core.asset_library_utils import update_library_path
|
|
from .constants import PLUGINS, PLUGINS_DIR, PLUGINS_ITEMS
|
|
|
|
|
|
def load_plugins():
|
|
print('Asset Library: Load Library Plugins')
|
|
|
|
plugin_files = list(PLUGINS_DIR.glob('*.py'))
|
|
# if self.plugin_directory:
|
|
# user_plugin_DIR = Path(self.plugin_directory)
|
|
# if user_plugin_DIR.exists():
|
|
# plugin_files += list(user_plugin_DIR.glob('*.py'))
|
|
|
|
for plugin_file in plugin_files:
|
|
if plugin_file.stem.startswith('_'):
|
|
continue
|
|
|
|
mod = import_module_from_path(plugin_file)
|
|
|
|
for name, obj in inspect.getmembers(mod):
|
|
if not inspect.isclass(obj):
|
|
continue
|
|
|
|
if AssetLibrary not in obj.__mro__ or obj is AssetLibrary:
|
|
continue
|
|
|
|
try:
|
|
print(f'Register Plugin {name}')
|
|
bpy.utils.register_class(obj)
|
|
setattr(Plugins, norm_str(obj.name), PointerProperty(type=obj))
|
|
PLUGINS.append(obj)
|
|
|
|
except Exception as e:
|
|
print(f'Could not register plugin {name}')
|
|
print(e)
|
|
|
|
plugin_items = [('NONE', 'None', '', 0)]
|
|
plugin_items += [(norm_str(p.name, format=str.upper), p.name, "", i+1) for i, a in enumerate(PLUGINS)]
|
|
|
|
PLUGINS_ITEMS[:] = plugin_items
|
|
|
|
return PLUGINS
|
|
|
|
class Plugins(PropertyGroup):
|
|
"""Container holding the registed library plugins"""
|
|
def __iter__(self):
|
|
return (getattr(self, p) for p in self.bl_rna.properties.keys() if p not in ('rna_type', 'name'))
|
|
|
|
|
|
class AssetLibrary(PropertyGroup):
|
|
"""Library item defining one library with his plugin and settings"""
|
|
|
|
name : StringProperty(name='Name', default='', update=update_library_path)
|
|
expand : BoolProperty(name='Expand', default=False)
|
|
use : BoolProperty(name='Use', default=True, update=update_library_path)
|
|
is_env : BoolProperty(default=False)
|
|
#path : StringProperty(subtype='DIR_PATH')
|
|
|
|
plugins : PointerProperty(type=Plugins)
|
|
plugin_name : EnumProperty(items=lambda s, c : PLUGINS_ITEMS)
|
|
|
|
@property
|
|
def index(self):
|
|
prefs = get_addon_prefs()
|
|
return list(prefs.libraries).index(self)
|
|
|
|
"""
|
|
def draw_operators(self, layout):
|
|
row = layout.row(align=True)
|
|
row.alignment = 'RIGHT'
|
|
row.prop(self, 'plugin_name', text='')
|
|
row.prop(self, 'auto_bundle', text='', icon='UV_SYNC_SELECT')
|
|
|
|
row.operator("assetlibrary.diff", text='', icon='FILE_REFRESH').name = self.name
|
|
|
|
op = row.operator("assetlibrary.sync", icon='MOD_BUILD', text='')
|
|
op.name = self.name
|
|
|
|
layout.separator(factor=3)
|
|
"""
|
|
|
|
def draw(self, layout):
|
|
prefs = get_addon_prefs()
|
|
#col = layout.column(align=True)
|
|
box = layout.box()
|
|
|
|
row = box.row(align=True)
|
|
icon = "DISCLOSURE_TRI_DOWN" if self.expand else "DISCLOSURE_TRI_RIGHT"
|
|
row.prop(self, 'expand', icon=icon, emboss=False, text='')
|
|
|
|
row.prop(self, 'use', text='')
|
|
#row.label(icon="ASSET_MANAGER")
|
|
row.prop(self, 'name', text='')
|
|
row.separator(factor=0.5)
|
|
|
|
op = row.operator("assetlibrary.synchronize", icon='UV_SYNC_SELECT', text='')
|
|
op.name = self.name
|
|
|
|
row.separator(factor=0.5)
|
|
row.operator("assetlibrary.remove_library", icon="REMOVE", text='', emboss=False).index = self.index
|
|
|
|
#self.draw_operators(row)
|
|
if self.expand:
|
|
col = box.column(align=False)
|
|
col.use_property_split = True
|
|
|
|
col.prop(self, "path")
|
|
|
|
|
|
classes = (
|
|
Plugins,
|
|
AssetLibrary,
|
|
)
|
|
|
|
|
|
def register():
|
|
for cls in classes:
|
|
bpy.utils.register_class(cls)
|
|
|
|
load_plugins()
|
|
|
|
def unregister():
|
|
for cls in reversed(classes):
|
|
bpy.utils.unregister_class(cls) |