black format

This commit is contained in:
Joseph HENRY 2026-01-07 16:05:47 +01:00
parent f7c125ae7b
commit 3d65bb6e4d
49 changed files with 2817 additions and 2280 deletions

View File

@ -23,8 +23,9 @@ from asset_library import pose
from asset_library import action from asset_library import action
from asset_library import collection from asset_library import collection
from asset_library import file from asset_library import file
from asset_library import (gui, keymaps, preferences, operators) from asset_library import gui, keymaps, preferences, operators
from asset_library import constants from asset_library import constants
# from asset_library.common.library_type import LibraryType # from asset_library.common.library_type import LibraryType
from asset_library.common.bl_utils import get_addon_prefs from asset_library.common.bl_utils import get_addon_prefs
from asset_library.common.functions import set_env_libraries from asset_library.common.functions import set_env_libraries
@ -33,7 +34,7 @@ from asset_library.common.template import Template
import re import re
if 'bpy' in locals(): if "bpy" in locals():
print("Reload Addon Asset Library") print("Reload Addon Asset Library")
import importlib import importlib
@ -56,40 +57,26 @@ import os
# addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = [] # addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = []
bl_modules = ( bl_modules = (operators, pose, action, collection, file, keymaps, gui, preferences)
operators,
pose,
action,
collection,
file,
keymaps,
gui,
preferences
)
def load_handler(): def load_handler():
print('load_handler') print("load_handler")
set_env_libraries() set_env_libraries()
bpy.ops.assetlib.set_paths(all=True) bpy.ops.assetlib.set_paths(all=True)
if not bpy.app.background: if not bpy.app.background:
bpy.ops.assetlib.bundle(blocking=False, mode='AUTO_BUNDLE') bpy.ops.assetlib.bundle(blocking=False, mode="AUTO_BUNDLE")
def register() -> None: def register() -> None:
for m in bl_modules: for m in bl_modules:
m.register() m.register()
# prefs = get_addon_prefs() # prefs = get_addon_prefs()
bpy.app.timers.register(load_handler, first_interval=1) bpy.app.timers.register(load_handler, first_interval=1)
@ -99,5 +86,3 @@ def unregister() -> None:
for m in reversed(bl_modules): for m in reversed(bl_modules):
m.unregister() m.unregister()

View File

@ -1,4 +1,3 @@
from asset_library.action import ( from asset_library.action import (
gui, gui,
keymaps, keymaps,
@ -10,7 +9,7 @@ from asset_library.action import (
# render_preview # render_preview
) )
if 'bpy' in locals(): if "bpy" in locals():
import importlib import importlib
importlib.reload(gui) importlib.reload(gui)
@ -24,10 +23,12 @@ if 'bpy' in locals():
import bpy import bpy
def register(): def register():
operators.register() operators.register()
keymaps.register() keymaps.register()
def unregister(): def unregister():
operators.unregister() operators.unregister()
keymaps.unregister() keymaps.unregister()

View File

@ -1,7 +1,6 @@
import sys import sys
from pathlib import Path from pathlib import Path
# sys.path.append(str(Path(__file__).parents[3])) # sys.path.append(str(Path(__file__).parents[3]))
from asset_library.action.concat_preview import mosaic_export from asset_library.action.concat_preview import mosaic_export
@ -19,17 +18,19 @@ import subprocess
from tempfile import gettempdir from tempfile import gettempdir
def rm_tree(pth): def rm_tree(pth):
pth = Path(pth) pth = Path(pth)
for child in pth.glob('*'): for child in pth.glob("*"):
if child.is_file(): if child.is_file():
child.unlink() child.unlink()
else: else:
rm_tree(child) rm_tree(child)
pth.rmdir() pth.rmdir()
def render_preview(directory, asset_catalog, render_actions, publish_actions, remove_folder):
def render_preview(
directory, asset_catalog, render_actions, publish_actions, remove_folder
):
scn = bpy.context.scene scn = bpy.context.scene
rnd = bpy.context.scene.render rnd = bpy.context.scene.render
@ -39,27 +40,32 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
blendfile = Path(bpy.data.filepath) blendfile = Path(bpy.data.filepath)
asset_catalog_data = read_catalog(asset_catalog) asset_catalog_data = read_catalog(asset_catalog)
anim_render_dir = Path(gettempdir()) / 'actionlib_render' #/tmp/actionlib_render. Removed at the end anim_render_dir = (
Path(gettempdir()) / "actionlib_render"
) # /tmp/actionlib_render. Removed at the end
anim_render_dir.mkdir(exist_ok=True, parents=True) anim_render_dir.mkdir(exist_ok=True, parents=True)
preview_render_dir = Path(directory) / 'preview' preview_render_dir = Path(directory) / "preview"
if preview_render_dir.exists() and remove_folder: if preview_render_dir.exists() and remove_folder:
rm_tree(preview_render_dir) rm_tree(preview_render_dir)
preview_render_dir.mkdir(exist_ok=True, parents=True) preview_render_dir.mkdir(exist_ok=True, parents=True)
for i in ('anim', 'pose'): for i in ("anim", "pose"):
Path(preview_render_dir / i).mkdir(exist_ok=True, parents=True) Path(preview_render_dir / i).mkdir(exist_ok=True, parents=True)
for f in preview_render_dir.rglob('*'): for f in preview_render_dir.rglob("*"):
if f.is_dir(): if f.is_dir():
print(f'{f} is dir. Skipped.') print(f"{f} is dir. Skipped.")
continue continue
if all(i not in f.parts for i in ('anim', 'pose')) and f.parent.parts[-1] != 'preview': if (
print(f'{f} is out of pipe. Approved or Rtk pictures. Skipped.') all(i not in f.parts for i in ("anim", "pose"))
and f.parent.parts[-1] != "preview"
):
print(f"{f} is out of pipe. Approved or Rtk pictures. Skipped.")
continue continue
if not any(f.stem.endswith(a) for a in publish_actions): if not any(f.stem.endswith(a) for a in publish_actions):
print(f'{str(f)} not in publish actions anymore. Removing...') print(f"{str(f)} not in publish actions anymore. Removing...")
f.unlink() f.unlink()
# Set Scene # Set Scene
@ -70,7 +76,7 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
scn.tool_settings.use_keyframe_insert_auto = False scn.tool_settings.use_keyframe_insert_auto = False
# Render Setting # Render Setting
rnd.engine = 'BLENDER_EEVEE' rnd.engine = "BLENDER_EEVEE"
rnd.use_simplify = False rnd.use_simplify = False
rnd.use_stamp_date = True rnd.use_stamp_date = True
rnd.use_stamp_time = True rnd.use_stamp_time = True
@ -89,7 +95,7 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
rnd.use_stamp = True rnd.use_stamp = True
rnd.stamp_font_size = 16 rnd.stamp_font_size = 16
rnd.use_stamp_labels = False rnd.use_stamp_labels = False
rnd.image_settings.file_format = 'JPEG' rnd.image_settings.file_format = "JPEG"
# Viewport Look # Viewport Look
# ---------- # ----------
@ -108,94 +114,104 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
""" """
# Cycles Mat Shading # Cycles Mat Shading
for a in bpy.context.screen.areas: for a in bpy.context.screen.areas:
if a.type == 'VIEW_3D': if a.type == "VIEW_3D":
a.spaces[0].overlay.show_overlays = False a.spaces[0].overlay.show_overlays = False
a.spaces[0].region_3d.view_perspective = 'CAMERA' a.spaces[0].region_3d.view_perspective = "CAMERA"
a.spaces[0].shading.show_cavity = True a.spaces[0].shading.show_cavity = True
a.spaces[0].shading.cavity_type = 'WORLD' a.spaces[0].shading.cavity_type = "WORLD"
a.spaces[0].shading.cavity_ridge_factor = 0.75 a.spaces[0].shading.cavity_ridge_factor = 0.75
a.spaces[0].shading.cavity_valley_factor = 1.0 a.spaces[0].shading.cavity_valley_factor = 1.0
# Add Subsurf # Add Subsurf
# ----------- # -----------
deform_ob = [m.object for o in scn.objects \ deform_ob = [
for m in o.modifiers if m.type == 'MESH_DEFORM' m.object for o in scn.objects for m in o.modifiers if m.type == "MESH_DEFORM"
] ]
deform_ob += [m.target for o in scn.objects \ deform_ob += [
for m in o.modifiers if m.type == 'SURFACE_DEFORM' m.target for o in scn.objects for m in o.modifiers if m.type == "SURFACE_DEFORM"
] ]
objects = [o for o in bpy.context.scene.objects if (o.type == 'MESH' objects = [
and o not in deform_ob and o not in bpy.context.scene.collection.objects[:]) o
for o in bpy.context.scene.objects
if (
o.type == "MESH"
and o not in deform_ob
and o not in bpy.context.scene.collection.objects[:]
)
] ]
for o in objects: for o in objects:
subsurf = False subsurf = False
for m in o.modifiers: for m in o.modifiers:
if m.type == 'SUBSURF': if m.type == "SUBSURF":
m.show_viewport = m.show_render m.show_viewport = m.show_render
m.levels = m.render_levels m.levels = m.render_levels
subsurf = True subsurf = True
break break
if not subsurf: if not subsurf:
subsurf = o.modifiers.new('', 'SUBSURF') subsurf = o.modifiers.new("", "SUBSURF")
subsurf.show_viewport = subsurf.show_render subsurf.show_viewport = subsurf.show_render
subsurf.levels = subsurf.render_levels subsurf.levels = subsurf.render_levels
# Loop through action and render # Loop through action and render
# ------------------------------ # ------------------------------
rig = next((o for o in scn.objects if o.type == 'ARMATURE'), None) rig = next((o for o in scn.objects if o.type == "ARMATURE"), None)
# actions = [a for a in bpy.data.actions if a.asset_data] # actions = [a for a in bpy.data.actions if a.asset_data]
rig.animation_data_create() rig.animation_data_create()
for action_name in render_actions: for action_name in render_actions:
action = bpy.data.actions.get(action_name) action = bpy.data.actions.get(action_name)
if not action: if not action:
print(f'\'{action_name}\' not found.') print(f"'{action_name}' not found.")
continue continue
print(f"-- Current --: {action.name}") print(f"-- Current --: {action.name}")
rnd.stamp_note_text = '{type} : {pose_name}' rnd.stamp_note_text = "{type} : {pose_name}"
action_data = action.asset_data action_data = action.asset_data
if 'camera' not in action_data.keys(): if "camera" not in action_data.keys():
report.append(f"'{action.name}' has no CameraData.") report.append(f"'{action.name}' has no CameraData.")
continue continue
catalog_name = next((v['name'] for v in asset_catalog_data.values() if action_data.catalog_id == v['id']), None) catalog_name = next(
pose_name = '/'.join([*catalog_name.split('-'), action.name]) (
filename = bpy.path.clean_name(f'{catalog_name}_{action.name}') v["name"]
ext = 'jpg' for v in asset_catalog_data.values()
if action_data.catalog_id == v["id"]
),
None,
)
pose_name = "/".join([*catalog_name.split("-"), action.name])
filename = bpy.path.clean_name(f"{catalog_name}_{action.name}")
ext = "jpg"
rig.animation_data.action = None rig.animation_data.action = None
bpy.context.view_layer.update() bpy.context.view_layer.update()
for b in rig.pose.bones: for b in rig.pose.bones:
if re.match('^[A-Z]+\.', b.name): if re.match("^[A-Z]+\.", b.name):
continue continue
reset_bone(b) reset_bone(b)
rest_pose = None rest_pose = None
if isinstance(action.asset_data.get('rest_pose'), str): if isinstance(action.asset_data.get("rest_pose"), str):
rest_pose = bpy.data.actions.get(action.asset_data['rest_pose']) rest_pose = bpy.data.actions.get(action.asset_data["rest_pose"])
rig.animation_data.action = rest_pose rig.animation_data.action = rest_pose
bpy.context.view_layer.update() bpy.context.view_layer.update()
rig.animation_data.action = action rig.animation_data.action = action
if 'camera' in action.asset_data.keys(): if "camera" in action.asset_data.keys():
action_cam = bpy.data.objects.get(action.asset_data['camera'], '') action_cam = bpy.data.objects.get(action.asset_data["camera"], "")
if action_cam: if action_cam:
scn.camera = action_cam scn.camera = action_cam
# Is Anim # Is Anim
if not action_data['is_single_frame'] or 'anim' in action_data.tags.keys(): if not action_data["is_single_frame"] or "anim" in action_data.tags.keys():
keyframes = get_keyframes(action) keyframes = get_keyframes(action)
if not keyframes: if not keyframes:
continue continue
@ -203,57 +219,65 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
anim_end = keyframes[-1] anim_end = keyframes[-1]
if anim_start < scn.frame_start: if anim_start < scn.frame_start:
report.append(f"Issue found for '{action.name}'. Has keyframes before 'Start Frame'.") report.append(
f"Issue found for '{action.name}'. Has keyframes before 'Start Frame'."
)
continue continue
scn.frame_preview_start = anim_start scn.frame_preview_start = anim_start
scn.frame_preview_end = anim_end scn.frame_preview_end = anim_end
rnd.stamp_note_text = rnd.stamp_note_text.format( rnd.stamp_note_text = rnd.stamp_note_text.format(
type='ANIM', type="ANIM",
pose_name=pose_name, pose_name=pose_name,
) )
rnd.filepath = f'{str(anim_render_dir)}/{filename}_####.{ext}' rnd.filepath = f"{str(anim_render_dir)}/{filename}_####.{ext}"
bpy.ops.render.opengl(animation=True) bpy.ops.render.opengl(animation=True)
ffmpeg_cmd = [ ffmpeg_cmd = [
'ffmpeg', '-y', "ffmpeg",
'-start_number', f'{anim_start:04d}', "-y",
'-i', rnd.filepath.replace('####', '%04d'), "-start_number",
'-c:v', 'libx264', f"{anim_start:04d}",
str((preview_render_dir/'anim'/filename).with_suffix('.mov')), "-i",
rnd.filepath.replace("####", "%04d"),
"-c:v",
"libx264",
str((preview_render_dir / "anim" / filename).with_suffix(".mov")),
] ]
subprocess.call(ffmpeg_cmd) subprocess.call(ffmpeg_cmd)
# Is Pose # Is Pose
elif action_data['is_single_frame'] or 'pose' in action_data.tags.keys(): elif action_data["is_single_frame"] or "pose" in action_data.tags.keys():
scn.frame_preview_start = scn.frame_preview_end = scn.frame_start scn.frame_preview_start = scn.frame_preview_end = scn.frame_start
rnd.stamp_note_text = rnd.stamp_note_text.format( rnd.stamp_note_text = rnd.stamp_note_text.format(
type='POSE', type="POSE",
pose_name=pose_name, pose_name=pose_name,
) )
rnd.filepath = f'{str(preview_render_dir)}/pose/{filename}_####.{ext}' rnd.filepath = f"{str(preview_render_dir)}/pose/{filename}_####.{ext}"
bpy.ops.render.opengl(animation=True) bpy.ops.render.opengl(animation=True)
filename = rnd.filepath.replace('####', f'{scn.frame_preview_end:04d}') filename = rnd.filepath.replace("####", f"{scn.frame_preview_end:04d}")
Path(filename).rename(re.sub('_[0-9]{4}.', '.', filename)) Path(filename).rename(re.sub("_[0-9]{4}.", ".", filename))
shutil.rmtree(anim_render_dir) shutil.rmtree(anim_render_dir)
# Report # Report
# ------ # ------
if report: if report:
report_file = blendfile.parent / Path(f'{blendfile.stem}report').with_suffix('.txt') report_file = blendfile.parent / Path(f"{blendfile.stem}report").with_suffix(
".txt"
)
if not report_file.exists(): if not report_file.exists():
report_file.touch(exist_ok=False) report_file.touch(exist_ok=False)
report_file.write_text('-') report_file.write_text("-")
report_file.write_text('\n'.join(report)) report_file.write_text("\n".join(report))
result = report_file result = report_file
@ -262,30 +286,39 @@ def render_preview(directory, asset_catalog, render_actions, publish_actions, re
open_file(result) open_file(result)
files = [str(f) for f in sorted((preview_render_dir/'pose').glob('*.jpg'))] files = [str(f) for f in sorted((preview_render_dir / "pose").glob("*.jpg"))]
mosaic_export( mosaic_export(
files=files, catalog_data=asset_catalog_data, files=files,
row=2, columns=2, auto_calculate=True, catalog_data=asset_catalog_data,
bg_color=(0.18, 0.18, 0.18,), resize_output=100 row=2,
columns=2,
auto_calculate=True,
bg_color=(
0.18,
0.18,
0.18,
),
resize_output=100,
) )
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Add Comment To the tracker",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
if __name__ == '__main__' : parser.add_argument("--directory")
parser = argparse.ArgumentParser(description='Add Comment To the tracker', parser.add_argument("--asset-catalog")
formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument("--render-actions", nargs="+")
parser.add_argument("--publish-actions", nargs="+")
parser.add_argument("--remove-folder", type=json.loads, default="false")
parser.add_argument('--directory') if "--" in sys.argv:
parser.add_argument('--asset-catalog') index = sys.argv.index("--")
parser.add_argument('--render-actions', nargs='+')
parser.add_argument('--publish-actions', nargs='+')
parser.add_argument('--remove-folder', type=json.loads, default='false')
if '--' in sys.argv :
index = sys.argv.index('--')
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()

View File

@ -1,22 +1,23 @@
import argparse import argparse
import bpy import bpy
import json import json
import sys import sys
from pathlib import Path from pathlib import Path
# sys.path.append(str(Path(__file__).parents[3])) # sys.path.append(str(Path(__file__).parents[3]))
from asset_library.common.bl_utils import ( from asset_library.common.bl_utils import (
get_preview, get_preview,
) )
def clear_asset(action_name='', use_fake_user=False):
def clear_asset(action_name="", use_fake_user=False):
scn = bpy.context.scene scn = bpy.context.scene
action = bpy.data.actions.get(action_name) action = bpy.data.actions.get(action_name)
if not action: if not action:
print(f'No {action_name} not found.') print(f"No {action_name} not found.")
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
action.asset_clear() action.asset_clear()
@ -28,20 +29,20 @@ def clear_asset(action_name='', use_fake_user=False):
preview.unlink() preview.unlink()
bpy.data.actions.remove(action) bpy.data.actions.remove(action)
bpy.ops.wm.save_mainfile( bpy.ops.wm.save_mainfile(filepath=bpy.data.filepath, compress=True, exit=True)
filepath=bpy.data.filepath, compress=True, exit=True
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Add Comment To the tracker",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
) )
parser.add_argument("--action-name")
parser.add_argument("--use-fake-user", type=json.loads, default="false")
if __name__ == '__main__' : if "--" in sys.argv:
parser = argparse.ArgumentParser(description='Add Comment To the tracker', index = sys.argv.index("--")
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--action-name')
parser.add_argument('--use-fake-user', type=json.loads, default='false')
if '--' in sys.argv :
index = sys.argv.index('--')
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()

View File

@ -1,4 +1,3 @@
import bpy import bpy
import math import math
import numpy as np import numpy as np
@ -19,9 +18,11 @@ def alpha_to_color(pixels_data, color):
# print(new_pixels_data)#Dbg # print(new_pixels_data)#Dbg
return new_pixels_data return new_pixels_data
def create_array(height, width): def create_array(height, width):
return np.zeros((height * width * 4), dtype=np.float32) return np.zeros((height * width * 4), dtype=np.float32)
def read_pixels_data(img, source_height, source_width): def read_pixels_data(img, source_height, source_width):
img_w, img_h = img.size img_w, img_h = img.size
@ -43,6 +44,7 @@ def read_pixels_data(img, source_height, source_width):
return array.reshape(source_height, source_width, 4) return array.reshape(source_height, source_width, 4)
def create_final(output_name, pixels_data, final_height, final_width): def create_final(output_name, pixels_data, final_height, final_width):
# print('output_name: ', output_name) # print('output_name: ', output_name)
@ -58,23 +60,35 @@ def create_final(output_name, pixels_data, final_height, final_width):
return new_img return new_img
def guess_input_format(img_list): def guess_input_format(img_list):
for i in img_list: for i in img_list:
if i.size[0] == i.size[1]: if i.size[0] == i.size[1]:
return i.size return i.size
def format_files(files, catalog_data): def format_files(files, catalog_data):
img_dict = {} img_dict = {}
for k, v in catalog_data.items(): for k, v in catalog_data.items():
if '/' not in k: if "/" not in k:
continue continue
img_dict[v['name']] = [f for f in files if v['name'] in f] img_dict[v["name"]] = [f for f in files if v["name"] in f]
return img_dict return img_dict
def mosaic_export( def mosaic_export(
files, catalog_data, row=2, columns=2, auto_calculate=True, files,
bg_color=(0.18, 0.18, 0.18,), resize_output=100, catalog_data,
row=2,
columns=2,
auto_calculate=True,
bg_color=(
0.18,
0.18,
0.18,
),
resize_output=100,
): ):
img_dict = format_files(files, catalog_data) img_dict = format_files(files, catalog_data)
@ -92,21 +106,21 @@ def mosaic_export(
chars = Path(files_list[0]).parts[-4] chars = Path(files_list[0]).parts[-4]
output_dir = str(Path(files_list[0]).parent.parent) output_dir = str(Path(files_list[0]).parent.parent)
ext = 'jpg' ext = "jpg"
output_name = f'{chars}_{cat}.{ext}' output_name = f"{chars}_{cat}.{ext}"
for img in files_list: for img in files_list:
img_list.append(bpy.data.images.load(img, check_existing=True)) img_list.append(bpy.data.images.load(img, check_existing=True))
for i in img_list: for i in img_list:
i.colorspace_settings.name = 'Raw' i.colorspace_settings.name = "Raw"
if auto_calculate: if auto_calculate:
rows = int(math.sqrt(len(img_list))) rows = int(math.sqrt(len(img_list)))
columns = math.ceil(len(img_list) / rows) columns = math.ceil(len(img_list) / rows)
if rows * columns < len(img_list): if rows * columns < len(img_list):
raise AttributeError('Grid too small for number of images') raise AttributeError("Grid too small for number of images")
src_w, src_h = img_list[0].size src_w, src_h = img_list[0].size
final_w = src_w * columns final_w = src_w * columns
@ -130,17 +144,20 @@ def mosaic_export(
else: else:
combined_stack = np.hstack((h_stack[:])) combined_stack = np.hstack((h_stack[:]))
combined_img = create_final(output_name, combined_stack.flatten(), final_h, final_w) combined_img = create_final(
output_name, combined_stack.flatten(), final_h, final_w
)
if resize_output != 100: if resize_output != 100:
w, h = combined_img.size w, h = combined_img.size
combined_img.scale(w*(resize_output*.01), h*(resize_output*.01)) combined_img.scale(w * (resize_output * 0.01), h * (resize_output * 0.01))
combined_img.filepath_raw = "/".join([output_dir, output_name])
combined_img.filepath_raw = '/'.join([output_dir, output_name]) combined_img.file_format = "JPEG"
combined_img.file_format = 'JPEG'
combined_img.save() combined_img.save()
print(f""" print(
f"""
Image saved: {combined_img.filepath_raw} Image saved: {combined_img.filepath_raw}
""") """
)

View File

@ -13,15 +13,7 @@ import functools
import re import re
import bpy import bpy
from bpy.types import ( from bpy.types import Action, Bone, Context, FCurve, Keyframe, Object, TimelineMarker
Action,
Bone,
Context,
FCurve,
Keyframe,
Object,
TimelineMarker
)
from asset_library.common.bl_utils import active_catalog_id, split_path from asset_library.common.bl_utils import active_catalog_id, split_path
@ -30,6 +22,7 @@ FCurveValue = Union[float, int]
pose_bone_re = re.compile(r'pose.bones\["([^"]+)"\]') pose_bone_re = re.compile(r'pose.bones\["([^"]+)"\]')
"""RegExp for matching FCurve data paths.""" """RegExp for matching FCurve data paths."""
def is_pose(action): def is_pose(action):
for fc in action.fcurves: for fc in action.fcurves:
if len(fc.keyframe_points) > 1: if len(fc.keyframe_points) > 1:
@ -37,6 +30,7 @@ def is_pose(action):
return True return True
def get_bone_visibility(data_path): def get_bone_visibility(data_path):
bone, prop = split_path(data_path) bone, prop = split_path(data_path)
@ -47,6 +41,7 @@ def get_bone_visibility(data_path):
return ob.data.layers[b_layers[0]] return ob.data.layers[b_layers[0]]
def get_keyframes(action, selected=False, includes=[]): def get_keyframes(action, selected=False, includes=[]):
if selected: if selected:
# keyframes = sorted([int(k.co[0]) for f in action.fcurves for k in f.keyframe_points if k.select_control_point and get_bone_visibility(f.data_path)]) # keyframes = sorted([int(k.co[0]) for f in action.fcurves for k in f.keyframe_points if k.select_control_point and get_bone_visibility(f.data_path)])
@ -65,14 +60,21 @@ def get_keyframes(action, selected=False, includes=[]):
if len(keyframes) <= 1: if len(keyframes) <= 1:
keyframes = [bpy.context.scene.frame_current] keyframes = [bpy.context.scene.frame_current]
else: else:
keyframes = sorted([int(k.co[0]) for f in action.fcurves for k in f.keyframe_points]) keyframes = sorted(
[int(k.co[0]) for f in action.fcurves for k in f.keyframe_points]
)
return keyframes return keyframes
def get_marker(action): def get_marker(action):
if action.pose_markers: if action.pose_markers:
markers = action.pose_markers markers = action.pose_markers
return next((m.name for m in markers if m.frame == bpy.context.scene.frame_current), None) return next(
(m.name for m in markers if m.frame == bpy.context.scene.frame_current),
None,
)
def reset_bone(bone, transform=True, custom_props=True): def reset_bone(bone, transform=True, custom_props=True):
if transform: if transform:
@ -95,29 +97,32 @@ def reset_bone(bone, transform=True, custom_props=True):
if not isinstance(value, (int, float)) or not id_prop: if not isinstance(value, (int, float)) or not id_prop:
continue continue
bone[key] = id_prop.as_dict()['default'] bone[key] = id_prop.as_dict()["default"]
def is_asset_action(action): def is_asset_action(action):
return action.asset_data and action.asset_data.catalog_id != str(uuid.UUID(int=0)) return action.asset_data and action.asset_data.catalog_id != str(uuid.UUID(int=0))
def conform_action(action): def conform_action(action):
tags = ('pose', 'anim') tags = ("pose", "anim")
if any(tag in action.asset_data.tags.keys() for tag in tags): if any(tag in action.asset_data.tags.keys() for tag in tags):
return return
for fc in action.fcurves: for fc in action.fcurves:
action.asset_data['is_single_frame'] = True action.asset_data["is_single_frame"] = True
if len(fc.keyframe_points) > 1: if len(fc.keyframe_points) > 1:
action.asset_data['is_single_frame'] = False action.asset_data["is_single_frame"] = False
break break
if action.asset_data['is_single_frame']: if action.asset_data["is_single_frame"]:
action.asset_data.tags.new('pose') action.asset_data.tags.new("pose")
else: else:
action.asset_data.tags.new('anim') action.asset_data.tags.new("anim")
def clean_action(action='', frame_start=0, frame_end=0, excludes=[], includes=[]):
def clean_action(action="", frame_start=0, frame_end=0, excludes=[], includes=[]):
## Clean Keyframe Before/After Range ## Clean Keyframe Before/After Range
for fc in action.fcurves: for fc in action.fcurves:
bone, prop = split_path(fc.data_path) bone, prop = split_path(fc.data_path)
@ -138,14 +143,16 @@ def clean_action(action='', frame_start=0, frame_end=0, excludes=[], includes=[]
fc.keyframe_points.remove(k) fc.keyframe_points.remove(k)
fc.update() fc.update()
def append_action(action_path='', action_name=''):
print(f'Loading {action_name} from: {action_path}') def append_action(action_path="", action_name=""):
print(f"Loading {action_name} from: {action_path}")
with bpy.data.libraries.load(str(action_path), link=False) as (data_from, data_to): with bpy.data.libraries.load(str(action_path), link=False) as (data_from, data_to):
data_to.actions = [action_name] data_to.actions = [action_name]
return data_to.actions[0] return data_to.actions[0]
def apply_anim(action_lib, ob, bones=[]): def apply_anim(action_lib, ob, bones=[]):
from mathutils import Vector from mathutils import Vector
@ -162,14 +169,23 @@ def apply_anim(action_lib, ob, bones=[]):
keys = sorted([k.co[0] for f in action_lib.fcurves for k in f.keyframe_points]) keys = sorted([k.co[0] for f in action_lib.fcurves for k in f.keyframe_points])
if not keys: if not keys:
print(f'The action {action_lib.name} has no keyframes') print(f"The action {action_lib.name} has no keyframes")
return return
first_key = keys[0] first_key = keys[0]
key_offset = scn.frame_current - first_key key_offset = scn.frame_current - first_key
key_attr = ('type', 'interpolation', 'handle_left_type', 'handle_right_type', key_attr = (
'amplitude', 'back', 'easing', 'period', 'handle_right', 'handle_left' "type",
"interpolation",
"handle_left_type",
"handle_right_type",
"amplitude",
"back",
"easing",
"period",
"handle_right",
"handle_left",
) )
for fc in action_lib.fcurves: for fc in action_lib.fcurves:
bone_name, prop_name = split_path(fc.data_path) bone_name, prop_name = split_path(fc.data_path)
@ -182,17 +198,16 @@ def apply_anim(action_lib, ob, bones=[]):
action_fc = action.fcurves.new( action_fc = action.fcurves.new(
fc.data_path, fc.data_path,
index=fc.array_index, index=fc.array_index,
action_group=fc.group.name if fc.group else fc.data_path.split('"')[1] action_group=fc.group.name if fc.group else fc.data_path.split('"')[1],
) )
for kf_lib in fc.keyframe_points: for kf_lib in fc.keyframe_points:
kf = action_fc.keyframe_points.insert( kf = action_fc.keyframe_points.insert(
frame=kf_lib.co[0] + key_offset, frame=kf_lib.co[0] + key_offset, value=kf_lib.co[1]
value=kf_lib.co[1]
) )
for attr in key_attr: for attr in key_attr:
src_val = getattr(kf_lib, attr) src_val = getattr(kf_lib, attr)
if attr.startswith('handle') and 'type' not in attr: if attr.startswith("handle") and "type" not in attr:
src_val += Vector((key_offset, 0)) src_val += Vector((key_offset, 0))
setattr(kf, attr, src_val) setattr(kf, attr, src_val)
@ -203,5 +218,5 @@ def apply_anim(action_lib, ob, bones=[]):
for window in bpy.context.window_manager.windows: for window in bpy.context.window_manager.windows:
screen = window.screen screen = window.screen
for area in screen.areas: for area in screen.areas:
if area.type == 'GRAPH_EDITOR': if area.type == "GRAPH_EDITOR":
area.tag_redraw() area.tag_redraw()

View File

@ -1,4 +1,3 @@
import bpy import bpy
@ -6,12 +5,14 @@ def draw_context_menu(layout):
params = bpy.context.space_data.params params = bpy.context.space_data.params
asset = bpy.context.asset_file_handle asset = bpy.context.asset_file_handle
layout.operator("assetlib.open_blend", text="Open blend file")#.asset = asset.name layout.operator(
"assetlib.open_blend", text="Open blend file"
) # .asset = asset.name
layout.operator("assetlib.play_preview", text="Play Preview") layout.operator("assetlib.play_preview", text="Play Preview")
layout.separator() layout.separator()
layout.operator_context = 'INVOKE_DEFAULT' layout.operator_context = "INVOKE_DEFAULT"
# layout.operator("assetlib.rename_asset", text="Rename Action") # layout.operator("assetlib.rename_asset", text="Rename Action")
layout.operator("assetlib.remove_assets", text="Remove Assets") layout.operator("assetlib.remove_assets", text="Remove Assets")
@ -21,29 +22,42 @@ def draw_context_menu(layout):
layout.separator() layout.separator()
layout.operator("actionlib.apply_selected_action", text="Apply Pose").flipped = False layout.operator("actionlib.apply_selected_action", text="Apply Pose").flipped = (
layout.operator("actionlib.apply_selected_action", text="Apply Pose (Flipped)").flipped = True False
)
layout.operator(
"actionlib.apply_selected_action", text="Apply Pose (Flipped)"
).flipped = True
layout.separator() layout.separator()
layout.operator("poselib.blend_pose_asset_for_keymap", text="Blend Pose").flipped = False layout.operator(
layout.operator("poselib.blend_pose_asset_for_keymap", text="Blend Pose (Flipped)").flipped = True "poselib.blend_pose_asset_for_keymap", text="Blend Pose"
).flipped = False
layout.operator(
"poselib.blend_pose_asset_for_keymap", text="Blend Pose (Flipped)"
).flipped = True
layout.separator() layout.separator()
layout.operator("poselib.pose_asset_select_bones", text="Select Bones").selected_side = 'CURRENT' layout.operator(
layout.operator("poselib.pose_asset_select_bones", text="Select Bones (Flipped)").selected_side = 'FLIPPED' "poselib.pose_asset_select_bones", text="Select Bones"
layout.operator("poselib.pose_asset_select_bones", text="Select Bones (Both)").selected_side = 'BOTH' ).selected_side = "CURRENT"
layout.operator(
"poselib.pose_asset_select_bones", text="Select Bones (Flipped)"
).selected_side = "FLIPPED"
layout.operator(
"poselib.pose_asset_select_bones", text="Select Bones (Both)"
).selected_side = "BOTH"
layout.separator() layout.separator()
# layout.operator("asset.library_refresh") # layout.operator("asset.library_refresh")
if params.display_type == 'THUMBNAIL': if params.display_type == "THUMBNAIL":
layout.prop_menu_enum(params, "display_size") layout.prop_menu_enum(params, "display_size")
def draw_header(layout): def draw_header(layout):
'''Draw the header of the Asset Browser Window''' """Draw the header of the Asset Browser Window"""
layout.separator() layout.separator()
layout.operator("actionlib.store_anim_pose", text='Add Action', icon='FILE_NEW') layout.operator("actionlib.store_anim_pose", text="Add Action", icon="FILE_NEW")

View File

@ -1,11 +1,10 @@
from typing import List, Tuple from typing import List, Tuple
import bpy import bpy
addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = [] addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = []
def register(): def register():
wm = bpy.context.window_manager wm = bpy.context.window_manager
addon = wm.keyconfigs.addon addon = wm.keyconfigs.addon
@ -15,34 +14,49 @@ def register():
km = addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER") km = addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER")
# DblClick to apply pose. # DblClick to apply pose.
kmi = km.keymap_items.new("actionlib.apply_selected_action", "LEFTMOUSE", "DOUBLE_CLICK") kmi = km.keymap_items.new(
"actionlib.apply_selected_action", "LEFTMOUSE", "DOUBLE_CLICK"
)
kmi.properties.flipped = False kmi.properties.flipped = False
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("actionlib.apply_selected_action", "LEFTMOUSE", "DOUBLE_CLICK", alt=True) kmi = km.keymap_items.new(
"actionlib.apply_selected_action", "LEFTMOUSE", "DOUBLE_CLICK", alt=True
)
kmi.properties.flipped = True kmi.properties.flipped = True
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("poselib.blend_pose_asset_for_keymap", "LEFTMOUSE", "DOUBLE_CLICK", shift=True) kmi = km.keymap_items.new(
"poselib.blend_pose_asset_for_keymap", "LEFTMOUSE", "DOUBLE_CLICK", shift=True
)
kmi.properties.flipped = False kmi.properties.flipped = False
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("poselib.blend_pose_asset_for_keymap", "LEFTMOUSE", "DOUBLE_CLICK", alt=True, shift=True) kmi = km.keymap_items.new(
"poselib.blend_pose_asset_for_keymap",
"LEFTMOUSE",
"DOUBLE_CLICK",
alt=True,
shift=True,
)
kmi.properties.flipped = True kmi.properties.flipped = True
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("poselib.pose_asset_select_bones", "S", "PRESS") kmi = km.keymap_items.new("poselib.pose_asset_select_bones", "S", "PRESS")
kmi.properties.selected_side = 'CURRENT' kmi.properties.selected_side = "CURRENT"
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("poselib.pose_asset_select_bones", "S", "PRESS", alt=True) kmi = km.keymap_items.new("poselib.pose_asset_select_bones", "S", "PRESS", alt=True)
kmi.properties.selected_side = 'FLIPPED' kmi.properties.selected_side = "FLIPPED"
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("poselib.pose_asset_select_bones", "S", "PRESS", alt=True, ctrl=True) kmi = km.keymap_items.new(
kmi.properties.selected_side = 'BOTH' "poselib.pose_asset_select_bones", "S", "PRESS", alt=True, ctrl=True
)
kmi.properties.selected_side = "BOTH"
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
def unregister(): def unregister():
for km, kmi in addon_keymaps: for km, kmi in addon_keymaps:
km.keymap_items.remove(kmi) km.keymap_items.remove(kmi)

View File

@ -26,13 +26,10 @@ from pprint import pprint
from asset_library.pose.pose_creation import ( from asset_library.pose.pose_creation import (
create_pose_asset_from_context, create_pose_asset_from_context,
assign_from_asset_browser assign_from_asset_browser,
) )
from asset_library.pose.pose_usage import( from asset_library.pose.pose_usage import select_bones, flip_side_name
select_bones,
flip_side_name
)
from asset_library.action.functions import ( from asset_library.action.functions import (
apply_anim, apply_anim,
@ -40,7 +37,7 @@ from asset_library.action.functions import(
clean_action, clean_action,
reset_bone, reset_bone,
is_asset_action, is_asset_action,
conform_action conform_action,
) )
from bpy.props import ( from bpy.props import (
@ -49,7 +46,7 @@ from bpy.props import (
EnumProperty, EnumProperty,
PointerProperty, PointerProperty,
StringProperty, StringProperty,
IntProperty IntProperty,
) )
from bpy.types import ( from bpy.types import (
@ -80,7 +77,7 @@ from asset_library.common.functions import (
# resync_lib, # resync_lib,
get_active_library, get_active_library,
get_active_catalog, get_active_catalog,
asset_warning_callback asset_warning_callback,
) )
from asset_library.common.bl_utils import ( from asset_library.common.bl_utils import (
@ -95,7 +92,7 @@ from asset_library.common.bl_utils import (
# load_assets_from, # load_assets_from,
get_asset_space_params, get_asset_space_params,
get_bl_cmd, get_bl_cmd,
get_overriden_col get_overriden_col,
) )
from asset_library.common.file_utils import ( from asset_library.common.file_utils import (
@ -137,17 +134,17 @@ class ACTIONLIB_OT_restore_previous_action(Operator):
self._timer = wm.event_timer_add(0.001, window=context.window) self._timer = wm.event_timer_add(0.001, window=context.window)
wm.modal_handler_add(self) wm.modal_handler_add(self)
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
def modal(self, context, event): def modal(self, context, event):
if event.type != 'TIMER': if event.type != "TIMER":
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
wm = context.window_manager wm = context.window_manager
wm.event_timer_remove(self._timer) wm.event_timer_remove(self._timer)
context.object.pose.apply_pose_from_action(self.pose_action) context.object.pose.apply_pose_from_action(self.pose_action)
return {'FINISHED'} return {"FINISHED"}
class ACTIONLIB_OT_assign_action(Operator): class ACTIONLIB_OT_assign_action(Operator):
@ -171,8 +168,8 @@ class ACTIONLIB_OT_assign_action(Operator):
class ACTIONLIB_OT_replace_pose(Operator): class ACTIONLIB_OT_replace_pose(Operator):
bl_idname = "actionlib.replace_pose" bl_idname = "actionlib.replace_pose"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Update Pose' bl_label = "Update Pose"
bl_description = 'Update selected Pose. ! Works only on Pose, not Anim !' bl_description = "Update selected Pose. ! Works only on Pose, not Anim !"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
@ -183,7 +180,11 @@ class ACTIONLIB_OT_replace_pose(Operator):
# else: # else:
# cls.poll_message_set(f"Current Action {context.id.name} different than Edit Action {wm.edit_pose_action}") # cls.poll_message_set(f"Current Action {context.id.name} different than Edit Action {wm.edit_pose_action}")
# return False # return False
if context.mode == 'POSE' and context.area.ui_type == 'ASSETS' and context.active_file: if (
context.mode == "POSE"
and context.area.ui_type == "ASSETS"
and context.active_file
):
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
@ -194,12 +195,13 @@ class ACTIONLIB_OT_replace_pose(Operator):
select_bones( select_bones(
context.object, context.object,
context.asset_file_handle.local_id, context.asset_file_handle.local_id,
selected_side='BOTH', selected_side="BOTH",
toggle=False) toggle=False,
)
data = { data = {
'name':active.name, "name": active.name,
'catalog_id':active.asset_data.catalog_id, "catalog_id": active.asset_data.catalog_id,
} }
data.update(dict(active.asset_data)) data.update(dict(active.asset_data))
@ -212,7 +214,7 @@ class ACTIONLIB_OT_replace_pose(Operator):
action = create_pose_asset_from_context( action = create_pose_asset_from_context(
context, context,
data['name'], data["name"],
) )
if not action: if not action:
self.report( # type: ignore self.report( # type: ignore
@ -226,33 +228,36 @@ class ACTIONLIB_OT_replace_pose(Operator):
_old_action.use_fake_user = False _old_action.use_fake_user = False
bpy.data.actions.remove(_old_action) bpy.data.actions.remove(_old_action)
action.name = data['name'] action.name = data["name"]
action.asset_data.catalog_id = data['catalog_id'] action.asset_data.catalog_id = data["catalog_id"]
for k, v in data.items(): for k, v in data.items():
if k in ('camera', 'is_single_frame'): if k in ("camera", "is_single_frame"):
action.asset_data[k] = v action.asset_data[k] = v
if not is_pose(action) and 'pose' in action.asset_data.tags.keys() and 'anim' not in action.asset_data.tags.keys(): if (
not is_pose(action)
and "pose" in action.asset_data.tags.keys()
and "anim" not in action.asset_data.tags.keys()
):
for tag in action.asset_data.tags: for tag in action.asset_data.tags:
if tag != 'pose': if tag != "pose":
continue continue
action.asset_data.tags.remove(tag) action.asset_data.tags.remove(tag)
action.asset_data.tags.new('anim') action.asset_data.tags.new("anim")
return {"FINISHED"}
return {'FINISHED'}
class ACTIONLIB_OT_apply_anim(Operator): class ACTIONLIB_OT_apply_anim(Operator):
bl_idname = "actionlib.apply_anim" bl_idname = "actionlib.apply_anim"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Apply Anim' bl_label = "Apply Anim"
bl_description = 'Apply selected Anim to selected bones' bl_description = "Apply selected Anim to selected bones"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
return context.mode == 'POSE' return context.mode == "POSE"
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
ob = context.object ob = context.object
@ -263,20 +268,20 @@ class ACTIONLIB_OT_apply_anim(Operator):
# params = get_asset_space_params(context.area) # params = get_asset_space_params(context.area)
asset_library_ref = context.asset_library_ref asset_library_ref = context.asset_library_ref
if asset_library_ref == 'LOCAL': if asset_library_ref == "LOCAL":
action = bpy.data.actions[active_action.name] action = bpy.data.actions[active_action.name]
to_remove = False to_remove = False
else: else:
asset_file_handle = bpy.context.asset_file_handle asset_file_handle = bpy.context.asset_file_handle
if asset_file_handle is None: if asset_file_handle is None:
return {'CANCELLED'} return {"CANCELLED"}
if asset_file_handle.local_id: if asset_file_handle.local_id:
return {'CANCELLED'} return {"CANCELLED"}
lib = get_active_library() lib = get_active_library()
if 'filepath' in asset_file_handle.asset_data: if "filepath" in asset_file_handle.asset_data:
action_path = asset_file_handle.asset_data['filepath'] action_path = asset_file_handle.asset_data["filepath"]
action_path = lib.library_type.format_path(action_path) action_path = lib.library_type.format_path(action_path)
else: else:
action_path = bpy.types.AssetHandle.get_full_library_path( action_path = bpy.types.AssetHandle.get_full_library_path(
@ -290,7 +295,9 @@ class ACTIONLIB_OT_apply_anim(Operator):
) )
to_remove = True to_remove = True
else: else:
self.report({"WARNING"}, f"Could not load action path {action_path} not exist") self.report(
{"WARNING"}, f"Could not load action path {action_path} not exist"
)
return {"CANCELLED"} return {"CANCELLED"}
bones = [b.name for b in context.selected_pose_bones_from_active_object] bones = [b.name for b in context.selected_pose_bones_from_active_object]
@ -298,7 +305,7 @@ class ACTIONLIB_OT_apply_anim(Operator):
if to_remove: if to_remove:
bpy.data.actions.remove(action) bpy.data.actions.remove(action)
return {'FINISHED'} return {"FINISHED"}
""" """
@ -477,8 +484,10 @@ class ACTIONLIB_OT_create_anim_asset(Operator):
return True return True
params = get_asset_space_params(asset_browse_area) params = get_asset_space_params(asset_browse_area)
if params.asset_library_ref != 'LOCAL': if params.asset_library_ref != "LOCAL":
cls.poll_message_set("Asset Browser must be set to the Current File library") cls.poll_message_set(
"Asset Browser must be set to the Current File library"
)
return False return False
return True return True
@ -498,67 +507,71 @@ class ACTIONLIB_OT_create_anim_asset(Operator):
data_dict = dict( data_dict = dict(
is_single_frame=False, is_single_frame=False,
camera= context.scene.camera.name if context.scene.camera else '', camera=context.scene.camera.name if context.scene.camera else "",
) )
data.tags.new('anim') data.tags.new("anim")
for k, v in data_dict.items(): for k, v in data_dict.items():
data[k] = v data[k] = v
### ###
return {'FINISHED'} return {"FINISHED"}
class ACTIONLIB_OT_apply_selected_action(Operator): class ACTIONLIB_OT_apply_selected_action(Operator):
bl_idname = "actionlib.apply_selected_action" bl_idname = "actionlib.apply_selected_action"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Apply Pose/Anim' bl_label = "Apply Pose/Anim"
bl_description = 'Apply selected Action to selected bones' bl_description = "Apply selected Action to selected bones"
flipped: BoolProperty(name="Flipped", default=False) # type: ignore flipped: BoolProperty(name="Flipped", default=False) # type: ignore
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
if context.mode == 'POSE' and context.area.ui_type == 'ASSETS': if context.mode == "POSE" and context.area.ui_type == "ASSETS":
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
active_action = context.active_file active_action = context.active_file
if 'pose' in active_action.asset_data.tags.keys(): if "pose" in active_action.asset_data.tags.keys():
bpy.ops.poselib.apply_pose_asset_for_keymap(flipped=self.flipped) bpy.ops.poselib.apply_pose_asset_for_keymap(flipped=self.flipped)
else: else:
bpy.ops.actionlib.apply_anim() bpy.ops.actionlib.apply_anim()
return {'FINISHED'} return {"FINISHED"}
class ACTIONLIB_OT_edit_action(Operator): class ACTIONLIB_OT_edit_action(Operator):
bl_idname = "actionlib.edit_action" bl_idname = "actionlib.edit_action"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Edit Action' bl_label = "Edit Action"
bl_description = 'Assign active action and set Camera' bl_description = "Assign active action and set Camera"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
if context.mode == 'POSE' and context.area.ui_type == 'ASSETS' and context.active_file: if (
context.mode == "POSE"
and context.area.ui_type == "ASSETS"
and context.active_file
):
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
scn = context.scene scn = context.scene
rest_pose = bpy.data.actions.get(context.id.asset_data.get('rest_pose', '')) rest_pose = bpy.data.actions.get(context.id.asset_data.get("rest_pose", ""))
context.object.animation_data_create() context.object.animation_data_create()
keyframes = get_keyframes(context.id) keyframes = get_keyframes(context.id)
if not keyframes: if not keyframes:
self.report({'ERROR'}, f'No Keyframes found for {context.id.name}.') self.report({"ERROR"}, f"No Keyframes found for {context.id.name}.")
return return
scn.frame_set(keyframes[0]) scn.frame_set(keyframes[0])
context.object.animation_data.action = None context.object.animation_data.action = None
for b in context.object.pose.bones: for b in context.object.pose.bones:
if re.match('^[A-Z]+\.', b.name): if re.match("^[A-Z]+\.", b.name):
continue continue
reset_bone(b) reset_bone(b)
@ -566,8 +579,8 @@ class ACTIONLIB_OT_edit_action(Operator):
context.view_layer.update() context.view_layer.update()
context.object.animation_data.action = context.id context.object.animation_data.action = context.id
if 'camera' in context.id.asset_data.keys(): if "camera" in context.id.asset_data.keys():
scn.camera = bpy.data.objects[context.id.asset_data['camera']] scn.camera = bpy.data.objects[context.id.asset_data["camera"]]
return {"FINISHED"} return {"FINISHED"}
@ -575,12 +588,16 @@ class ACTIONLIB_OT_edit_action(Operator):
class ACTIONLIB_OT_clear_action(Operator): class ACTIONLIB_OT_clear_action(Operator):
bl_idname = "actionlib.clear_action" bl_idname = "actionlib.clear_action"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Clear Action' bl_label = "Clear Action"
bl_description = 'Assign active action and set Camera' bl_description = "Assign active action and set Camera"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
if context.mode == 'POSE' and context.area.ui_type == 'ASSETS' and context.active_file: if (
context.mode == "POSE"
and context.area.ui_type == "ASSETS"
and context.active_file
):
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
@ -591,7 +608,7 @@ class ACTIONLIB_OT_clear_action(Operator):
context.id.asset_generate_preview() context.id.asset_generate_preview()
for a in context.screen.areas: for a in context.screen.areas:
if a.type == 'DOPESHEET_EDITOR' and a.ui_type == 'DOPESHEET': if a.type == "DOPESHEET_EDITOR" and a.ui_type == "DOPESHEET":
a.tag_redraw() a.tag_redraw()
return {"FINISHED"} return {"FINISHED"}
@ -600,12 +617,12 @@ class ACTIONLIB_OT_clear_action(Operator):
class ACTIONLIB_OT_generate_preview(Operator): class ACTIONLIB_OT_generate_preview(Operator):
bl_idname = "actionlib.generate_preview" bl_idname = "actionlib.generate_preview"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Generate Preview' bl_label = "Generate Preview"
bl_description = 'Genreate Preview for Active Action' bl_description = "Genreate Preview for Active Action"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
if context.object.type == 'ARMATURE' and context.mode == 'POSE': if context.object.type == "ARMATURE" and context.mode == "POSE":
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
@ -616,18 +633,18 @@ class ACTIONLIB_OT_generate_preview(Operator):
actions = context.selected_files + [_action] actions = context.selected_files + [_action]
for a in context.selected_files: for a in context.selected_files:
rest_pose = bpy.data.actions.get(a.asset_data.get('rest_pose', '')) rest_pose = bpy.data.actions.get(a.asset_data.get("rest_pose", ""))
if rest_pose: if rest_pose:
context.object.animation_data.action = rest_pose context.object.animation_data.action = rest_pose
bpy.context.view_layer.update() bpy.context.view_layer.update()
else: else:
context.object.animation_data.action = None context.object.animation_data.action = None
for b in context.object.pose.bones: for b in context.object.pose.bones:
if re.match('^[A-Z]+\.', b.name): if re.match("^[A-Z]+\.", b.name):
continue continue
reset_bone(b) reset_bone(b)
a_cam = bpy.data.objects.get(a.asset_data['camera']) a_cam = bpy.data.objects.get(a.asset_data["camera"])
if a_cam: if a_cam:
context.scene.camera = a_cam context.scene.camera = a_cam
bpy.context.view_layer.update() bpy.context.view_layer.update()
@ -644,15 +661,15 @@ class ACTIONLIB_OT_generate_preview(Operator):
class ACTIONLIB_OT_update_action_data(Operator): class ACTIONLIB_OT_update_action_data(Operator):
bl_idname = "actionlib.update_action_data" bl_idname = "actionlib.update_action_data"
bl_options = {"REGISTER", "INTERNAL"} bl_options = {"REGISTER", "INTERNAL"}
bl_label = 'Udpate Action Data' bl_label = "Udpate Action Data"
bl_description = 'Update Action Metadata' bl_description = "Update Action Metadata"
tags: EnumProperty( tags: EnumProperty(
name='Tags', name="Tags",
items=( items=(
('POSE', "pose", ""), ("POSE", "pose", ""),
('ANIM', "anim", ""), ("ANIM", "anim", ""),
) ),
) )
use_tags: BoolProperty(default=False) use_tags: BoolProperty(default=False)
@ -688,7 +705,7 @@ class ACTIONLIB_OT_update_action_data(Operator):
row.prop(self, "use_camera", text="") row.prop(self, "use_camera", text="")
sub = row.row() sub = row.row()
sub.active = self.use_camera sub.active = self.use_camera
sub.prop(scn.actionlib, "camera", text="", icon='CAMERA_DATA') sub.prop(scn.actionlib, "camera", text="", icon="CAMERA_DATA")
heading = layout.column(align=True, heading="Rest Pose") heading = layout.column(align=True, heading="Rest Pose")
row = heading.row(align=True) row = heading.row(align=True)
@ -702,11 +719,11 @@ class ACTIONLIB_OT_update_action_data(Operator):
for action in context.selected_files: for action in context.selected_files:
if self.use_camera: if self.use_camera:
action.asset_data['camera'] = context.scene.actionlib.camera.name action.asset_data["camera"] = context.scene.actionlib.camera.name
if self.use_tags: if self.use_tags:
tag = self.tags.lower() tag = self.tags.lower()
not_tag = 'anim' if tag == 'pose' else 'pose' not_tag = "anim" if tag == "pose" else "pose"
if tag not in action.asset_data.tags.keys(): if tag not in action.asset_data.tags.keys():
if not_tag in action.asset_data.tags.keys(): if not_tag in action.asset_data.tags.keys():
@ -717,29 +734,27 @@ class ACTIONLIB_OT_update_action_data(Operator):
action.asset_data.tags.new(tag) action.asset_data.tags.new(tag)
if 'pose' in action.asset_data.tags.keys(): if "pose" in action.asset_data.tags.keys():
action.asset_data['is_single_frame'] = True action.asset_data["is_single_frame"] = True
else: else:
action.asset_data['is_single_frame'] = False action.asset_data["is_single_frame"] = False
if self.use_rest_pose: if self.use_rest_pose:
name = scn.actionlib.rest_pose.name if scn.actionlib.rest_pose else '' name = scn.actionlib.rest_pose.name if scn.actionlib.rest_pose else ""
action.asset_data['rest_pose'] = name action.asset_data["rest_pose"] = name
return {'FINISHED'}
return {"FINISHED"}
class ACTIONLIB_OT_assign_rest_pose(Operator): class ACTIONLIB_OT_assign_rest_pose(Operator):
bl_idname = "actionlib.mark_as_rest_pose" bl_idname = "actionlib.mark_as_rest_pose"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Mark As Rest Pose' bl_label = "Mark As Rest Pose"
bl_description = 'Mark Pose as Rest Pose' bl_description = "Mark Pose as Rest Pose"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
if context.area.ui_type == 'ASSETS': if context.area.ui_type == "ASSETS":
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
@ -747,14 +762,14 @@ class ACTIONLIB_OT_assign_rest_pose(Operator):
context.scene.actionlib.rest_pose = context.id context.scene.actionlib.rest_pose = context.id
print(f"'{active_action.name.title()}' marked as Rest Pose.") print(f"'{active_action.name.title()}' marked as Rest Pose.")
return {'FINISHED'} return {"FINISHED"}
class ACTIONLIB_OT_open_blendfile(Operator): class ACTIONLIB_OT_open_blendfile(Operator):
bl_idname = "actionlib.open_blendfile" bl_idname = "actionlib.open_blendfile"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Set Paths' bl_label = "Set Paths"
bl_description = 'Open Containing File' bl_description = "Open Containing File"
replace_local: BoolProperty(default=False) replace_local: BoolProperty(default=False)
@ -764,7 +779,7 @@ class ACTIONLIB_OT_open_blendfile(Operator):
return False return False
sp = context.space_data sp = context.space_data
if sp.params.asset_library_ref == 'LOCAL': if sp.params.asset_library_ref == "LOCAL":
return False return False
return True return True
@ -777,13 +792,11 @@ class ACTIONLIB_OT_open_blendfile(Operator):
blendfile=str(asset_path), blendfile=str(asset_path),
) )
subprocess.Popen(cmd) subprocess.Popen(cmd)
return {'FINISHED'} return {"FINISHED"}
# LIBRARY_ITEMS = [] # LIBRARY_ITEMS = []
''' """
def callback_operator(modal_func, operator, override={}): def callback_operator(modal_func, operator, override={}):
def wrap(self, context, event): def wrap(self, context, event):
@ -795,7 +808,7 @@ def callback_operator(modal_func, operator, override={}):
return retset return retset
return wrap return wrap
''' """
class ACTIONLIB_OT_make_custom_preview(Operator): class ACTIONLIB_OT_make_custom_preview(Operator):
@ -807,34 +820,47 @@ class ACTIONLIB_OT_make_custom_preview(Operator):
prefs = get_addons_prefs() prefs = get_addons_prefs()
if not prefs.preview_modal: if not prefs.preview_modal:
with context.temp_override(area=self.source_area, region=self.source_area.regions[-1]): with context.temp_override(
bpy.ops.actionlib.store_anim_pose("INVOKE_DEFAULT", clear_previews=False, **prefs.add_asset_dict) area=self.source_area, region=self.source_area.regions[-1]
):
bpy.ops.actionlib.store_anim_pose(
"INVOKE_DEFAULT", clear_previews=False, **prefs.add_asset_dict
)
return {"FINISHED"} return {"FINISHED"}
return {'PASS_THROUGH'} return {"PASS_THROUGH"}
def invoke(self, context, event): def invoke(self, context, event):
self.source_area = bpy.context.area self.source_area = bpy.context.area
view3d = get_viewport() view3d = get_viewport()
with context.temp_override(area=view3d, region=view3d.regions[-1], window=context.window): with context.temp_override(
area=view3d, region=view3d.regions[-1], window=context.window
):
# To close the popup # To close the popup
bpy.ops.screen.screen_full_area() bpy.ops.screen.screen_full_area()
bpy.ops.screen.back_to_previous() bpy.ops.screen.back_to_previous()
view3d = get_viewport() view3d = get_viewport()
with context.temp_override(area=view3d, region=view3d.regions[-1], window=context.window): with context.temp_override(
bpy.ops.assetlib.make_custom_preview('INVOKE_DEFAULT', modal=True) area=view3d, region=view3d.regions[-1], window=context.window
):
bpy.ops.assetlib.make_custom_preview("INVOKE_DEFAULT", modal=True)
context.window_manager.modal_handler_add(self) context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
def get_preview_items(self, context): def get_preview_items(self, context):
prefs = get_addon_prefs() prefs = get_addon_prefs()
return sorted([(k, k, '', v.icon_id, index) for index, (k, v) in enumerate(prefs.previews.items())], reverse=True) return sorted(
[
(k, k, "", v.icon_id, index)
for index, (k, v) in enumerate(prefs.previews.items())
],
reverse=True,
)
class ACTIONLIB_OT_store_anim_pose(Operator): class ACTIONLIB_OT_store_anim_pose(Operator):
@ -842,32 +868,40 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
bl_label = "Add Action to the current library" bl_label = "Add Action to the current library"
bl_description = "Store current pose/anim to local library" bl_description = "Store current pose/anim to local library"
warning: StringProperty(name='') warning: StringProperty(name="")
path: StringProperty(name='Path') path: StringProperty(name="Path")
catalog: StringProperty(name='Catalog', update=asset_warning_callback, options={'TEXTEDIT_UPDATE'}) catalog: StringProperty(
name: StringProperty(name='Name', update=asset_warning_callback, options={'TEXTEDIT_UPDATE'}) name="Catalog", update=asset_warning_callback, options={"TEXTEDIT_UPDATE"}
action_type: EnumProperty(name='Type', items=[('POSE', 'Pose', ''), ('ANIMATION', 'Animation', '')]) )
name: StringProperty(
name="Name", update=asset_warning_callback, options={"TEXTEDIT_UPDATE"}
)
action_type: EnumProperty(
name="Type", items=[("POSE", "Pose", ""), ("ANIMATION", "Animation", "")]
)
frame_start: IntProperty(name="Frame Start") frame_start: IntProperty(name="Frame Start")
frame_end: IntProperty(name="Frame End") frame_end: IntProperty(name="Frame End")
tags: StringProperty(name='Tags', description='Tags need to separate with a comma (,)') tags: StringProperty(
description: StringProperty(name='Description') name="Tags", description="Tags need to separate with a comma (,)"
)
description: StringProperty(name="Description")
preview: EnumProperty(items=get_preview_items) preview: EnumProperty(items=get_preview_items)
clear_previews: BoolProperty(default=True) clear_previews: BoolProperty(default=True)
store_library: StringProperty(name='Store Library') store_library: StringProperty(name="Store Library")
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
ob = context.object ob = context.object
if not ob: if not ob:
cls.poll_message_set(f'You have no active object') cls.poll_message_set(f"You have no active object")
return False return False
if not ob.type == 'ARMATURE': if not ob.type == "ARMATURE":
cls.poll_message_set(f'Active object {ob.name} is not of type ARMATURE') cls.poll_message_set(f"Active object {ob.name} is not of type ARMATURE")
return False return False
if not ob.animation_data or not ob.animation_data.action: if not ob.animation_data or not ob.animation_data.action:
cls.poll_message_set(f'Active object {ob.name} has no action or keyframes') cls.poll_message_set(f"Active object {ob.name} has no action or keyframes")
return False return False
return True return True
@ -882,7 +916,16 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
# col.operator("asset.tag_remove", icon='REMOVE', text="") # col.operator("asset.tag_remove", icon='REMOVE', text="")
def to_dict(self): def to_dict(self):
keys = ("catalog", "name", "action_type", "frame_start", "frame_end", "tags", "description", "store_library") keys = (
"catalog",
"name",
"action_type",
"frame_start",
"frame_end",
"tags",
"description",
"store_library",
)
return {k: getattr(self, k) for k in keys} return {k: getattr(self, k) for k in keys}
def draw(self, context): def draw(self, context):
@ -893,41 +936,43 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
split = layout.split(factor=0.39, align=True) split = layout.split(factor=0.39, align=True)
# row = split.row(align=False) # row = split.row(align=False)
# split.use_property_split = False # split.use_property_split = False
split.alignment = 'RIGHT' split.alignment = "RIGHT"
split.label(text='Preview') split.label(text="Preview")
sub = split.row(align=True) sub = split.row(align=True)
sub.template_icon_view(self, "preview", show_labels=False) sub.template_icon_view(self, "preview", show_labels=False)
sub.separator() sub.separator()
sub.operator("actionlib.make_custom_preview", icon='RESTRICT_RENDER_OFF', text='') sub.operator(
"actionlib.make_custom_preview", icon="RESTRICT_RENDER_OFF", text=""
)
prefs.add_asset_dict.clear() prefs.add_asset_dict.clear()
prefs.add_asset_dict.update(self.to_dict()) prefs.add_asset_dict.update(self.to_dict())
sub.label(icon='BLANK1') sub.label(icon="BLANK1")
# layout.ui_units_x = 50 # layout.ui_units_x = 50
# row = layout.row(align=True) # row = layout.row(align=True)
layout.use_property_split = True layout.use_property_split = True
if self.current_library.merge_libraries: if self.current_library.merge_libraries:
layout.prop(self.current_library, 'store_library', expand=False) layout.prop(self.current_library, "store_library", expand=False)
# row.label(text='Action Name') # row.label(text='Action Name')
layout.prop(self, "action_type", text="Type", expand=True) layout.prop(self, "action_type", text="Type", expand=True)
if self.action_type == 'ANIMATION': if self.action_type == "ANIMATION":
col = layout.column(align=True) col = layout.column(align=True)
col.prop(self, "frame_start") col.prop(self, "frame_start")
col.prop(self, "frame_end", text='End') col.prop(self, "frame_end", text="End")
layout.prop(self, "catalog", text="Catalog") layout.prop(self, "catalog", text="Catalog")
layout.prop(self, "name", text="Name") layout.prop(self, "name", text="Name")
# layout.separator() # layout.separator()
# self.draw_tags(self.asset_action, layout) # self.draw_tags(self.asset_action, layout)
layout.prop(self, 'tags') layout.prop(self, "tags")
layout.prop(self, 'description', text='Description') layout.prop(self, "description", text="Description")
# layout.prop(prefs, 'author', text='Author') # layout.prop(prefs, 'author', text='Author')
# layout.prop() # layout.prop()
@ -941,7 +986,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
col.label(text=self.path) col.label(text=self.path)
if self.warning: if self.warning:
col.label(icon='ERROR', text=self.warning) col.label(icon="ERROR", text=self.warning)
def set_action_type(self): def set_action_type(self):
ob = bpy.context.object ob = bpy.context.object
@ -953,9 +998,9 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
self.frame_start = min(keyframes_selected) self.frame_start = min(keyframes_selected)
self.frame_end = max(keyframes_selected) self.frame_end = max(keyframes_selected)
self.action_type = 'POSE' self.action_type = "POSE"
if (self.frame_start != self.frame_end): if self.frame_start != self.frame_end:
self.action_type = 'ANIMATION' self.action_type = "ANIMATION"
def invoke(self, context, event): def invoke(self, context, event):
prefs = get_addon_prefs() prefs = get_addon_prefs()
@ -975,8 +1020,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
self.store_library = lib.name self.store_library = lib.name
# lib = self.current_library # lib = self.current_library
self.tags = '' self.tags = ""
# print(self, self.library_items) # print(self, self.library_items)
catalog_item = lib.catalog.active_item catalog_item = lib.catalog.active_item
@ -991,7 +1035,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
view3d = get_viewport() view3d = get_viewport()
with context.temp_override(area=view3d, region=view3d.regions[-1]): with context.temp_override(area=view3d, region=view3d.regions[-1]):
bpy.ops.assetlib.make_custom_preview('INVOKE_DEFAULT') bpy.ops.assetlib.make_custom_preview("INVOKE_DEFAULT")
else: else:
preview_items = get_preview_items(self, context) preview_items = get_preview_items(self, context)
@ -1012,17 +1056,17 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
action=action, action=action,
frame_start=self.frame_start, frame_start=self.frame_start,
frame_end=self.frame_end, frame_end=self.frame_end,
excludes=['world', 'walk'], excludes=["world", "walk"],
includes=bones, includes=bones,
) )
## Define Tags ## Define Tags
tags = [t.strip() for t in self.tags.split(',') if t] tags = [t.strip() for t in self.tags.split(",") if t]
tag_range = f'f{self.frame_start}' tag_range = f"f{self.frame_start}"
is_single_frame = True is_single_frame = True
if self.action_type == 'ANIM': if self.action_type == "ANIM":
is_single_frame = False is_single_frame = False
tag_range = f'f{self.frame_start}-f{self.frame_end}' tag_range = f"f{self.frame_start}-f{self.frame_end}"
tags.append(self.action_type) tags.append(self.action_type)
tags.append(tag_range) tags.append(tag_range)
@ -1030,15 +1074,15 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
for tag in tags: for tag in tags:
action.asset_data.tags.new(tag) action.asset_data.tags.new(tag)
action.asset_data['is_single_frame'] = is_single_frame action.asset_data["is_single_frame"] = is_single_frame
action.asset_data['rig'] = bpy.context.object.name action.asset_data["rig"] = bpy.context.object.name
action.asset_data.description = self.description action.asset_data.description = self.description
action.asset_data.author = prefs.author action.asset_data.author = prefs.author
col = get_overriden_col(bpy.context.object) col = get_overriden_col(bpy.context.object)
if col: if col:
action.asset_data['col'] = col.name action.asset_data["col"] = col.name
return action return action
@ -1050,21 +1094,21 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
space = area.spaces.active space = area.spaces.active
attrs = [ attrs = [
(scn, 'use_preview_range', True), (scn, "use_preview_range", True),
(scn, 'frame_preview_start', self.frame_start), (scn, "frame_preview_start", self.frame_start),
(scn, 'frame_preview_end', self.frame_end), (scn, "frame_preview_end", self.frame_end),
(scn.render, 'resolution_percentage', 100), (scn.render, "resolution_percentage", 100),
(space.overlay, 'show_overlays', False), (space.overlay, "show_overlays", False),
(space.region_3d, 'view_perspective', 'CAMERA'), (space.region_3d, "view_perspective", "CAMERA"),
(scn.render, 'resolution_x', 1280), (scn.render, "resolution_x", 1280),
(scn.render, 'resolution_y', 720), (scn.render, "resolution_y", 720),
(scn.render.image_settings, 'file_format', 'FFMPEG'), (scn.render.image_settings, "file_format", "FFMPEG"),
(scn.render.ffmpeg, 'format', 'QUICKTIME'), (scn.render.ffmpeg, "format", "QUICKTIME"),
(scn.render.ffmpeg, 'codec', 'H264'), (scn.render.ffmpeg, "codec", "H264"),
(scn.render.ffmpeg, 'ffmpeg_preset', 'GOOD'), (scn.render.ffmpeg, "ffmpeg_preset", "GOOD"),
(scn.render.ffmpeg, 'constant_rate_factor', 'HIGH'), (scn.render.ffmpeg, "constant_rate_factor", "HIGH"),
(scn.render.ffmpeg, 'gopsize', 12), (scn.render.ffmpeg, "gopsize", 12),
(scn.render, 'filepath', str(video_path)), (scn.render, "filepath", str(video_path)),
] ]
if self.action_type == "ANIMATION": if self.action_type == "ANIMATION":
@ -1076,7 +1120,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
pass pass
def refresh(self, area): def refresh(self, area):
bpy.ops.asset.library_refresh({"area": area, 'region': area.regions[3]}) bpy.ops.asset.library_refresh({"area": area, "region": area.regions[3]})
# space_data.activate_asset_by_id(asset, deferred=deferred) # space_data.activate_asset_by_id(asset, deferred=deferred)
def execute(self, context: Context): def execute(self, context: Context):
@ -1094,8 +1138,12 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
lib_type = lib.library_type lib_type = lib.library_type
asset_path = lib_type.get_asset_path(name=self.name, catalog=self.catalog) asset_path = lib_type.get_asset_path(name=self.name, catalog=self.catalog)
img_path = lib_type.get_image_path(name=self.name, catalog=self.catalog, filepath=asset_path) img_path = lib_type.get_image_path(
video_path = lib_type.get_video_path(name=self.name, catalog=self.catalog, filepath=asset_path) name=self.name, catalog=self.catalog, filepath=asset_path
)
video_path = lib_type.get_video_path(
name=self.name, catalog=self.catalog, filepath=asset_path
)
## Copy Action ## Copy Action
current_action = ob.animation_data.action current_action = ob.animation_data.action
@ -1128,7 +1176,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
# print('asset_info') # print('asset_info')
# pprint(asset_info) # pprint(asset_info)
diff = [dict(a, operation='ADD') for a in lib_type.flatten_cache([asset_info])] diff = [dict(a, operation="ADD") for a in lib_type.flatten_cache([asset_info])]
ob.animation_data.action = current_action ob.animation_data.action = current_action
@ -1137,7 +1185,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
bpy.data.actions.remove(asset_action) bpy.data.actions.remove(asset_action)
# TODO Write a proper method for this # TODO Write a proper method for this
diff_path = Path(bpy.app.tempdir, 'diff.json') diff_path = Path(bpy.app.tempdir, "diff.json")
# diff = [dict(a, operation='ADD') for a in [asset_info])] # diff = [dict(a, operation='ADD') for a in [asset_info])]
diff_path.write_text(json.dumps(diff, indent=4)) diff_path.write_text(json.dumps(diff, indent=4))
@ -1146,7 +1194,7 @@ class ACTIONLIB_OT_store_anim_pose(Operator):
# self.area.tag_redraw() # self.area.tag_redraw()
self.report({'INFO'}, f'"{self.name}" has been added to the library.') self.report({"INFO"}, f'"{self.name}" has been added to the library.')
return {"FINISHED"} return {"FINISHED"}
@ -1165,7 +1213,7 @@ classes = (
ACTIONLIB_OT_update_action_data, ACTIONLIB_OT_update_action_data,
ACTIONLIB_OT_assign_rest_pose, ACTIONLIB_OT_assign_rest_pose,
ACTIONLIB_OT_store_anim_pose, ACTIONLIB_OT_store_anim_pose,
ACTIONLIB_OT_make_custom_preview ACTIONLIB_OT_make_custom_preview,
) )
register, unregister = bpy.utils.register_classes_factory(classes) register, unregister = bpy.utils.register_classes_factory(classes)

View File

@ -2,20 +2,19 @@ import bpy
from bpy.types import PropertyGroup from bpy.types import PropertyGroup
from bpy.props import PointerProperty, StringProperty, BoolProperty from bpy.props import PointerProperty, StringProperty, BoolProperty
class ACTIONLIB_PG_scene(PropertyGroup): class ACTIONLIB_PG_scene(PropertyGroup):
flipped: BoolProperty( flipped: BoolProperty(
name="Flip Pose", name="Flip Pose",
default=False, default=False,
) )
previous_action: PointerProperty(type=bpy.types.Action) previous_action: PointerProperty(type=bpy.types.Action)
publish_path : StringProperty(subtype='FILE_PATH') publish_path: StringProperty(subtype="FILE_PATH")
camera : PointerProperty(type=bpy.types.Object, poll=lambda s, o: o.type == 'CAMERA') camera: PointerProperty(type=bpy.types.Object, poll=lambda s, o: o.type == "CAMERA")
rest_pose: PointerProperty(type=bpy.types.Action, poll=lambda s, a: a.asset_data) rest_pose: PointerProperty(type=bpy.types.Action, poll=lambda s, a: a.asset_data)
classes = ( classes = (ACTIONLIB_PG_scene,)
ACTIONLIB_PG_scene,
)
def register(): def register():
@ -24,6 +23,7 @@ def register():
bpy.types.Scene.actionlib = PointerProperty(type=ACTIONLIB_PG_scene) bpy.types.Scene.actionlib = PointerProperty(type=ACTIONLIB_PG_scene)
def unregister(): def unregister():
try: try:
del bpy.types.Scene.actionlib del bpy.types.Scene.actionlib

View File

@ -1,4 +1,3 @@
import argparse import argparse
import bpy import bpy
import json import json
@ -6,17 +5,19 @@ import re
import sys import sys
from pathlib import Path from pathlib import Path
# sys.path.append(str(Path(__file__).parents[3])) # sys.path.append(str(Path(__file__).parents[3]))
from asset_library.common.bl_utils import ( from asset_library.common.bl_utils import (
get_preview, get_preview,
) )
def rename_pose(src_name='', dst_name=''):
def rename_pose(src_name="", dst_name=""):
scn = bpy.context.scene scn = bpy.context.scene
action = bpy.data.actions.get(src_name) action = bpy.data.actions.get(src_name)
if not action: if not action:
print(f'No {src_name} not found.') print(f"No {src_name} not found.")
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
action.name = dst_name action.name = dst_name
@ -24,20 +25,20 @@ def rename_pose(src_name='', dst_name=''):
if preview: if preview:
preview.rename(re.sub(src_name, dst_name, str(preview))) preview.rename(re.sub(src_name, dst_name, str(preview)))
bpy.ops.wm.save_mainfile( bpy.ops.wm.save_mainfile(filepath=bpy.data.filepath, compress=True, exit=True)
filepath=bpy.data.filepath, compress=True, exit=True
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Add Comment To the tracker",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
) )
parser.add_argument("--src-name")
parser.add_argument("--dst-name")
if __name__ == '__main__' : if "--" in sys.argv:
parser = argparse.ArgumentParser(description='Add Comment To the tracker', index = sys.argv.index("--")
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--src-name')
parser.add_argument('--dst-name')
if '--' in sys.argv :
index = sys.argv.index('--')
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()

View File

@ -1,5 +1,3 @@
from bpy.types import PropertyGroup from bpy.types import PropertyGroup
@ -7,6 +5,11 @@ class Adapter(PropertyGroup):
# def __init__(self): # def __init__(self):
name = "Base Adapter" name = "Base Adapter"
# library = None # library = None
def to_dict(self): def to_dict(self):
return {p: getattr(self, p) for p in self.bl_rna.properties.keys() if p !='rna_type'} return {
p: getattr(self, p)
for p in self.bl_rna.properties.keys()
if p != "rna_type"
}

View File

@ -1,4 +1,3 @@
from asset_library.collection import ( from asset_library.collection import (
gui, gui,
operators, operators,
@ -7,7 +6,7 @@ from asset_library.collection import (
# create_collection_library, # create_collection_library,
) )
if 'bpy' in locals(): if "bpy" in locals():
import importlib import importlib
importlib.reload(gui) importlib.reload(gui)
@ -18,10 +17,12 @@ if 'bpy' in locals():
import bpy import bpy
def register(): def register():
operators.register() operators.register()
keymaps.register() keymaps.register()
def unregister(): def unregister():
operators.unregister() operators.unregister()
keymaps.unregister() keymaps.unregister()

View File

@ -29,6 +29,7 @@ from asset_library.constants import ASSETLIB_FILENAME
] ]
""" """
def build_collection_blends(path, categories=None, clean=True): def build_collection_blends(path, categories=None, clean=True):
t0 = time() t0 = time()
@ -43,30 +44,33 @@ def build_collection_blends(path, categories=None, clean=True):
category_datas = json.loads(json_path.read_text()) category_datas = json.loads(json_path.read_text())
for category_data in category_datas: for category_data in category_datas:
if categories and category_data['name'] not in categories: if categories and category_data["name"] not in categories:
continue continue
bpy.ops.wm.read_homefile(use_empty=True) bpy.ops.wm.read_homefile(use_empty=True)
# category_data = next(c for c in category_datas if c['name'] == category) # category_data = next(c for c in category_datas if c['name'] == category)
# _col_datas = category_data['children'] # _col_datas = category_data['children']
cat_name = category_data['name'] cat_name = category_data["name"]
build_path = Path(path) / cat_name / f'{cat_name}.blend' build_path = Path(path) / cat_name / f"{cat_name}.blend"
## re-iterate in grouped filepath ## re-iterate in grouped filepath
col_datas = sorted(category_data['children'], key=lambda x: x['filepath']) col_datas = sorted(category_data["children"], key=lambda x: x["filepath"])
for filepath, col_data_groups in groupby(col_datas, key=lambda x: x['filepath']): for filepath, col_data_groups in groupby(
col_datas, key=lambda x: x["filepath"]
):
# f = Path(f) # f = Path(f)
if not Path(filepath).exists(): if not Path(filepath).exists():
print(f'Not exists: {filepath}') print(f"Not exists: {filepath}")
continue continue
col_data_groups = list(col_data_groups) col_data_groups = list(col_data_groups)
col_names = [a['name'] for a in col_data_groups] col_names = [a["name"] for a in col_data_groups]
linked_cols = load_datablocks(filepath, col_names, link=True, type='collections') linked_cols = load_datablocks(
filepath, col_names, link=True, type="collections"
)
for i, col in enumerate(linked_cols): for i, col in enumerate(linked_cols):
# iterate in linked collection and associated data # iterate in linked collection and associated data
@ -78,10 +82,10 @@ def build_collection_blends(path, categories=None, clean=True):
## Directly link as collection inside a marked collection with same name ## Directly link as collection inside a marked collection with same name
marked_col = col_as_asset(col, verbose=True) marked_col = col_as_asset(col, verbose=True)
marked_col.asset_data.description = asset_data.get('description', '') marked_col.asset_data.description = asset_data.get("description", "")
marked_col.asset_data.catalog_id = category_data['id'] # assign catalog marked_col.asset_data.catalog_id = category_data["id"] # assign catalog
for k, v in asset_data.get('metadata', {}).items(): for k, v in asset_data.get("metadata", {}).items():
marked_col.asset_data[k] = v marked_col.asset_data[k] = v
## exclude collections and generate preview ## exclude collections and generate preview
@ -93,31 +97,35 @@ def build_collection_blends(path, categories=None, clean=True):
## clear all objects (can be very long with a lot of objects...): ## clear all objects (can be very long with a lot of objects...):
if clean: if clean:
print('Removing links...') print("Removing links...")
for lib in reversed(bpy.data.libraries): for lib in reversed(bpy.data.libraries):
bpy.data.libraries.remove(lib) bpy.data.libraries.remove(lib)
# Créer les dossiers intermediaires # Créer les dossiers intermediaires
build_path.parent.mkdir(parents=True, exist_ok=True) build_path.parent.mkdir(parents=True, exist_ok=True)
print('Saving to', build_path) print("Saving to", build_path)
bpy.ops.wm.save_as_mainfile(filepath=str(build_path), compress=False) bpy.ops.wm.save_as_mainfile(filepath=str(build_path), compress=False)
print("build time:", f'{time() - t0:.1f}s') print("build time:", f"{time() - t0:.1f}s")
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
if __name__ == '__main__' : if __name__ == "__main__":
parser = argparse.ArgumentParser(description='build_collection_blends', parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter) description="build_collection_blends",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument('-path') # Trouve/créer le json assetlib.json en sous-dossier de libdir parser.add_argument(
parser.add_argument('--category') # Lit la category dans le json et a link tout dans le blend "-path"
) # Trouve/créer le json assetlib.json en sous-dossier de libdir
parser.add_argument(
"--category"
) # Lit la category dans le json et a link tout dans le blend
if '--' in sys.argv : if "--" in sys.argv:
index = sys.argv.index('--') index = sys.argv.index("--")
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()

View File

@ -44,10 +44,11 @@ from asset_library.constants import ASSETLIB_FILENAME
] ]
""" """
def create_collection_json(path, source_directory): def create_collection_json(path, source_directory):
'''Create a Json from every marked collection in blends """Create a Json from every marked collection in blends
contained in folderpath (respect hierachy) contained in folderpath (respect hierachy)
''' """
json_path = Path(path) / ASSETLIB_FILENAME json_path = Path(path) / ASSETLIB_FILENAME
@ -56,75 +57,80 @@ def create_collection_json(path, source_directory):
# or open all blends and look only for marked collection ? (if versionned, get still get only last) # or open all blends and look only for marked collection ? (if versionned, get still get only last)
# get all blend in dir and subdirs (only last when versionned _v???) # get all blend in dir and subdirs (only last when versionned _v???)
blends = get_last_files(source_directory, pattern=r'(_v\d{3})?\.blend$', only_matching=True) blends = get_last_files(
source_directory, pattern=r"(_v\d{3})?\.blend$", only_matching=True
)
root_path = Path(source_directory).as_posix().rstrip('/') + '/' root_path = Path(source_directory).as_posix().rstrip("/") + "/"
print('root_path: ', root_path) print("root_path: ", root_path)
# open and check data block marked as asset # open and check data block marked as asset
category_datas = [] category_datas = []
for i, blend in enumerate(blends): for i, blend in enumerate(blends):
fp = Path(blend) fp = Path(blend)
print(f'{i+1}/{len(blends)}') print(f"{i+1}/{len(blends)}")
## What is considered a grouping category ? top level folders ? parents[1] ? ## What is considered a grouping category ? top level folders ? parents[1] ?
## Remove root path and extension ## Remove root path and extension
## top level folder ('chars'), problem if blends at root ## top level folder ('chars'), problem if blends at root
category = fp.as_posix().replace(root_path, '').split('/')[0] category = fp.as_posix().replace(root_path, "").split("/")[0]
## full blend path (chars/perso/perso) ## full blend path (chars/perso/perso)
# category = fp.as_posix().replace(root_path, '').rsplit('.', 1)[0] # category = fp.as_posix().replace(root_path, '').rsplit('.', 1)[0]
print(category) print(category)
with bpy.data.libraries.load(blend, link=True, assets_only=True) as (data_from, data_to): with bpy.data.libraries.load(blend, link=True, assets_only=True) as (
data_from,
data_to,
):
## just listing ## just listing
col_name_list = [c for c in data_from.collections] col_name_list = [c for c in data_from.collections]
if not col_name_list: if not col_name_list:
continue continue
col_list = next((c['children'] for c in category_datas if c['name'] == category), None) col_list = next(
(c["children"] for c in category_datas if c["name"] == category), None
)
if col_list is None: if col_list is None:
col_list = [] col_list = []
category_data = { category_data = {
'name': category, "name": category,
'id': str(uuid.uuid4()), "id": str(uuid.uuid4()),
'children': col_list, "children": col_list,
} }
category_datas.append(category_data) category_datas.append(category_data)
blend_source_path = blend.as_posix() blend_source_path = blend.as_posix()
if (project_root := os.environ.get('PROJECT_ROOT')): if project_root := os.environ.get("PROJECT_ROOT"):
blend_source_path = blend_source_path.replace(project_root, '$PROJECT_ROOT') blend_source_path = blend_source_path.replace(project_root, "$PROJECT_ROOT")
for name in col_name_list: for name in col_name_list:
data = { data = {
'filepath' : blend, "filepath": blend,
'name' : name, "name": name,
# 'tags' : [], # 'tags' : [],
'metadata' : {'filepath': blend_source_path}, "metadata": {"filepath": blend_source_path},
} }
col_list.append(data) col_list.append(data)
json_path.write_text(json.dumps(category_datas, indent='\t')) json_path.write_text(json.dumps(category_datas, indent="\t"))
## create text catalog from json (keep_existing_category ?) ## create text catalog from json (keep_existing_category ?)
create_catalog_file(json_path, keep_existing_category=True) create_catalog_file(json_path, keep_existing_category=True)
def create_collection_library(path, source_directory=None): def create_collection_library(path, source_directory=None):
''' """
path: store collection library (json and blends database) path: store collection library (json and blends database)
source_directory: if a source is set, rebuild json and library source_directory: if a source is set, rebuild json and library
''' """
if source_directory: if source_directory:
if not Path(source_directory).exists(): if not Path(source_directory).exists():
print(f'Source directory not exists: {source_directory}') print(f"Source directory not exists: {source_directory}")
return return
## scan source and build json in assetlib dir root ## scan source and build json in assetlib dir root
@ -132,31 +138,44 @@ def create_collection_library(path, source_directory=None):
json_path = Path(path) / ASSETLIB_FILENAME json_path = Path(path) / ASSETLIB_FILENAME
if not json_path.exists(): if not json_path.exists():
print(f'No json found at: {json_path}') print(f"No json found at: {json_path}")
return return
file_datas = json.loads(json_path.read()) file_datas = json.loads(json_path.read())
## For each category in json, execute build_assets_blend script ## For each category in json, execute build_assets_blend script
script = Path(__file__).parent / 'build_collection_blends.py' script = Path(__file__).parent / "build_collection_blends.py"
# empty_blend = Path(__file__).parent / 'empty_scene.blend' # empty_blend = Path(__file__).parent / 'empty_scene.blend'
# for category, asset_datas in file_datas.items(): # for category, asset_datas in file_datas.items():
for category_data in file_datas: for category_data in file_datas:
## add an empty blend as second arg ## add an empty blend as second arg
cmd = [bpy.app.binary_path, '--python', str(script), '--', '--path', path, '--category', category_data['name']] cmd = [
print('cmd: ', cmd) bpy.app.binary_path,
"--python",
str(script),
"--",
"--path",
path,
"--category",
category_data["name"],
]
print("cmd: ", cmd)
subprocess.call(cmd) subprocess.call(cmd)
if __name__ == '__main__' : if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Create Collection Library', parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter) description="Create Collection Library",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument('--path') # trouve/créer le json assetlib.json en sous-dossier de libdir parser.add_argument(
"--path"
) # trouve/créer le json assetlib.json en sous-dossier de libdir
if '--' in sys.argv : if "--" in sys.argv:
index = sys.argv.index('--') index = sys.argv.index("--")
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()

View File

@ -1,4 +1,3 @@
import bpy import bpy
@ -9,6 +8,6 @@ def draw_context_menu(layout):
def draw_header(layout): def draw_header(layout):
'''Draw the header of the Asset Browser Window''' """Draw the header of the Asset Browser Window"""
return return

View File

@ -1,11 +1,10 @@
from typing import List, Tuple from typing import List, Tuple
import bpy import bpy
addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = [] addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = []
def register(): def register():
wm = bpy.context.window_manager wm = bpy.context.window_manager
addon = wm.keyconfigs.addon addon = wm.keyconfigs.addon
@ -13,9 +12,12 @@ def register():
return return
km = addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER") km = addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER")
kmi = km.keymap_items.new("assetlib.load_asset", "LEFTMOUSE", "DOUBLE_CLICK") # , shift=True kmi = km.keymap_items.new(
"assetlib.load_asset", "LEFTMOUSE", "DOUBLE_CLICK"
) # , shift=True
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
def unregister(): def unregister():
for km, kmi in addon_keymaps: for km, kmi in addon_keymaps:
km.keymap_items.remove(kmi) km.keymap_items.remove(kmi)

View File

@ -14,12 +14,11 @@ from asset_library.common.bl_utils import load_col
from asset_library.common.functions import get_active_library from asset_library.common.functions import get_active_library
class ASSETLIB_OT_load_asset(Operator): class ASSETLIB_OT_load_asset(Operator):
bl_idname = "assetlib.load_asset" bl_idname = "assetlib.load_asset"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Load Asset' bl_label = "Load Asset"
bl_description = 'Link and override asset in current file' bl_description = "Link and override asset in current file"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
@ -28,10 +27,10 @@ class ASSETLIB_OT_load_asset(Operator):
return False return False
lib = get_active_library() lib = get_active_library()
if not lib or lib.data_type != 'COLLECTION': if not lib or lib.data_type != "COLLECTION":
return False return False
if not context.active_file or 'filepath' not in context.active_file.asset_data: if not context.active_file or "filepath" not in context.active_file.asset_data:
cls.poll_message_set("Has not filepath property") cls.poll_message_set("Has not filepath property")
return False return False
@ -39,51 +38,49 @@ class ASSETLIB_OT_load_asset(Operator):
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
print('Load Asset') print("Load Asset")
lib = get_active_library() lib = get_active_library()
asset = context.active_file asset = context.active_file
if not asset: if not asset:
self.report({"ERROR"}, 'No asset selected') self.report({"ERROR"}, "No asset selected")
return {'CANCELLED'} return {"CANCELLED"}
active_lib = lib.library_type.get_active_asset_library() active_lib = lib.library_type.get_active_asset_library()
asset_path = asset.asset_data['filepath'] asset_path = asset.asset_data["filepath"]
asset_path = active_lib.library_type.format_path(asset_path) asset_path = active_lib.library_type.format_path(asset_path)
name = asset.name name = asset.name
## set mode to object ## set mode to object
if context.mode != 'OBJECT': if context.mode != "OBJECT":
bpy.ops.object.mode_set(mode='OBJECT') bpy.ops.object.mode_set(mode="OBJECT")
if not Path(asset_path).exists(): if not Path(asset_path).exists():
self.report({'ERROR'}, f'Not exists: {asset_path}') self.report({"ERROR"}, f"Not exists: {asset_path}")
return {'CANCELLED'} return {"CANCELLED"}
print('Load collection', asset_path, name) print("Load collection", asset_path, name)
res = load_col(asset_path, name, link=True, override=True, rig_pattern='*_rig') res = load_col(asset_path, name, link=True, override=True, rig_pattern="*_rig")
if res: if res:
if res.type == 'ARMATURE': if res.type == "ARMATURE":
self.report({'INFO'}, f'Override rig {res.name}') self.report({"INFO"}, f"Override rig {res.name}")
elif res.type == 'EMPTY': elif res.type == "EMPTY":
self.report({'INFO'}, f'Instance collection {res.name}') self.report({"INFO"}, f"Instance collection {res.name}")
return {'FINISHED'} return {"FINISHED"}
### --- REGISTER --- ### --- REGISTER ---
classes = ( classes = (ASSETLIB_OT_load_asset,)
ASSETLIB_OT_load_asset,
)
def register(): def register():
for cls in classes: for cls in classes:
bpy.utils.register_class(cls) bpy.utils.register_class(cls)
def unregister(): def unregister():
for cls in reversed(classes): for cls in reversed(classes):
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(cls)

View File

@ -1,12 +1,10 @@
from asset_library.common import file_utils from asset_library.common import file_utils
from asset_library.common import functions from asset_library.common import functions
from asset_library.common import synchronize from asset_library.common import synchronize
from asset_library.common import template from asset_library.common import template
from asset_library.common import catalog from asset_library.common import catalog
if 'bpy' in locals(): if "bpy" in locals():
import importlib import importlib
importlib.reload(file_utils) importlib.reload(file_utils)

View File

@ -1,4 +1,3 @@
""" """
Generic Blender functions Generic Blender functions
""" """
@ -6,21 +5,23 @@ Generic Blender functions
from pathlib import Path from pathlib import Path
from fnmatch import fnmatch from fnmatch import fnmatch
from typing import Any, List, Iterable, Optional, Tuple from typing import Any, List, Iterable, Optional, Tuple
Datablock = Any Datablock = Any
import bpy import bpy
from bpy_extras import asset_utils from bpy_extras import asset_utils
from asset_library.constants import RESOURCES_DIR from asset_library.constants import RESOURCES_DIR
# from asset_library.common.file_utils import no # from asset_library.common.file_utils import no
from os.path import abspath from os.path import abspath
import subprocess import subprocess
class attr_set(): class attr_set:
'''Receive a list of tuple [(data_path, "attribute" [, wanted value)] ] """Receive a list of tuple [(data_path, "attribute" [, wanted value)] ]
entering with-statement : Store existing values, assign wanted value (if any) entering with-statement : Store existing values, assign wanted value (if any)
exiting with-statement: Restore values to their old values exiting with-statement: Restore values to their old values
''' """
def __init__(self, attrib_list): def __init__(self, attrib_list):
self.store = [] self.store = []
@ -36,7 +37,7 @@ class attr_set():
try: try:
setattr(prop, attr, item[2]) setattr(prop, attr, item[2])
except TypeError: except TypeError:
print(f'Cannot set attribute {attr} to {prop}') print(f"Cannot set attribute {attr} to {prop}")
def __enter__(self): def __enter__(self):
return self return self
@ -48,24 +49,37 @@ class attr_set():
for prop, attr, old_val in self.store: for prop, attr, old_val in self.store:
setattr(prop, attr, old_val) setattr(prop, attr, old_val)
def get_overriden_col(ob, scene=None): def get_overriden_col(ob, scene=None):
scn = scene or bpy.context.scene scn = scene or bpy.context.scene
cols = [c for c in bpy.data.collections if scn.user_of_id(c)] cols = [c for c in bpy.data.collections if scn.user_of_id(c)]
return next((c for c in cols if ob in c.all_objects[:] return next(
if all(not c.override_library for c in get_col_parents(c))), None) (
c
for c in cols
if ob in c.all_objects[:]
if all(not c.override_library for c in get_col_parents(c))
),
None,
)
def get_view3d_persp(): def get_view3d_persp():
windows = bpy.context.window_manager.windows windows = bpy.context.window_manager.windows
view_3ds = [a for w in windows for a in w.screen.areas if a.type == 'VIEW_3D'] view_3ds = [a for w in windows for a in w.screen.areas if a.type == "VIEW_3D"]
view_3d = next((a for a in view_3ds if a.spaces.active.region_3d.view_perspective == 'PERSP'), view_3ds[0]) view_3d = next(
(a for a in view_3ds if a.spaces.active.region_3d.view_perspective == "PERSP"),
view_3ds[0],
)
return view_3d return view_3d
def get_viewport(): def get_viewport():
screen = bpy.context.screen screen = bpy.context.screen
areas = [a for a in screen.areas if a.type == 'VIEW_3D'] areas = [a for a in screen.areas if a.type == "VIEW_3D"]
areas.sort(key=lambda x: x.width * x.height) areas.sort(key=lambda x: x.width * x.height)
return areas[-1] return areas[-1]
@ -81,7 +95,7 @@ def biggest_asset_browser_area(screen: bpy.types.Screen) -> Optional[bpy.types.A
def area_sorting_key(area: bpy.types.Area) -> Tuple[bool, int]: def area_sorting_key(area: bpy.types.Area) -> Tuple[bool, int]:
"""Return area size in pixels.""" """Return area size in pixels."""
return (area.width * area.height) return area.width * area.height
areas = list(suitable_areas(screen)) areas = list(suitable_areas(screen))
if not areas: if not areas:
@ -89,6 +103,7 @@ def biggest_asset_browser_area(screen: bpy.types.Screen) -> Optional[bpy.types.A
return max(areas, key=area_sorting_key) return max(areas, key=area_sorting_key)
def suitable_areas(screen: bpy.types.Screen) -> Iterable[bpy.types.Area]: def suitable_areas(screen: bpy.types.Screen) -> Iterable[bpy.types.Area]:
"""Generator, yield Asset Browser areas.""" """Generator, yield Asset Browser areas."""
@ -98,6 +113,7 @@ def suitable_areas(screen: bpy.types.Screen) -> Iterable[bpy.types.Area]:
continue continue
yield area yield area
def area_from_context(context: bpy.types.Context) -> Optional[bpy.types.Area]: def area_from_context(context: bpy.types.Context) -> Optional[bpy.types.Area]:
"""Return an Asset Browser suitable for the given category. """Return an Asset Browser suitable for the given category.
@ -122,6 +138,7 @@ def area_from_context(context: bpy.types.Context) -> Optional[bpy.types.Area]:
return None return None
def activate_asset( def activate_asset(
asset: bpy.types.Action, asset_browser: bpy.types.Area, *, deferred: bool asset: bpy.types.Action, asset_browser: bpy.types.Area, *, deferred: bool
) -> None: ) -> None:
@ -131,19 +148,25 @@ def activate_asset(
assert asset_utils.SpaceAssetInfo.is_asset_browser(space_data) assert asset_utils.SpaceAssetInfo.is_asset_browser(space_data)
space_data.activate_asset_by_id(asset, deferred=deferred) space_data.activate_asset_by_id(asset, deferred=deferred)
def active_catalog_id(asset_browser: bpy.types.Area) -> str: def active_catalog_id(asset_browser: bpy.types.Area) -> str:
"""Return the ID of the catalog shown in the asset browser.""" """Return the ID of the catalog shown in the asset browser."""
return params(asset_browser).catalog_id return params(asset_browser).catalog_id
def get_asset_space_params(asset_browser: bpy.types.Area) -> bpy.types.FileAssetSelectParams:
def get_asset_space_params(
asset_browser: bpy.types.Area,
) -> bpy.types.FileAssetSelectParams:
"""Return the asset browser parameters given its Area.""" """Return the asset browser parameters given its Area."""
space_data = asset_browser.spaces[0] space_data = asset_browser.spaces[0]
assert asset_utils.SpaceAssetInfo.is_asset_browser(space_data) assert asset_utils.SpaceAssetInfo.is_asset_browser(space_data)
return space_data.params return space_data.params
def refresh_asset_browsers(): def refresh_asset_browsers():
for area in suitable_areas(bpy.context.screen): for area in suitable_areas(bpy.context.screen):
bpy.ops.asset.library_refresh({"area": area, 'region': area.regions[3]}) bpy.ops.asset.library_refresh({"area": area, "region": area.regions[3]})
def tag_redraw(screen: bpy.types.Screen) -> None: def tag_redraw(screen: bpy.types.Screen) -> None:
"""Tag all asset browsers for redrawing.""" """Tag all asset browsers for redrawing."""
@ -151,6 +174,7 @@ def tag_redraw(screen: bpy.types.Screen) -> None:
for area in suitable_areas(screen): for area in suitable_areas(screen):
area.tag_redraw() area.tag_redraw()
# def get_blender_command(file=None, script=None, background=True, **args): # def get_blender_command(file=None, script=None, background=True, **args):
# '''Return a Blender Command as a list to be used in a subprocess''' # '''Return a Blender Command as a list to be used in a subprocess'''
@ -169,6 +193,7 @@ def tag_redraw(screen: bpy.types.Screen) -> None:
# return cmd # return cmd
def norm_value(value): def norm_value(value):
if isinstance(value, (tuple, list)): if isinstance(value, (tuple, list)):
values = [] values = []
@ -186,31 +211,35 @@ def norm_value(value):
value = json.dumps(value) value = json.dumps(value)
return value return value
def norm_arg(arg_name, format=str.lower, prefix='--', separator='-'):
def norm_arg(arg_name, format=str.lower, prefix="--", separator="-"):
arg_name = norm_str(arg_name, format=format, separator=separator) arg_name = norm_str(arg_name, format=format, separator=separator)
return prefix + arg_name return prefix + arg_name
def get_bl_cmd(blender=None, background=False, focus=True, blendfile=None, script=None, **kargs):
def get_bl_cmd(
blender=None, background=False, focus=True, blendfile=None, script=None, **kargs
):
cmd = [str(blender)] if blender else [bpy.app.binary_path] cmd = [str(blender)] if blender else [bpy.app.binary_path]
if background: if background:
cmd += ['--background'] cmd += ["--background"]
if not focus and not background: if not focus and not background:
cmd += ['--no-window-focus'] cmd += ["--no-window-focus"]
cmd += ['--window-geometry', '5000', '0', '10', '10'] cmd += ["--window-geometry", "5000", "0", "10", "10"]
cmd += ['--python-use-system-env'] cmd += ["--python-use-system-env"]
if blendfile: if blendfile:
cmd += [str(blendfile)] cmd += [str(blendfile)]
if script: if script:
cmd += ['--python', str(script)] cmd += ["--python", str(script)]
if kargs: if kargs:
cmd += ['--'] cmd += ["--"]
for k, v in kargs.items(): for k, v in kargs.items():
k = norm_arg(k) k = norm_arg(k)
v = norm_value(v) v = norm_value(v)
@ -223,18 +252,18 @@ def get_bl_cmd(blender=None, background=False, focus=True, blendfile=None, scrip
return cmd return cmd
def get_addon_prefs():
addon_name = __package__.split('.')[0]
return bpy.context.preferences.addons[addon_name].preferences
def get_addon_prefs():
addon_name = __package__.split(".")[0]
return bpy.context.preferences.addons[addon_name].preferences
def thumbnail_blend_file(input_blend, output_img): def thumbnail_blend_file(input_blend, output_img):
input_blend = Path(input_blend).resolve() input_blend = Path(input_blend).resolve()
output_img = Path(output_img).resolve() output_img = Path(output_img).resolve()
print(f'Thumbnailing {input_blend} to {output_img}') print(f"Thumbnailing {input_blend} to {output_img}")
blender_thumbnailer = Path(bpy.app.binary_path).parent / 'blender-thumbnailer' blender_thumbnailer = Path(bpy.app.binary_path).parent / "blender-thumbnailer"
output_img.parent.mkdir(exist_ok=True, parents=True) output_img.parent.mkdir(exist_ok=True, parents=True)
@ -243,16 +272,17 @@ def thumbnail_blend_file(input_blend, output_img):
success = output_img.exists() success = output_img.exists()
if not success: if not success:
empty_preview = RESOURCES_DIR / 'empty_preview.png' empty_preview = RESOURCES_DIR / "empty_preview.png"
shutil.copy(str(empty_preview), str(output_img)) shutil.copy(str(empty_preview), str(output_img))
return success return success
def get_col_parents(col, root=None, cols=None): def get_col_parents(col, root=None, cols=None):
'''Return a list of parents collections of passed col """Return a list of parents collections of passed col
root : Pass a collection to search in (recursive) root : Pass a collection to search in (recursive)
else search in master collection else search in master collection
''' """
if cols is None: if cols is None:
cols = [] cols = []
@ -267,14 +297,23 @@ def get_col_parents(col, root=None, cols=None):
cols = get_col_parents(col, root=sub, cols=cols) cols = get_col_parents(col, root=sub, cols=cols)
return cols return cols
def get_overriden_col(ob, scene=None): def get_overriden_col(ob, scene=None):
'''Get the collection use for making the override''' """Get the collection use for making the override"""
scn = scene or bpy.context.scene scn = scene or bpy.context.scene
cols = [c for c in bpy.data.collections if scn.user_of_id(c)] cols = [c for c in bpy.data.collections if scn.user_of_id(c)]
return next((c for c in cols if ob in c.all_objects[:] return next(
if all(not c.override_library for c in get_col_parents(c))), None) (
c
for c in cols
if ob in c.all_objects[:]
if all(not c.override_library for c in get_col_parents(c))
),
None,
)
def load_assets_from(filepath: Path) -> List[Datablock]: def load_assets_from(filepath: Path) -> List[Datablock]:
if not has_assets(filepath): if not has_assets(filepath):
@ -306,6 +345,7 @@ def load_assets_from(filepath: Path) -> List[Datablock]:
loaded_assets.append(datablock) loaded_assets.append(datablock)
return loaded_assets return loaded_assets
def has_assets(filepath: Path) -> bool: def has_assets(filepath: Path) -> bool:
with bpy.data.libraries.load(str(filepath), assets_only=True) as ( with bpy.data.libraries.load(str(filepath), assets_only=True) as (
data_from, data_from,
@ -318,17 +358,13 @@ def has_assets(filepath: Path) -> bool:
return False return False
def copy_frames(start, end, offset, path): def copy_frames(start, end, offset, path):
for i in range(start, end): for i in range(start, end):
src = path.replace('####', f'{i:04d}') src = path.replace("####", f"{i:04d}")
dst = src.replace(src.split('_')[-1].split('.')[0], f'{i+offset:04d}') dst = src.replace(src.split("_")[-1].split(".")[0], f"{i+offset:04d}")
shutil.copy2(src, dst) shutil.copy2(src, dst)
def split_path(path): def split_path(path):
try: try:
bone_name = path.split('["')[1].split('"]')[0] bone_name = path.split('["')[1].split('"]')[0]
@ -337,15 +373,14 @@ def split_path(path) :
try: try:
prop_name = path.split('["')[2].split('"]')[0] prop_name = path.split('["')[2].split('"]')[0]
except: except:
prop_name = path.split('.')[-1] prop_name = path.split(".")[-1]
return bone_name, prop_name return bone_name, prop_name
def load_datablocks(
src, names=None, type="objects", link=True, expr=None, assets_only=False
) -> list:
def load_datablocks(src, names=None, type='objects', link=True, expr=None, assets_only=False) -> list:
return_list = not isinstance(names, str) return_list = not isinstance(names, str)
names = names or [] names = names or []
@ -356,7 +391,10 @@ def load_datablocks(src, names=None, type='objects', link=True, expr=None, asset
pattern = expr pattern = expr
expr = lambda x: fnmatch(x, pattern) expr = lambda x: fnmatch(x, pattern)
with bpy.data.libraries.load(str(src), link=link,assets_only=assets_only) as (data_from, data_to): with bpy.data.libraries.load(str(src), link=link, assets_only=assets_only) as (
data_from,
data_to,
):
datablocks = getattr(data_from, type) datablocks = getattr(data_from, type)
if expr: if expr:
names += [i for i in datablocks if expr(i)] names += [i for i in datablocks if expr(i)]
@ -373,23 +411,26 @@ def load_datablocks(src, names=None, type='objects', link=True, expr=None, asset
elif datablocks: elif datablocks:
return datablocks[0] return datablocks[0]
""" """
# --- Collection handling # --- Collection handling
""" """
def col_as_asset(col, verbose=False): def col_as_asset(col, verbose=False):
if col is None: if col is None:
return return
if verbose: if verbose:
print('linking:', col.name) print("linking:", col.name)
pcol = bpy.data.collections.new(col.name) pcol = bpy.data.collections.new(col.name)
bpy.context.scene.collection.children.link(pcol) bpy.context.scene.collection.children.link(pcol)
pcol.children.link(col) pcol.children.link(col)
pcol.asset_mark() pcol.asset_mark()
return pcol return pcol
def load_col(filepath, name, link=True, override=True, rig_pattern=None, context=None): def load_col(filepath, name, link=True, override=True, rig_pattern=None, context=None):
'''Link a collection by name from a file and override if has armature''' """Link a collection by name from a file and override if has armature"""
# with bpy.data.libraries.load(filepath, link=link) as (data_from, data_to): # with bpy.data.libraries.load(filepath, link=link) as (data_from, data_to):
# data_to.collections = [c for c in data_from.collections if c == name] # data_to.collections = [c for c in data_from.collections if c == name]
@ -398,12 +439,12 @@ def load_col(filepath, name, link=True, override=True, rig_pattern=None, context
# return data_to.collections[0] # return data_to.collections[0]
context = context or bpy.context context = context or bpy.context
col = load_datablocks(filepath, name, link=link, type='collections') col = load_datablocks(filepath, name, link=link, type="collections")
## create instance object ## create instance object
inst = bpy.data.objects.new(col.name, None) inst = bpy.data.objects.new(col.name, None)
inst.instance_collection = col inst.instance_collection = col
inst.instance_type = 'COLLECTION' inst.instance_type = "COLLECTION"
context.scene.collection.objects.link(inst) context.scene.collection.objects.link(inst)
# make active # make active
@ -413,7 +454,7 @@ def load_col(filepath, name, link=True, override=True, rig_pattern=None, context
## simple object (no armatures) ## simple object (no armatures)
if not link or not override: if not link or not override:
return inst return inst
if not next((o for o in col.all_objects if o.type == 'ARMATURE'), None): if not next((o for o in col.all_objects if o.type == "ARMATURE"), None):
return inst return inst
## Create the override ## Create the override
@ -421,18 +462,21 @@ def load_col(filepath, name, link=True, override=True, rig_pattern=None, context
parent_cols = inst.users_collection parent_cols = inst.users_collection
child_cols = [child for pcol in parent_cols for child in pcol.children] child_cols = [child for pcol in parent_cols for child in pcol.children]
params = {'active_object': inst, 'selected_objects': [inst]} params = {"active_object": inst, "selected_objects": [inst]}
try: try:
bpy.ops.object.make_override_library(params) bpy.ops.object.make_override_library(params)
## check which collection is new in parents collection ## check which collection is new in parents collection
asset_col = next((c for pcol in parent_cols for c in pcol.children if c not in child_cols), None) asset_col = next(
(c for pcol in parent_cols for c in pcol.children if c not in child_cols),
None,
)
if not asset_col: if not asset_col:
print('Overriden, but no collection found !!') print("Overriden, but no collection found !!")
return return
for ob in asset_col.all_objects: for ob in asset_col.all_objects:
if ob.type != 'ARMATURE': if ob.type != "ARMATURE":
continue continue
if rig_pattern and not fnmatch(ob.name, rig_pattern): if rig_pattern and not fnmatch(ob.name, rig_pattern):
continue continue
@ -444,16 +488,19 @@ def load_col(filepath, name, link=True, override=True, rig_pattern=None, context
return ob return ob
except Exception as e: except Exception as e:
print(f'Override failed on {col.name}') print(f"Override failed on {col.name}")
print(e) print(e)
return inst return inst
def get_preview(asset_path='', asset_name=''): def get_preview(asset_path="", asset_name=""):
asset_preview_dir = Path(asset_path).parents[1] asset_preview_dir = Path(asset_path).parents[1]
name = asset_name.lower() name = asset_name.lower()
return next((f for f in asset_preview_dir.rglob('*') if f.stem.lower().endswith(name)), None) return next(
(f for f in asset_preview_dir.rglob("*") if f.stem.lower().endswith(name)), None
)
def get_object_libraries(ob): def get_object_libraries(ob):
if ob is None: if ob is None:
@ -463,7 +510,7 @@ def get_object_libraries(ob):
if ob.data: if ob.data:
libraries += [ob.data.library] libraries += [ob.data.library]
if ob.type in ('MESH', 'CURVE'): if ob.type in ("MESH", "CURVE"):
libraries += [m.library for m in ob.data.materials if m] libraries += [m.library for m in ob.data.materials if m]
filepaths = [] filepaths = []

View File

@ -1,4 +1,3 @@
from pathlib import Path from pathlib import Path
import uuid import uuid
import bpy import bpy
@ -6,6 +5,7 @@ import bpy
class CatalogItem: class CatalogItem:
"""Represent a single item of a catalog""" """Represent a single item of a catalog"""
def __init__(self, catalog, path=None, name=None, id=None): def __init__(self, catalog, path=None, name=None, id=None):
self.catalog = catalog self.catalog = catalog
@ -25,10 +25,10 @@ class CatalogItem:
def norm_name(self, name): def norm_name(self, name):
"""Get a norm name from a catalog_path entry""" """Get a norm name from a catalog_path entry"""
return name.replace('/', '-') return name.replace("/", "-")
def __repr__(self): def __repr__(self):
return f'CatalogItem(name={self.name}, path={self.path}, id={self.id})' return f"CatalogItem(name={self.name}, path={self.path}, id={self.id})"
class CatalogContext: class CatalogContext:
@ -60,11 +60,12 @@ class CatalogContext:
if self.active_item: if self.active_item:
return self.active_item.path return self.active_item.path
return '' return ""
class Catalog: class Catalog:
"""Represent the catalog of the blender asset browser library""" """Represent the catalog of the blender asset browser library"""
def __init__(self, directory=None): def __init__(self, directory=None):
self.directory = None self.directory = None
@ -79,7 +80,7 @@ class Catalog:
def filepath(self): def filepath(self):
"""Get the filepath of the catalog text file relative to the directory""" """Get the filepath of the catalog text file relative to the directory"""
if self.directory: if self.directory:
return self.directory /'blender_assets.cats.txt' return self.directory / "blender_assets.cats.txt"
def read(self): def read(self):
"""Read the catalog file of the library target directory or of the specified directory""" """Read the catalog file of the library target directory or of the specified directory"""
@ -89,13 +90,15 @@ class Catalog:
self._data.clear() self._data.clear()
print(f'Read catalog from {self.filepath}') print(f"Read catalog from {self.filepath}")
for line in self.filepath.read_text(encoding="utf-8").split('\n'): for line in self.filepath.read_text(encoding="utf-8").split("\n"):
if line.startswith(('VERSION', '#')) or not line: if line.startswith(("VERSION", "#")) or not line:
continue continue
cat_id, cat_path, cat_name = line.split(':') cat_id, cat_path, cat_name = line.split(":")
self._data[cat_id] = CatalogItem(self, name=cat_name, id=cat_id, path=cat_path) self._data[cat_id] = CatalogItem(
self, name=cat_name, id=cat_id, path=cat_path
)
return self return self
@ -103,9 +106,9 @@ class Catalog:
"""Write the catalog file in the library target directory or of the specified directory""" """Write the catalog file in the library target directory or of the specified directory"""
if not self.filepath: if not self.filepath:
raise Exception(f'Cannot write catalog {self} no filepath setted') raise Exception(f"Cannot write catalog {self} no filepath setted")
lines = ['VERSION 1', ''] lines = ["VERSION 1", ""]
catalog_items = list(self) catalog_items = list(self)
if sort: if sort:
@ -114,8 +117,8 @@ class Catalog:
for catalog_item in catalog_items: for catalog_item in catalog_items:
lines.append(f"{catalog_item.id}:{catalog_item.path}:{catalog_item.name}") lines.append(f"{catalog_item.id}:{catalog_item.path}:{catalog_item.name}")
print(f'Write Catalog at: {self.filepath}') print(f"Write Catalog at: {self.filepath}")
self.filepath.write_text('\n'.join(lines), encoding="utf-8") self.filepath.write_text("\n".join(lines), encoding="utf-8")
def get(self, path=None, id=None, fallback=None): def get(self, path=None, id=None, fallback=None):
"""Found a catalog item by is path or id""" """Found a catalog item by is path or id"""
@ -140,7 +143,7 @@ class Catalog:
if catalog_item: if catalog_item:
return self._data.pop(catalog_item.id) return self._data.pop(catalog_item.id)
print(f'Warning: {catalog_item} cannot be remove, not in {self}') print(f"Warning: {catalog_item} cannot be remove, not in {self}")
return None return None
def add(self, catalog_path): def add(self, catalog_path):
@ -163,7 +166,7 @@ class Catalog:
return cat_item return cat_item
def update(self, catalogs): def update(self, catalogs):
'Add or remove catalog entries if on the list given or not' "Add or remove catalog entries if on the list given or not"
catalogs = set(catalogs) # Remove doubles catalogs = set(catalogs) # Remove doubles
@ -171,9 +174,13 @@ class Catalog:
removed = [c.path for c in self if c.path not in catalogs] removed = [c.path for c in self if c.path not in catalogs]
if added: if added:
print(f'{len(added)} Catalog Entry Added \n{tuple(c.name for c in added[:10])}...\n') print(
f"{len(added)} Catalog Entry Added \n{tuple(c.name for c in added[:10])}...\n"
)
if removed: if removed:
print(f'{len(removed)} Catalog Entry Removed \n{tuple(c.name for c in removed[:10])}...\n') print(
f"{len(removed)} Catalog Entry Removed \n{tuple(c.name for c in removed[:10])}...\n"
)
for catalog_item in removed: for catalog_item in removed:
self.remove(catalog_item) self.remove(catalog_item)
@ -197,4 +204,4 @@ class Catalog:
return item in self return item in self
def __repr__(self): def __repr__(self):
return f'Catalog(filepath={self.filepath})' return f"Catalog(filepath={self.filepath})"

View File

@ -1,4 +1,3 @@
"""Generic python functions to make operation on file and names""" """Generic python functions to make operation on file and names"""
import fnmatch import fnmatch
@ -15,6 +14,7 @@ import shutil
import contextlib import contextlib
@contextlib.contextmanager @contextlib.contextmanager
def cd(path): def cd(path):
"""Changes working directory and returns to previous on exit.""" """Changes working directory and returns to previous on exit."""
@ -25,20 +25,24 @@ def cd(path):
finally: finally:
os.chdir(prev_cwd) os.chdir(prev_cwd)
def install_module(module_name, package_name=None): def install_module(module_name, package_name=None):
'''Install a python module with pip or return it if already installed''' """Install a python module with pip or return it if already installed"""
try: try:
module = importlib.import_module(module_name) module = importlib.import_module(module_name)
except ModuleNotFoundError: except ModuleNotFoundError:
print(f'Installing Module {module_name} ....') print(f"Installing Module {module_name} ....")
subprocess.call([sys.executable, '-m', 'ensurepip']) subprocess.call([sys.executable, "-m", "ensurepip"])
subprocess.call([sys.executable, '-m', 'pip', 'install', package_name or module_name]) subprocess.call(
[sys.executable, "-m", "pip", "install", package_name or module_name]
)
module = importlib.import_module(module_name) module = importlib.import_module(module_name)
return module return module
def import_module_from_path(path): def import_module_from_path(path):
from importlib import util from importlib import util
@ -51,45 +55,59 @@ def import_module_from_path(path):
return mod return mod
except Exception as e: except Exception as e:
print(f'Cannot import file {path}') print(f"Cannot import file {path}")
print(e) print(e)
def norm_str(string, separator='_', format=str.lower, padding=0):
def norm_str(string, separator="_", format=str.lower, padding=0):
string = str(string) string = str(string)
string = string.replace('_', ' ') string = string.replace("_", " ")
string = string.replace('-', ' ') string = string.replace("-", " ")
string = re.sub('[ ]+', ' ', string) string = re.sub("[ ]+", " ", string)
string = re.sub('[ ]+\/[ ]+', '/', string) string = re.sub("[ ]+\/[ ]+", "/", string)
string = string.strip() string = string.strip()
if format: if format:
string = format(string) string = format(string)
# Padd rightest number # Padd rightest number
string = re.sub(r'(\d+)(?!.*\d)', lambda x : x.group(1).zfill(padding), string) string = re.sub(r"(\d+)(?!.*\d)", lambda x: x.group(1).zfill(padding), string)
string = string.replace(' ', separator) string = string.replace(" ", separator)
string = unicodedata.normalize('NFKD', string).encode('ASCII', 'ignore').decode("utf-8") string = (
unicodedata.normalize("NFKD", string).encode("ASCII", "ignore").decode("utf-8")
)
return string return string
def remove_version(filepath): def remove_version(filepath):
pattern = '_v[0-9]+\.' pattern = "_v[0-9]+\."
search = re.search(pattern, filepath) search = re.search(pattern, filepath)
if search: if search:
filepath = filepath.replace(search.group()[:-1], '') filepath = filepath.replace(search.group()[:-1], "")
return Path(filepath).name return Path(filepath).name
def is_exclude(name, patterns) -> bool: def is_exclude(name, patterns) -> bool:
# from fnmatch import fnmatch # from fnmatch import fnmatch
if not isinstance(patterns, (list, tuple)): if not isinstance(patterns, (list, tuple)):
patterns = [patterns] patterns = [patterns]
return any([fnmatch(name, p) for p in patterns]) return any([fnmatch(name, p) for p in patterns])
def get_last_files(root, pattern=r'_v\d{3}\.\w+', only_matching=False, ex_file=None, ex_dir=None, keep=1, verbose=False) -> list:
'''Recursively get last(s) file(s) (when there is multiple versions) in passed directory def get_last_files(
root,
pattern=r"_v\d{3}\.\w+",
only_matching=False,
ex_file=None,
ex_dir=None,
keep=1,
verbose=False,
) -> list:
"""Recursively get last(s) file(s) (when there is multiple versions) in passed directory
root -> str: Filepath of the folder to scan. root -> str: Filepath of the folder to scan.
pattern -> str: Regex pattern to group files. pattern -> str: Regex pattern to group files.
only_matching -> bool: Discard files that aren't matched by regex pattern. only_matching -> bool: Discard files that aren't matched by regex pattern.
@ -97,7 +115,7 @@ def get_last_files(root, pattern=r'_v\d{3}\.\w+', only_matching=False, ex_file=N
ex_dir -> list : List of fn_match pattern of directory name to skip. ex_dir -> list : List of fn_match pattern of directory name to skip.
keep -> int: Number of lasts versions to keep when there are mutliple versionned files (e.g: 1 keep only last). keep -> int: Number of lasts versions to keep when there are mutliple versionned files (e.g: 1 keep only last).
verbose -> bool: Print infos in console. verbose -> bool: Print infos in console.
''' """
files = [] files = []
if ex_file is None: if ex_file is None:
@ -111,7 +129,9 @@ def get_last_files(root, pattern=r'_v\d{3}\.\w+', only_matching=False, ex_file=N
dirs = [f for f in all_items if f.is_dir()] dirs = [f for f in all_items if f.is_dir()]
for i in range(len(allfiles)-1,-1,-1):# fastest way to iterate on index in reverse for i in range(
len(allfiles) - 1, -1, -1
): # fastest way to iterate on index in reverse
if not re.search(pattern, allfiles[i].name): if not re.search(pattern, allfiles[i].name):
if only_matching: if only_matching:
allfiles.pop(i) allfiles.pop(i)
@ -119,7 +139,10 @@ def get_last_files(root, pattern=r'_v\d{3}\.\w+', only_matching=False, ex_file=N
files.append(allfiles.pop(i).path) files.append(allfiles.pop(i).path)
# separate remaining files in prefix grouped lists # separate remaining files in prefix grouped lists
lilist = [list(v) for k, v in groupby(allfiles, key=lambda x: re.split(pattern, x.name)[0])] lilist = [
list(v)
for k, v in groupby(allfiles, key=lambda x: re.split(pattern, x.name)[0])
]
# get only item last of each sorted grouplist # get only item last of each sorted grouplist
for l in lilist: for l in lilist:
@ -128,17 +151,26 @@ def get_last_files(root, pattern=r'_v\d{3}\.\w+', only_matching=False, ex_file=N
files.append(f.path) files.append(f.path)
if verbose and len(l) > 1: if verbose and len(l) > 1:
print(f'{root}: keep {str([x.name for x in versions])} out of {len(l)} elements') print(
f"{root}: keep {str([x.name for x in versions])} out of {len(l)} elements"
)
for d in dirs: # recursively treat all detected directory for d in dirs: # recursively treat all detected directory
if ex_dir and is_exclude(d.name, ex_dir): if ex_dir and is_exclude(d.name, ex_dir):
# skip folder with excluded name # skip folder with excluded name
continue continue
files += get_last_files( files += get_last_files(
d.path, pattern=pattern, only_matching=only_matching, ex_file=ex_file, ex_dir=ex_dir, keep=keep) d.path,
pattern=pattern,
only_matching=only_matching,
ex_file=ex_file,
ex_dir=ex_dir,
keep=keep,
)
return sorted(files) return sorted(files)
def copy_file(src, dst, only_new=False, only_recent=False): def copy_file(src, dst, only_new=False, only_recent=False):
if dst.exists(): if dst.exists():
if only_new: if only_new:
@ -147,19 +179,20 @@ def copy_file(src, dst, only_new=False, only_recent=False):
return return
dst.parent.mkdir(exist_ok=True, parents=True) dst.parent.mkdir(exist_ok=True, parents=True)
print(f'Copy file from {src} to {dst}') print(f"Copy file from {src} to {dst}")
if platform.system() == 'Windows': if platform.system() == "Windows":
subprocess.call(['copy', str(src), str(dst)], shell=True) subprocess.call(["copy", str(src), str(dst)], shell=True)
else: else:
subprocess.call(['cp', str(src), str(dst)]) subprocess.call(["cp", str(src), str(dst)])
def copy_dir(src, dst, only_new=False, only_recent=False, excludes=['.*'], includes=[]):
def copy_dir(src, dst, only_new=False, only_recent=False, excludes=[".*"], includes=[]):
src, dst = Path(src), Path(dst) src, dst = Path(src), Path(dst)
if includes: if includes:
includes = r'|'.join([fnmatch.translate(x) for x in includes]) includes = r"|".join([fnmatch.translate(x) for x in includes])
if excludes: if excludes:
excludes = r'|'.join([fnmatch.translate(x) for x in excludes]) excludes = r"|".join([fnmatch.translate(x) for x in excludes])
if dst.is_dir(): if dst.is_dir():
dst.mkdir(exist_ok=True, parents=True) dst.mkdir(exist_ok=True, parents=True)
@ -170,7 +203,7 @@ def copy_dir(src, dst, only_new=False, only_recent=False, excludes=['.*'], inclu
copy_file(src, dst, only_new=only_new, only_recent=only_recent) copy_file(src, dst, only_new=only_new, only_recent=only_recent)
elif src.is_dir(): elif src.is_dir():
src_files = list(src.rglob('*')) src_files = list(src.rglob("*"))
if excludes: if excludes:
src_files = [f for f in src_files if not re.match(excludes, f.name)] src_files = [f for f in src_files if not re.match(excludes, f.name)]
@ -183,110 +216,115 @@ def copy_dir(src, dst, only_new=False, only_recent=False, excludes=['.*'], inclu
if src_file.is_dir(): if src_file.is_dir():
dst_file.mkdir(exist_ok=True, parents=True) dst_file.mkdir(exist_ok=True, parents=True)
else: else:
copy_file(src_file, dst_file, only_new=only_new, only_recent=only_recent) copy_file(
src_file, dst_file, only_new=only_new, only_recent=only_recent
)
def open_file(filepath, select=False): def open_file(filepath, select=False):
'''Open a filepath inside the os explorer''' """Open a filepath inside the os explorer"""
if platform.system() == 'Darwin': # macOS if platform.system() == "Darwin": # macOS
cmd = ['open'] cmd = ["open"]
elif platform.system() == 'Windows': # Windows elif platform.system() == "Windows": # Windows
cmd = ['explorer'] cmd = ["explorer"]
if select: if select:
cmd += ['/select,'] cmd += ["/select,"]
else: # linux variants else: # linux variants
cmd = ['xdg-open'] cmd = ["xdg-open"]
if select: if select:
cmd = ['nemo'] cmd = ["nemo"]
cmd += [str(filepath)] cmd += [str(filepath)]
subprocess.Popen(cmd) subprocess.Popen(cmd)
def open_blender_file(filepath=None): def open_blender_file(filepath=None):
filepath = filepath or bpy.data.filepath filepath = filepath or bpy.data.filepath
cmd = sys.argv cmd = sys.argv
# if no filepath, use command as is to reopen blender # if no filepath, use command as is to reopen blender
if filepath != '': if filepath != "":
if len(cmd) > 1 and cmd[1].endswith('.blend'): if len(cmd) > 1 and cmd[1].endswith(".blend"):
cmd[1] = str(filepath) cmd[1] = str(filepath)
else: else:
cmd.insert(1, str(filepath)) cmd.insert(1, str(filepath))
subprocess.Popen(cmd) subprocess.Popen(cmd)
def read_file(path):
'''Read a file with an extension in (json, yaml, yml, txt)'''
exts = ('.json', '.yaml', '.yml', '.txt') def read_file(path):
"""Read a file with an extension in (json, yaml, yml, txt)"""
exts = (".json", ".yaml", ".yml", ".txt")
if not path: if not path:
print('Try to read empty file') print("Try to read empty file")
path = Path(path) path = Path(path)
if not path.exists(): if not path.exists():
print('File not exist', path) print("File not exist", path)
return return
if path.suffix not in exts: if path.suffix not in exts:
print(f'Cannot read file {path}, extension must be in {exts}') print(f"Cannot read file {path}, extension must be in {exts}")
return return
txt = path.read_text() txt = path.read_text()
data = None data = None
if path.suffix.lower() in ('.yaml', '.yml'): if path.suffix.lower() in (".yaml", ".yml"):
yaml = install_module('yaml') yaml = install_module("yaml")
try: try:
data = yaml.safe_load(txt) data = yaml.safe_load(txt)
except Exception: except Exception:
print(f'Could not load yaml file {path}') print(f"Could not load yaml file {path}")
return return
elif path.suffix.lower() == '.json': elif path.suffix.lower() == ".json":
try: try:
data = json.loads(txt) data = json.loads(txt)
except Exception: except Exception:
print(f'Could not load json file {path}') print(f"Could not load json file {path}")
return return
else: else:
data = txt data = txt
return data return data
def write_file(path, data, indent=4):
'''Read a file with an extension in (json, yaml, yml, text)'''
exts = ('.json', '.yaml', '.yml', '.txt') def write_file(path, data, indent=4):
"""Read a file with an extension in (json, yaml, yml, text)"""
exts = (".json", ".yaml", ".yml", ".txt")
if not path: if not path:
print('Try to write empty file') print("Try to write empty file")
path = Path(path) path = Path(path)
path.parent.mkdir(parents=True, exist_ok=True) path.parent.mkdir(parents=True, exist_ok=True)
if path.suffix not in exts: if path.suffix not in exts:
print(f'Cannot read file {path}, extension must be in {exts}') print(f"Cannot read file {path}, extension must be in {exts}")
return return
if path.suffix.lower() in ('.yaml', '.yml'): if path.suffix.lower() in (".yaml", ".yml"):
yaml = install_module('yaml') yaml = install_module("yaml")
try: try:
path.write_text(yaml.dump(data), encoding='utf8') path.write_text(yaml.dump(data), encoding="utf8")
except Exception as e: except Exception as e:
print(e) print(e)
print(f'Could not write yaml file {path}') print(f"Could not write yaml file {path}")
return return
elif path.suffix.lower() == '.json': elif path.suffix.lower() == ".json":
try: try:
path.write_text(json.dumps(data, indent=indent), encoding='utf8') path.write_text(json.dumps(data, indent=indent), encoding="utf8")
except Exception as e: except Exception as e:
print(e) print(e)
print(f'Could not write json file {path}') print(f"Could not write json file {path}")
return return
else: else:
data = path.write_text(data, encoding='utf8') data = path.write_text(data, encoding="utf8")
def synchronize(src, dst, only_new=False, only_recent=False, clear=False): def synchronize(src, dst, only_new=False, only_recent=False, clear=False):
@ -300,19 +338,22 @@ def synchronize(src, dst, only_new=False, only_recent=False, clear=False):
# set_actionlib_dir(custom=custom) # set_actionlib_dir(custom=custom)
script = Path(__file__).parent / 'synchronize.py' script = Path(__file__).parent / "synchronize.py"
cmd = [ cmd = [
sys.executable, sys.executable,
script, script,
'--src', str(src), "--src",
'--dst', str(dst), str(src),
'--only-new', json.dumps(only_new), "--dst",
'--only-recent', json.dumps(only_recent), str(dst),
"--only-new",
json.dumps(only_new),
"--only-recent",
json.dumps(only_recent),
] ]
subprocess.Popen(cmd) subprocess.Popen(cmd)
except Exception as e: except Exception as e:
print(e) print(e)

View File

@ -11,6 +11,7 @@ import os
import re import re
import time import time
# from asset_library.constants import ASSETLIB_FILENAME # from asset_library.constants import ASSETLIB_FILENAME
import inspect import inspect
from asset_library.common.file_utils import read_file from asset_library.common.file_utils import read_file
@ -21,34 +22,37 @@ import bpy
def command(func): def command(func):
'''Decorator to be used from printed functions argument and run time''' """Decorator to be used from printed functions argument and run time"""
func_name = func.__name__.replace('_', ' ').title() func_name = func.__name__.replace("_", " ").title()
def _command(*args, **kargs): def _command(*args, **kargs):
bound = inspect.signature(func).bind(*args, **kargs) bound = inspect.signature(func).bind(*args, **kargs)
bound.apply_defaults() bound.apply_defaults()
args_str = ', '.join([f'{k}={v}' for k, v in bound.arguments.items()]) args_str = ", ".join([f"{k}={v}" for k, v in bound.arguments.items()])
print(f'\n[>-] {func_name} ({args_str}) --- Start ---') print(f"\n[>-] {func_name} ({args_str}) --- Start ---")
t0 = time.time() t0 = time.time()
result = func(*args, **kargs) result = func(*args, **kargs)
print(f'[>-] {func_name} --- Finished (total time : {time.time() - t0:.2f}s) ---') print(
f"[>-] {func_name} --- Finished (total time : {time.time() - t0:.2f}s) ---"
)
return result return result
return _command return _command
def asset_warning_callback(self, context): def asset_warning_callback(self, context):
"""Callback function to display a warning message when ading or modifying an asset""" """Callback function to display a warning message when ading or modifying an asset"""
self.warning = '' self.warning = ""
if not self.name: if not self.name:
self.warning = 'You need to specify a name' self.warning = "You need to specify a name"
return return
if not self.catalog: if not self.catalog:
self.warning = 'You need to specify a catalog' self.warning = "You need to specify a catalog"
return return
lib = get_active_library() lib = get_active_library()
@ -60,10 +64,11 @@ def asset_warning_callback(self, context):
lib = prefs.libraries[lib.store_library] lib = prefs.libraries[lib.store_library]
if not lib.library_type.get_asset_path(self.name, self.catalog).parents[1].exists(): if not lib.library_type.get_asset_path(self.name, self.catalog).parents[1].exists():
self.warning = 'A new folder will be created' self.warning = "A new folder will be created"
def get_active_library(): def get_active_library():
'''Get the pref library properties from the active library of the asset browser''' """Get the pref library properties from the active library of the asset browser"""
prefs = get_addon_prefs() prefs = get_addon_prefs()
asset_lib_ref = bpy.context.space_data.params.asset_library_ref asset_lib_ref = bpy.context.space_data.params.asset_library_ref
@ -72,18 +77,20 @@ def get_active_library():
if l.library_name == asset_lib_ref: if l.library_name == asset_lib_ref:
return l return l
def get_active_catalog(): def get_active_catalog():
'''Get the active catalog path''' """Get the active catalog path"""
lib = get_active_library() lib = get_active_library()
cat_data = lib.library_type.read_catalog() cat_data = lib.library_type.read_catalog()
cat_data = {v['id']:k for k,v in cat_data.items()} cat_data = {v["id"]: k for k, v in cat_data.items()}
cat_id = bpy.context.space_data.params.catalog_id cat_id = bpy.context.space_data.params.catalog_id
if cat_id in cat_data: if cat_id in cat_data:
return cat_data[cat_id] return cat_data[cat_id]
return '' return ""
""" """
def norm_asset_datas(asset_file_datas): def norm_asset_datas(asset_file_datas):
@ -181,7 +188,7 @@ def get_asset_source(replace_local=False):
return source_path return source_path
""" """
''' """
def get_catalog_path(filepath=None): def get_catalog_path(filepath=None):
filepath = filepath or bpy.data.filepath filepath = filepath or bpy.data.filepath
filepath = Path(filepath) filepath = Path(filepath)
@ -196,7 +203,7 @@ def get_catalog_path(filepath=None):
catalog.touch(exist_ok=False) catalog.touch(exist_ok=False)
return catalog return catalog
''' """
# def read_catalog(path, key='path'): # def read_catalog(path, key='path'):
# cat_data = {} # cat_data = {}
@ -302,14 +309,15 @@ def create_catalog_file(json_path : str|Path, keep_existing_category : bool = Tr
return return
""" """
def clear_env_libraries(): def clear_env_libraries():
print('clear_env_libraries') print("clear_env_libraries")
prefs = get_addon_prefs() prefs = get_addon_prefs()
asset_libraries = bpy.context.preferences.filepaths.asset_libraries asset_libraries = bpy.context.preferences.filepaths.asset_libraries
for env_lib in prefs.env_libraries: for env_lib in prefs.env_libraries:
name = env_lib.get('asset_library') name = env_lib.get("asset_library")
if not name: if not name:
continue continue
@ -322,7 +330,8 @@ def clear_env_libraries():
prefs.env_libraries.clear() prefs.env_libraries.clear()
'''
"""
env_libs = get_env_libraries() env_libs = get_env_libraries()
paths = [Path(l['path']).resolve().as_posix() for n, l in env_libs.items()] paths = [Path(l['path']).resolve().as_posix() for n, l in env_libs.items()]
@ -331,10 +340,11 @@ def clear_env_libraries():
if (l.name in env_libs or lib_path in paths): if (l.name in env_libs or lib_path in paths):
libs.remove(i) libs.remove(i)
''' """
def set_env_libraries(path=None) -> list: def set_env_libraries(path=None) -> list:
'''Read the environments variables and create the libraries''' """Read the environments variables and create the libraries"""
# from asset_library.prefs import AssetLibraryOptions # from asset_library.prefs import AssetLibraryOptions
prefs = get_addon_prefs() prefs = get_addon_prefs()
@ -359,7 +369,8 @@ def set_env_libraries(path=None) -> list:
return libs return libs
'''
"""
def get_env_libraries(): def get_env_libraries():
env_libraries = {} env_libraries = {}
@ -391,21 +402,17 @@ def get_env_libraries():
} }
return env_libraries return env_libraries
''' """
def resync_lib(name, waiting_time): def resync_lib(name, waiting_time):
bpy.app.timers.register( bpy.app.timers.register(
lambda: bpy.ops.assetlib.synchronize(only_recent=True, name=name), lambda: bpy.ops.assetlib.synchronize(only_recent=True, name=name),
first_interval=waiting_time first_interval=waiting_time,
) )
"""
'''
def set_assetlib_paths(): def set_assetlib_paths():
prefs = bpy.context.preferences prefs = bpy.context.preferences
@ -452,16 +459,4 @@ def set_actionlib_paths():
prefs.filepaths.asset_libraries[lib_id].name = actionlib_name prefs.filepaths.asset_libraries[lib_id].name = actionlib_name
#prefs.filepaths.asset_libraries[lib_id].path = str(actionlib_dir) #prefs.filepaths.asset_libraries[lib_id].path = str(actionlib_dir)
''' """

View File

@ -32,10 +32,7 @@ class AssetCache:
@property @property
def metadata(self): def metadata(self):
metadata = { metadata = {".library_id": self.library_id, ".filepath": self.filepath}
'.library_id': self.library_id,
'.filepath': self.filepath
}
metadata.update(self._metadata) metadata.update(self._metadata)
@ -43,23 +40,23 @@ class AssetCache:
@property @property
def norm_name(self): def norm_name(self):
return self.name.replace(' ', '_').lower() return self.name.replace(" ", "_").lower()
def unique_name(self): def unique_name(self):
return (self.filepath / self.name).as_posix() return (self.filepath / self.name).as_posix()
def set_data(self, data): def set_data(self, data):
catalog = data['catalog'] catalog = data["catalog"]
if isinstance(catalog, (list, tuple)): if isinstance(catalog, (list, tuple)):
catalog = '/'.join(catalog) catalog = "/".join(catalog)
self.catalog = catalog self.catalog = catalog
self.author = data.get('author', '') self.author = data.get("author", "")
self.description = data.get('description', '') self.description = data.get("description", "")
self.tags = data.get('tags', []) self.tags = data.get("tags", [])
self.type = data.get('type') self.type = data.get("type")
self.name = data['name'] self.name = data["name"]
self._metadata = data.get('metadata', {}) self._metadata = data.get("metadata", {})
def to_dict(self): def to_dict(self):
return dict( return dict(
@ -69,11 +66,11 @@ class AssetCache:
description=self.description, description=self.description,
tags=self.tags, tags=self.tags,
type=self.type, type=self.type,
name=self.name name=self.name,
) )
def __repr__(self): def __repr__(self):
return f'AssetCache(name={self.name}, catalog={self.catalog})' return f"AssetCache(name={self.name}, catalog={self.catalog})"
def __eq__(self, other): def __eq__(self, other):
return self.to_dict() == other.to_dict() return self.to_dict() == other.to_dict()
@ -111,7 +108,7 @@ class AssetsCache:
return next((a for a in self if a.name == name), None) return next((a for a in self if a.name == name), None)
def __repr__(self): def __repr__(self):
return f'AssetsCache({list(self)})' return f"AssetsCache({list(self)})"
class FileCache: class FileCache:
@ -132,15 +129,15 @@ class FileCache:
def set_data(self, data): def set_data(self, data):
if 'filepath' in data: if "filepath" in data:
self.filepath = Path(data['filepath']) self.filepath = Path(data["filepath"])
self.modified = data.get('modified', time.time_ns()) self.modified = data.get("modified", time.time_ns())
if data.get('type') == 'FILE': if data.get("type") == "FILE":
self.assets.add(data) self.assets.add(data)
for asset_cache_data in data.get('assets', []): for asset_cache_data in data.get("assets", []):
self.assets.add(asset_cache_data) self.assets.add(asset_cache_data)
def to_dict(self): def to_dict(self):
@ -148,7 +145,7 @@ class FileCache:
filepath=self.filepath.as_posix(), filepath=self.filepath.as_posix(),
modified=self.modified, modified=self.modified,
library_id=self.library_id, library_id=self.library_id,
assets=[asset_cache.to_dict() for asset_cache in self] assets=[asset_cache.to_dict() for asset_cache in self],
) )
def __iter__(self): def __iter__(self):
@ -158,7 +155,7 @@ class FileCache:
return self._data[key] return self._data[key]
def __repr__(self): def __repr__(self):
return f'FileCache(filepath={self.filepath})' return f"FileCache(filepath={self.filepath})"
class AssetCacheDiff: class AssetCacheDiff:
@ -192,29 +189,46 @@ class LibraryCacheDiff:
def compare(self, old_cache, new_cache): def compare(self, old_cache, new_cache):
if old_cache is None or new_cache is None: if old_cache is None or new_cache is None:
print('Cannot Compare cache with None') print("Cannot Compare cache with None")
cache_dict = {a.unique_name: a for a in old_cache.asset_caches} cache_dict = {a.unique_name: a for a in old_cache.asset_caches}
new_cache_dict = {a.unique_name: a for a in new_cache.asset_caches} new_cache_dict = {a.unique_name: a for a in new_cache.asset_caches}
assets_added = self.add([v for k, v in new_cache_dict.items() if k not in cache_dict], 'ADD') assets_added = self.add(
assets_removed = self.add([v for k, v in cache_dict.items() if k not in new_cache_dict], 'REMOVED') [v for k, v in new_cache_dict.items() if k not in cache_dict], "ADD"
assets_modified = self.add([v for k, v in cache_dict.items() if v not in assets_removed and v!= new_cache_dict[k]], 'MODIFIED') )
assets_removed = self.add(
[v for k, v in cache_dict.items() if k not in new_cache_dict], "REMOVED"
)
assets_modified = self.add(
[
v
for k, v in cache_dict.items()
if v not in assets_removed and v != new_cache_dict[k]
],
"MODIFIED",
)
if assets_added: if assets_added:
print(f'{len(assets_added)} Assets Added \n{tuple(a.name for a in assets_added[:10])}...\n') print(
f"{len(assets_added)} Assets Added \n{tuple(a.name for a in assets_added[:10])}...\n"
)
if assets_removed: if assets_removed:
print(f'{len(assets_removed)} Assets Removed \n{tuple(a.name for a in assets_removed[:10])}...\n') print(
f"{len(assets_removed)} Assets Removed \n{tuple(a.name for a in assets_removed[:10])}...\n"
)
if assets_modified: if assets_modified:
print(f'{len(assets_modified)} Assets Modified \n{tuple(a.name for a in assets_modified[:10])}...\n') print(
f"{len(assets_modified)} Assets Modified \n{tuple(a.name for a in assets_modified[:10])}...\n"
)
if len(self) == 0: if len(self) == 0:
print('No change in the library') print("No change in the library")
return self return self
def group_by(self, key): def group_by(self, key):
'''Return groups of file cache diff using the key provided''' """Return groups of file cache diff using the key provided"""
data = list(self).sort(key=key) data = list(self).sort(key=key)
return groupby(data, key=key) return groupby(data, key=key)
@ -228,7 +242,7 @@ class LibraryCacheDiff:
return len(self._data) return len(self._data)
def __repr__(self): def __repr__(self):
return f'LibraryCacheDiff(operations={[o for o in self][:2]}...)' return f"LibraryCacheDiff(operations={[o for o in self][:2]}...)"
class LibraryCache: class LibraryCache:
@ -248,7 +262,7 @@ class LibraryCache:
@property @property
def library_id(self): def library_id(self):
return self.filepath.stem.split('.')[-1] return self.filepath.stem.split(".")[-1]
# @property # @property
# def filepath(self): # def filepath(self):
@ -260,7 +274,7 @@ class LibraryCache:
@property @property
def asset_caches(self): def asset_caches(self):
'''Return an iterator to get all asset caches''' """Return an iterator to get all asset caches"""
return (asset_cache for file_cache in self for asset_cache in file_cache) return (asset_cache for file_cache in self for asset_cache in file_cache)
@property @property
@ -268,7 +282,7 @@ class LibraryCache:
return Path(bpy.app.tempdir) / self.filename return Path(bpy.app.tempdir) / self.filename
def read(self): def read(self):
print(f'Read cache from {self.filepath}') print(f"Read cache from {self.filepath}")
for file_cache_data in read_file(self.filepath): for file_cache_data in read_file(self.filepath):
self.add(file_cache_data) self.add(file_cache_data)
@ -280,7 +294,7 @@ class LibraryCache:
if tmp: if tmp:
filepath = self.tmp_filepath filepath = self.tmp_filepath
print(f'Write cache file to {filepath}') print(f"Write cache file to {filepath}")
write_file(filepath, self._data) write_file(filepath, self._data)
return filepath return filepath
@ -293,7 +307,7 @@ class LibraryCache:
def add_asset_cache(self, asset_cache_data, filepath=None): def add_asset_cache(self, asset_cache_data, filepath=None):
if filepath is None: if filepath is None:
filepath = asset_cache_data['filepath'] filepath = asset_cache_data["filepath"]
file_cache = self.get(filepath) file_cache = self.get(filepath)
if not file_cache: if not file_cache:
@ -341,19 +355,19 @@ class LibraryCache:
for asset_cache_diff in cache_diff: for asset_cache_diff in cache_diff:
file_cache = self.get(asset_cache_diff.filepath) file_cache = self.get(asset_cache_diff.filepath)
if not asset_cache: if not asset_cache:
print(f'Filepath {asset_cache_diff.filepath} not in {self}' ) print(f"Filepath {asset_cache_diff.filepath} not in {self}")
continue continue
asset_cache = file_cache.get(asset_cache_diff.name) asset_cache = file_cache.get(asset_cache_diff.name)
if not asset_cache: if not asset_cache:
print(f'Asset {asset_cache_diff.name} not in file_cache {file_cache}' ) print(f"Asset {asset_cache_diff.name} not in file_cache {file_cache}")
continue continue
if asset_cache_diff.operation == 'REMOVE': if asset_cache_diff.operation == "REMOVE":
file_cache.assets.remove(asset_cache_diff.name) file_cache.assets.remove(asset_cache_diff.name)
elif asset_cache_diff.operation in ('MODIFY', 'ADD'): elif asset_cache_diff.operation in ("MODIFY", "ADD"):
asset_cache.set_data(asset_cache_diff.asset_cache.to_dict()) asset_cache.set_data(asset_cache_diff.asset_cache.to_dict())
return self return self
@ -377,5 +391,4 @@ class LibraryCache:
return next((a for a in self if a.filepath == filepath), None) return next((a for a in self if a.filepath == filepath), None)
def __repr__(self): def __repr__(self):
return f'LibraryCache(library_id={self.library_id})' return f"LibraryCache(library_id={self.library_id})"

View File

@ -1,4 +1,3 @@
import argparse import argparse
import fnmatch import fnmatch
import importlib.util import importlib.util
@ -19,24 +18,29 @@ spec.loader.exec_module(utils)
def synchronize(src, dst, only_new=False, only_recent=False): def synchronize(src, dst, only_new=False, only_recent=False):
excludes=['*.sync-conflict-*', '.*'] excludes = ["*.sync-conflict-*", ".*"]
includes=['*.blend', 'blender_assets.cats.txt'] includes = ["*.blend", "blender_assets.cats.txt"]
utils.copy_dir( utils.copy_dir(
src, dst, src,
only_new=only_new, only_recent=only_recent, dst,
excludes=excludes, includes=includes only_new=only_new,
only_recent=only_recent,
excludes=excludes,
includes=includes,
) )
if __name__ == '__main__' : if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Add Comment To the tracker', parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter) description="Add Comment To the tracker",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument('--src') parser.add_argument("--src")
parser.add_argument('--dst') parser.add_argument("--dst")
parser.add_argument('--only-new', type=json.loads, default='false') parser.add_argument("--only-new", type=json.loads, default="false")
parser.add_argument('--only-recent', type=json.loads, default='false') parser.add_argument("--only-recent", type=json.loads, default="false")
args = parser.parse_args() args = parser.parse_args()
synchronize(**vars(args)) synchronize(**vars(args))

View File

@ -9,25 +9,26 @@ import string
class TemplateFormatter(string.Formatter): class TemplateFormatter(string.Formatter):
def format_field(self, value, format_spec): def format_field(self, value, format_spec):
if isinstance(value, str): if isinstance(value, str):
spec, sep = [*format_spec.split(':'), None][:2] spec, sep = [*format_spec.split(":"), None][:2]
if sep: if sep:
value = value.replace('_', ' ') value = value.replace("_", " ")
value = value = re.sub(r'([a-z])([A-Z])', rf'\1{sep}\2', value) value = value = re.sub(r"([a-z])([A-Z])", rf"\1{sep}\2", value)
value = value.replace(' ', sep) value = value.replace(" ", sep)
if spec == 'u': if spec == "u":
value = value.upper() value = value.upper()
elif spec == 'l': elif spec == "l":
value = value.lower() value = value.lower()
elif spec == 't': elif spec == "t":
value = value.title() value = value.title()
return super().format(value, format_spec) return super().format(value, format_spec)
class Template: class Template:
field_pattern = re.compile(r'{(\w+)\*{0,2}}') field_pattern = re.compile(r"{(\w+)\*{0,2}}")
field_pattern_recursive = re.compile(r'{(\w+)\*{2}}') field_pattern_recursive = re.compile(r"{(\w+)\*{2}}")
def __init__(self, template): def __init__(self, template):
# asset_data_path = Path(lib_path) / ASSETLIB_FILENAME # asset_data_path = Path(lib_path) / ASSETLIB_FILENAME
@ -37,16 +38,16 @@ class Template:
@property @property
def glob_pattern(self): def glob_pattern(self):
pattern = self.field_pattern_recursive.sub('**', self.raw) pattern = self.field_pattern_recursive.sub("**", self.raw)
pattern = self.field_pattern.sub('*', pattern) pattern = self.field_pattern.sub("*", pattern)
return pattern return pattern
@property @property
def re_pattern(self): def re_pattern(self):
pattern = self.field_pattern_recursive.sub('([\\\w -_.\/]+)', self.raw) pattern = self.field_pattern_recursive.sub("([\\\w -_.\/]+)", self.raw)
pattern = self.field_pattern.sub('([\\\w -_.]+)', pattern) pattern = self.field_pattern.sub("([\\\w -_.]+)", pattern)
pattern = pattern.replace('?', '.') pattern = pattern.replace("?", ".")
pattern = pattern.replace('*', '.*') pattern = pattern.replace("*", ".*")
return re.compile(pattern) return re.compile(pattern)
@ -61,7 +62,7 @@ class Template:
res = self.re_pattern.findall(path) res = self.re_pattern.findall(path)
if not res: if not res:
print('Could not parse {path} with {self.re_pattern}') print("Could not parse {path} with {self.re_pattern}")
return {} return {}
fields = self.fields fields = self.fields
@ -92,14 +93,14 @@ class Template:
# print('FORMAT', self.raw, data) # print('FORMAT', self.raw, data)
path = self.formatter.format(self.raw, **self.norm_data(data)) path = self.formatter.format(self.raw, **self.norm_data(data))
except KeyError as e: except KeyError as e:
print(f'Cannot format {self.raw} with {data}, field {e} is missing') print(f"Cannot format {self.raw} with {data}, field {e} is missing")
return return
path = os.path.expandvars(path) path = os.path.expandvars(path)
return Path(path) return Path(path)
def glob(self, directory, pattern=None): def glob(self, directory, pattern=None):
'''If pattern is given it need to be absolute''' """If pattern is given it need to be absolute"""
if pattern is None: if pattern is None:
pattern = Path(directory, self.glob_pattern).as_posix() pattern = Path(directory, self.glob_pattern).as_posix()
@ -114,7 +115,7 @@ class Template:
pattern = self.format(data, **kargs) pattern = self.format(data, **kargs)
pattern_str = str(pattern) pattern_str = str(pattern)
if '*' not in pattern_str and '?' not in pattern_str: if "*" not in pattern_str and "?" not in pattern_str:
return pattern return pattern
paths = glob(pattern.as_posix()) paths = glob(pattern.as_posix())
@ -124,4 +125,4 @@ class Template:
# return pattern # return pattern
def __repr__(self): def __repr__(self):
return f'Template({self.raw})' return f"Template({self.raw})"

View File

@ -5,22 +5,23 @@ import bpy
DATA_TYPE_ITEMS = [ DATA_TYPE_ITEMS = [
("ACTION", "Action", "", "ACTION", 0), ("ACTION", "Action", "", "ACTION", 0),
("COLLECTION", "Collection", "", "OUTLINER_OB_GROUP_INSTANCE", 1), ("COLLECTION", "Collection", "", "OUTLINER_OB_GROUP_INSTANCE", 1),
("FILE", "File", "", "FILE", 2) ("FILE", "File", "", "FILE", 2),
] ]
DATA_TYPES = [i[0] for i in DATA_TYPE_ITEMS] DATA_TYPES = [i[0] for i in DATA_TYPE_ITEMS]
ICONS = {identifier: icon for identifier, name, description, icon, number in DATA_TYPE_ITEMS} ICONS = {
identifier: icon for identifier, name, description, icon, number in DATA_TYPE_ITEMS
}
ASSETLIB_FILENAME = "blender_assets.libs.json" ASSETLIB_FILENAME = "blender_assets.libs.json"
MODULE_DIR = Path(__file__).parent MODULE_DIR = Path(__file__).parent
RESOURCES_DIR = MODULE_DIR / 'resources' RESOURCES_DIR = MODULE_DIR / "resources"
LIBRARY_TYPE_DIR = MODULE_DIR / 'library_types' LIBRARY_TYPE_DIR = MODULE_DIR / "library_types"
LIBRARY_TYPES = [] LIBRARY_TYPES = []
ADAPTER_DIR = MODULE_DIR / 'adapters' ADAPTER_DIR = MODULE_DIR / "adapters"
ADAPTERS = [] ADAPTERS = []
PREVIEW_ASSETS_SCRIPT = MODULE_DIR / 'common' / 'preview_assets.py' PREVIEW_ASSETS_SCRIPT = MODULE_DIR / "common" / "preview_assets.py"
# ADD_ASSET_DICT = {} # ADD_ASSET_DICT = {}

View File

@ -1,8 +1,6 @@
from asset_library.file import operators, gui, keymaps
from asset_library.file import ( if "bpy" in locals():
operators, gui, keymaps)
if 'bpy' in locals():
import importlib import importlib
importlib.reload(operators) importlib.reload(operators)
@ -11,10 +9,12 @@ if 'bpy' in locals():
import bpy import bpy
def register(): def register():
operators.register() operators.register()
keymaps.register() keymaps.register()
def unregister(): def unregister():
operators.unregister() operators.unregister()
keymaps.unregister() keymaps.unregister()

View File

@ -1,4 +1,3 @@
import argparse import argparse
import sys import sys
import json import json
@ -13,18 +12,22 @@ from asset_library.common.bl_utils import thumbnail_blend_file
from asset_library.common.functions import command from asset_library.common.functions import command
@command @command
def bundle_library(source_directory, bundle_directory, template_info, thumbnail_template, def bundle_library(
template=None, data_file=None): source_directory,
bundle_directory,
template_info,
thumbnail_template,
template=None,
data_file=None,
):
field_pattern = r'{(\w+)}' field_pattern = r"{(\w+)}"
asset_data_path = Path(bundle_directory) / ASSETLIB_FILENAME asset_data_path = Path(bundle_directory) / ASSETLIB_FILENAME
glob_pattern = re.sub(field_pattern, '*', template) glob_pattern = re.sub(field_pattern, "*", template)
re_pattern = re.sub(field_pattern, r'([\\w -_.]+)', template) re_pattern = re.sub(field_pattern, r"([\\w -_.]+)", template)
re_pattern = re_pattern.replace('?', '.') re_pattern = re_pattern.replace("?", ".")
field_names = re.findall(field_pattern, template) field_names = re.findall(field_pattern, template)
@ -35,29 +38,31 @@ def bundle_library(source_directory, bundle_directory, template_info, thumbnail_
field_values = re.findall(re_pattern, rel_path)[0] field_values = re.findall(re_pattern, rel_path)[0]
field_data = {k: v for k, v in zip(field_names, field_values)} field_data = {k: v for k, v in zip(field_names, field_values)}
name = field_data.get('name', f.stem) name = field_data.get("name", f.stem)
thumbnail = (f / thumbnail_template.format(name=name)).resolve() thumbnail = (f / thumbnail_template.format(name=name)).resolve()
asset_data = (f / template_info.format(name=name)).resolve() asset_data = (f / template_info.format(name=name)).resolve()
catalogs = sorted([v for k,v in sorted(field_data.items()) if re.findall('cat[0-9]+', k)]) catalogs = sorted(
catalogs = [c.replace('_', ' ').title() for c in catalogs] [v for k, v in sorted(field_data.items()) if re.findall("cat[0-9]+", k)]
)
catalogs = [c.replace("_", " ").title() for c in catalogs]
if not thumbnail.exists(): if not thumbnail.exists():
thumbnail_blend_file(f, thumbnail) thumbnail_blend_file(f, thumbnail)
asset_data = { asset_data = {
'catalog' : '/'.join(catalogs), "catalog": "/".join(catalogs),
'preview' : thumbnail.as_posix(), #'./' + bpy.path.relpath(str(thumbnail), start=str(f))[2:], "preview": thumbnail.as_posix(), #'./' + bpy.path.relpath(str(thumbnail), start=str(f))[2:],
'filepath' : f.as_posix(), #'./' + bpy.path.relpath(str(f), start=str(asset_data_path))[2:], "filepath": f.as_posix(), #'./' + bpy.path.relpath(str(f), start=str(asset_data_path))[2:],
'name': name, "name": name,
'tags': [], "tags": [],
'metadata': {'filepath': f.as_posix()} "metadata": {"filepath": f.as_posix()},
} }
asset_file_datas.append(asset_data) asset_file_datas.append(asset_data)
# Write json data file to store all asset found # Write json data file to store all asset found
print(f'Writing asset data file to, {asset_data_path}') print(f"Writing asset data file to, {asset_data_path}")
asset_data_path.write_text(json.dumps(asset_file_datas, indent=4)) asset_data_path.write_text(json.dumps(asset_file_datas, indent=4))
# script = MODULE_DIR / 'common' / 'bundle_blend.py' # script = MODULE_DIR / 'common' / 'bundle_blend.py'
@ -65,6 +70,7 @@ def bundle_library(source_directory, bundle_directory, template_info, thumbnail_
# print(cmd) # print(cmd)
# subprocess.call(cmd) # subprocess.call(cmd)
@command @command
def bundle_blend(filepath, depth=0): def bundle_blend(filepath, depth=0):
# print('Bundle Blend...') # print('Bundle Blend...')
@ -73,11 +79,11 @@ def bundle_blend(filepath, depth=0):
# asset_data_path = get_asset_datas_file(filepath) # asset_data_path = get_asset_datas_file(filepath)
asset_data_path = filepath / ASSETLIB_FILENAME asset_data_path = filepath / ASSETLIB_FILENAME
blend_name = filepath.name.replace(' ', '_').lower() blend_name = filepath.name.replace(" ", "_").lower()
blend_path = (filepath / blend_name).with_suffix('.blend') blend_path = (filepath / blend_name).with_suffix(".blend")
if not asset_data_path.exists(): if not asset_data_path.exists():
raise Exception(f'The file {asset_data_path} not exist') raise Exception(f"The file {asset_data_path} not exist")
catalog_path = get_catalog_path(filepath) catalog_path = get_catalog_path(filepath)
catalog_data = read_catalog(catalog_path) catalog_data = read_catalog(catalog_path)
@ -88,8 +94,8 @@ def bundle_blend(filepath, depth=0):
if depth == 0: if depth == 0:
groups = [asset_file_data] groups = [asset_file_data]
else: else:
asset_file_data.sort(key=lambda x :x['catalog'].split('/')[:depth]) asset_file_data.sort(key=lambda x: x["catalog"].split("/")[:depth])
groups = groupby(asset_file_data, key=lambda x :x['catalog'].split('/')[:depth]) groups = groupby(asset_file_data, key=lambda x: x["catalog"].split("/")[:depth])
# progress = 0 # progress = 0
total_assets = len(asset_file_data) total_assets = len(asset_file_data)
@ -99,39 +105,37 @@ def bundle_blend(filepath, depth=0):
bpy.ops.wm.read_homefile(use_empty=True) bpy.ops.wm.read_homefile(use_empty=True)
for asset_data in asset_datas: for asset_data in asset_datas:
blend_name = sub_path[-1].replace(' ', '_').lower() blend_name = sub_path[-1].replace(" ", "_").lower()
blend_path = Path(filepath, *sub_path, blend_name).with_suffix('.blend') blend_path = Path(filepath, *sub_path, blend_name).with_suffix(".blend")
if i % int(total_assets / 100) == 0: if i % int(total_assets / 100) == 0:
print(f'Progress: {int(i / total_assets * 100)}') print(f"Progress: {int(i / total_assets * 100)}")
col = bpy.data.collections.new(name=asset_data['name']) col = bpy.data.collections.new(name=asset_data["name"])
# Seems slow # Seems slow
# bpy.context.scene.collection.children.link(col) # bpy.context.scene.collection.children.link(col)
col.asset_mark() col.asset_mark()
with bpy.context.temp_override(id=col): with bpy.context.temp_override(id=col):
bpy.ops.ed.lib_id_load_custom_preview( bpy.ops.ed.lib_id_load_custom_preview(filepath=asset_data["preview"])
filepath=asset_data['preview']
)
col.asset_data.description = asset_data.get('description', '') col.asset_data.description = asset_data.get("description", "")
catalog_name = asset_data['catalog'] catalog_name = asset_data["catalog"]
catalog = catalog_data.get(catalog_name) catalog = catalog_data.get(catalog_name)
if not catalog: if not catalog:
catalog = {'id': str(uuid.uuid4()), 'name': catalog_name} catalog = {"id": str(uuid.uuid4()), "name": catalog_name}
catalog_data[catalog_name] = catalog catalog_data[catalog_name] = catalog
col.asset_data.catalog_id = catalog['id'] col.asset_data.catalog_id = catalog["id"]
for k, v in asset_data.get('metadata', {}).items(): for k, v in asset_data.get("metadata", {}).items():
col.asset_data[k] = v col.asset_data[k] = v
i += 1 i += 1
print(f'Saving Blend to {blend_path}') print(f"Saving Blend to {blend_path}")
blend_path.mkdir(exist_ok=True, parents=True) blend_path.mkdir(exist_ok=True, parents=True)
bpy.ops.wm.save_as_mainfile(filepath=str(blend_path), compress=True) bpy.ops.wm.save_as_mainfile(filepath=str(blend_path), compress=True)
@ -141,20 +145,22 @@ def bundle_blend(filepath, depth=0):
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
if __name__ == '__main__' : if __name__ == "__main__":
parser = argparse.ArgumentParser(description='bundle_blend', parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter) description="bundle_blend",
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument('--source-path') parser.add_argument("--source-path")
parser.add_argument('--bundle-path') parser.add_argument("--bundle-path")
parser.add_argument('--asset-data-template') parser.add_argument("--asset-data-template")
parser.add_argument('--thumbnail-template') parser.add_argument("--thumbnail-template")
parser.add_argument('--template', default=None) parser.add_argument("--template", default=None)
parser.add_argument('--data-file', default=None) parser.add_argument("--data-file", default=None)
parser.add_argument('--depth', default=0, type=int) parser.add_argument("--depth", default=0, type=int)
if '--' in sys.argv : if "--" in sys.argv:
index = sys.argv.index('--') index = sys.argv.index("--")
sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]] sys.argv = [sys.argv[index - 1], *sys.argv[index + 1 :]]
args = parser.parse_args() args = parser.parse_args()
@ -165,6 +171,7 @@ if __name__ == '__main__' :
template_info=args.template_info, template_info=args.template_info,
thumbnail_template=args.thumbnail_template, thumbnail_template=args.thumbnail_template,
template=args.template, template=args.template,
data_file=args.data_file) data_file=args.data_file,
)
bundle_blend(filepath=args.bundle_directory, depth=args.depth) bundle_blend(filepath=args.bundle_directory, depth=args.depth)

View File

@ -1,4 +1,3 @@
import bpy import bpy
from pathlib import Path from pathlib import Path
@ -23,7 +22,9 @@ def draw_context_menu(layout):
lib = get_active_library() lib = get_active_library()
filepath = lib.library_type.get_active_asset_path() filepath = lib.library_type.get_active_asset_path()
layout.operator("assetlib.open_blend_file", text="Open Blend File")#.filepath = asset.asset_data['filepath'] layout.operator(
"assetlib.open_blend_file", text="Open Blend File"
) # .filepath = asset.asset_data['filepath']
op = layout.operator("wm.link", text="Link") op = layout.operator("wm.link", text="Link")
op.filepath = str(filepath) op.filepath = str(filepath)
@ -32,7 +33,7 @@ def draw_context_menu(layout):
def draw_header(layout): def draw_header(layout):
'''Draw the header of the Asset Browser Window''' """Draw the header of the Asset Browser Window"""
layout.separator() layout.separator()
# layout.operator("actionlib.store_anim_pose", text='Add Action', icon='FILE_NEW') # layout.operator("actionlib.store_anim_pose", text='Add Action', icon='FILE_NEW')

View File

@ -1,5 +1,3 @@
from typing import List, Tuple from typing import List, Tuple
import bpy import bpy
@ -7,13 +5,16 @@ from bpy.app.handlers import persistent
addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = [] addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = []
def register() -> None: def register() -> None:
wm = bpy.context.window_manager wm = bpy.context.window_manager
if not wm.keyconfigs.addon: if not wm.keyconfigs.addon:
# This happens when Blender is running in the background. # This happens when Blender is running in the background.
return return
km = wm.keyconfigs.addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER") km = wm.keyconfigs.addon.keymaps.new(
name="File Browser Main", space_type="FILE_BROWSER"
)
kmi = km.keymap_items.new("assetlib.open_blend_file", "LEFTMOUSE", "DOUBLE_CLICK") kmi = km.keymap_items.new("assetlib.open_blend_file", "LEFTMOUSE", "DOUBLE_CLICK")
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))

View File

@ -1,12 +1,14 @@
import bpy import bpy
from bpy.types import Context, Operator from bpy.types import Context, Operator
from bpy_extras import asset_utils from bpy_extras import asset_utils
from bpy.props import StringProperty from bpy.props import StringProperty
from typing import List, Tuple, Set from typing import List, Tuple, Set
from asset_library.common.file_utils import (open_blender_file, from asset_library.common.file_utils import (
synchronize, open_blender_file) open_blender_file,
synchronize,
open_blender_file,
)
from asset_library.common.functions import get_active_library from asset_library.common.functions import get_active_library
@ -14,8 +16,8 @@ from asset_library.common.functions import get_active_library
class ASSETLIB_OT_open_blend_file(Operator): class ASSETLIB_OT_open_blend_file(Operator):
bl_idname = "assetlib.open_blend_file" bl_idname = "assetlib.open_blend_file"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Open Blender File' bl_label = "Open Blender File"
bl_description = 'Open blender file' bl_description = "Open blender file"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
@ -24,10 +26,10 @@ class ASSETLIB_OT_open_blend_file(Operator):
return False return False
lib = get_active_library() lib = get_active_library()
if not lib or lib.data_type != 'FILE': if not lib or lib.data_type != "FILE":
return False return False
if not context.active_file or 'filepath' not in context.active_file.asset_data: if not context.active_file or "filepath" not in context.active_file.asset_data:
cls.poll_message_set("Has not filepath property") cls.poll_message_set("Has not filepath property")
return False return False
@ -41,17 +43,17 @@ class ASSETLIB_OT_open_blend_file(Operator):
open_blender_file(filepath) open_blender_file(filepath)
return {'FINISHED'} return {"FINISHED"}
classes = ( classes = (ASSETLIB_OT_open_blend_file,)
ASSETLIB_OT_open_blend_file,
)
def register(): def register():
for cls in classes: for cls in classes:
bpy.utils.register_class(cls) bpy.utils.register_class(cls)
def unregister(): def unregister():
for cls in reversed(classes): for cls in reversed(classes):
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(cls)

90
gui.py
View File

@ -25,40 +25,36 @@ from asset_library.common.bl_utils import (
get_object_libraries, get_object_libraries,
) )
from asset_library.common.functions import ( from asset_library.common.functions import get_active_library
get_active_library
)
def pose_library_panel_poll(): def pose_library_panel_poll():
return bpy.context.object and bpy.context.object.mode == 'POSE' return bpy.context.object and bpy.context.object.mode == "POSE"
class PoseLibraryPanel: class PoseLibraryPanel:
@classmethod @classmethod
def pose_library_panel_poll(cls, context: Context) -> bool: def pose_library_panel_poll(cls, context: Context) -> bool:
return bool( return bool(context.object and context.object.mode == "POSE")
context.object
and context.object.mode == 'POSE'
)
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
return cls.pose_library_panel_poll(context); return cls.pose_library_panel_poll(context)
class AssetLibraryMenu: class AssetLibraryMenu:
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
from bpy_extras.asset_utils import SpaceAssetInfo from bpy_extras.asset_utils import SpaceAssetInfo
return SpaceAssetInfo.is_asset_browser_poll(context) return SpaceAssetInfo.is_asset_browser_poll(context)
class ASSETLIB_PT_libraries(Panel): class ASSETLIB_PT_libraries(Panel):
bl_label = "Libraries" bl_label = "Libraries"
bl_space_type = 'VIEW_3D' bl_space_type = "VIEW_3D"
bl_region_type = 'UI' bl_region_type = "UI"
bl_category = 'Item' bl_category = "Item"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
@ -70,9 +66,10 @@ class ASSETLIB_PT_libraries(Panel):
for f in get_object_libraries(context.object): for f in get_object_libraries(context.object):
row = layout.row(align=True) row = layout.row(align=True)
row.label(text=f) row.label(text=f)
row.operator("assetlib.open_blend", icon='FILE_BLEND', text='').filepath = f row.operator("assetlib.open_blend", icon="FILE_BLEND", text="").filepath = f
'''
"""
class ASSETLIB_PT_pose_library_usage(Panel): class ASSETLIB_PT_pose_library_usage(Panel):
bl_space_type = 'FILE_BROWSER' bl_space_type = 'FILE_BROWSER'
bl_region_type = "TOOLS" bl_region_type = "TOOLS"
@ -117,11 +114,13 @@ class ASSETLIB_PT_pose_library_usage(Panel):
col = layout.column(align=True) col = layout.column(align=True)
row = col.row(align=True) row = col.row(align=True)
row.operator("actionlib.store_anim_pose", text='Store Anim/Pose', icon='ACTION') row.operator("actionlib.store_anim_pose", text='Store Anim/Pose', icon='ACTION')
''' """
class ASSETLIB_PT_pose_library_editing(PoseLibraryPanel, asset_utils.AssetBrowserPanel, Panel): class ASSETLIB_PT_pose_library_editing(
bl_space_type = 'FILE_BROWSER' PoseLibraryPanel, asset_utils.AssetBrowserPanel, Panel
):
bl_space_type = "FILE_BROWSER"
bl_region_type = "TOOL_PROPS" bl_region_type = "TOOL_PROPS"
bl_label = "Metadata" bl_label = "Metadata"
# bl_options = {'HIDE_HEADER'} # bl_options = {'HIDE_HEADER'}
@ -131,7 +130,7 @@ class ASSETLIB_PT_pose_library_editing(PoseLibraryPanel, asset_utils.AssetBrowse
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
sp = context.space_data sp = context.space_data
if not (sp and sp.type == 'FILE_BROWSER' and sp.browse_mode == 'ASSETS'): if not (sp and sp.type == "FILE_BROWSER" and sp.browse_mode == "ASSETS"):
return False return False
if not (context.active_file and context.active_file.asset_data): if not (context.active_file and context.active_file.asset_data):
@ -144,16 +143,16 @@ class ASSETLIB_PT_pose_library_editing(PoseLibraryPanel, asset_utils.AssetBrowse
layout.use_property_split = True layout.use_property_split = True
asset_data = context.active_file.asset_data asset_data = context.active_file.asset_data
metadata = ['camera', 'is_single_frame', 'rest_pose'] metadata = ["camera", "is_single_frame", "rest_pose"]
if 'camera' in asset_data.keys(): if "camera" in asset_data.keys():
layout.prop(asset_data, f'["camera"]', text='Camera', icon='CAMERA_DATA') layout.prop(asset_data, f'["camera"]', text="Camera", icon="CAMERA_DATA")
if 'is_single_frame' in asset_data.keys(): if "is_single_frame" in asset_data.keys():
layout.prop(asset_data, f'["is_single_frame"]', text='Is Single Frame') layout.prop(asset_data, f'["is_single_frame"]', text="Is Single Frame")
if 'rest_pose' in asset_data.keys(): if "rest_pose" in asset_data.keys():
layout.prop(asset_data, f'["rest_pose"]', text='Rest Pose', icon='ACTION') layout.prop(asset_data, f'["rest_pose"]', text="Rest Pose", icon="ACTION")
if 'filepath' in asset_data.keys(): if "filepath" in asset_data.keys():
layout.prop(asset_data, f'["filepath"]', text='Filepath') layout.prop(asset_data, f'["filepath"]', text="Filepath")
class ASSETLIB_MT_context_menu(AssetLibraryMenu, Menu): class ASSETLIB_MT_context_menu(AssetLibraryMenu, Menu):
@ -190,7 +189,7 @@ def is_option_region_visible(context, space):
return False return False
for region in context.area.regions: for region in context.area.regions:
if region.type == 'TOOL_PROPS' and region.width <= 1: if region.type == "TOOL_PROPS" and region.width <= 1:
return False return False
return True return True
@ -209,7 +208,7 @@ def draw_assetbrowser_header(self, context):
row = self.layout.row(align=True) row = self.layout.row(align=True)
row.separator() row.separator()
row.operator("assetlib.bundle", icon='UV_SYNC_SELECT', text='').name = lib.name row.operator("assetlib.bundle", icon="UV_SYNC_SELECT", text="").name = lib.name
# op # op
# op.clean = False # op.clean = False
# op.only_recent = True # op.only_recent = True
@ -224,7 +223,7 @@ def draw_assetbrowser_header(self, context):
sub = row.row() sub = row.row()
sub.ui_units_x = 10 sub.ui_units_x = 10
sub.prop(params, "filter_search", text="", icon='VIEWZOOM') sub.prop(params, "filter_search", text="", icon="VIEWZOOM")
row.separator_spacer() row.separator_spacer()
@ -239,17 +238,18 @@ def draw_assetbrowser_header(self, context):
row.operator( row.operator(
"screen.region_toggle", "screen.region_toggle",
text="", text="",
icon='PREFERENCES', icon="PREFERENCES",
depress=is_option_region_visible(context, space_data) depress=is_option_region_visible(context, space_data),
).region_type = 'TOOL_PROPS' ).region_type = "TOOL_PROPS"
### Messagebus subscription to monitor asset library changes. ### Messagebus subscription to monitor asset library changes.
_msgbus_owner = object() _msgbus_owner = object()
def _on_asset_library_changed() -> None: def _on_asset_library_changed() -> None:
"""Update areas when a different asset library is selected.""" """Update areas when a different asset library is selected."""
refresh_area_types = {'DOPESHEET_EDITOR', 'VIEW_3D'} refresh_area_types = {"DOPESHEET_EDITOR", "VIEW_3D"}
for win in bpy.context.window_manager.windows: for win in bpy.context.window_manager.windows:
for area in win.screen.areas: for area in win.screen.areas:
if area.type not in refresh_area_types: if area.type not in refresh_area_types:
@ -257,6 +257,7 @@ def _on_asset_library_changed() -> None:
area.tag_redraw() area.tag_redraw()
def register_message_bus() -> None: def register_message_bus() -> None:
bpy.msgbus.subscribe_rna( bpy.msgbus.subscribe_rna(
@ -264,17 +265,20 @@ def register_message_bus() -> None:
owner=_msgbus_owner, owner=_msgbus_owner,
args=(), args=(),
notify=_on_asset_library_changed, notify=_on_asset_library_changed,
options={'PERSISTENT'}, options={"PERSISTENT"},
) )
def unregister_message_bus() -> None: def unregister_message_bus() -> None:
bpy.msgbus.clear_by_owner(_msgbus_owner) bpy.msgbus.clear_by_owner(_msgbus_owner)
@bpy.app.handlers.persistent @bpy.app.handlers.persistent
def _on_blendfile_load_pre(none, other_none) -> None: def _on_blendfile_load_pre(none, other_none) -> None:
# The parameters are required, but both are None. # The parameters are required, but both are None.
unregister_message_bus() unregister_message_bus()
@bpy.app.handlers.persistent @bpy.app.handlers.persistent
def _on_blendfile_load_post(none, other_none) -> None: def _on_blendfile_load_post(none, other_none) -> None:
# The parameters are required, but both are None. # The parameters are required, but both are None.
@ -285,7 +289,7 @@ classes = (
ASSETLIB_PT_pose_library_editing, ASSETLIB_PT_pose_library_editing,
# ASSETLIB_PT_pose_library_usage, # ASSETLIB_PT_pose_library_usage,
ASSETLIB_MT_context_menu, ASSETLIB_MT_context_menu,
ASSETLIB_PT_libraries ASSETLIB_PT_libraries,
) )
@ -293,8 +297,12 @@ def register() -> None:
for cls in classes: for cls in classes:
bpy.utils.register_class(cls) bpy.utils.register_class(cls)
bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons = bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons = (
bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons = draw_assetbrowser_header bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons
)
bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons = (
draw_assetbrowser_header
)
# WorkSpace.active_pose_asset_index = bpy.props.IntProperty( # WorkSpace.active_pose_asset_index = bpy.props.IntProperty(
# name="Active Pose Asset", # name="Active Pose Asset",
@ -319,7 +327,9 @@ def unregister() -> None:
for cls in reversed(classes): for cls in reversed(classes):
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(cls)
bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons = bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons bpy.types.FILEBROWSER_HT_header.draw_asset_browser_buttons = (
bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons
)
del bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons del bpy.types.FILEBROWSER_HT_header._draw_asset_browser_buttons
unregister_message_bus() unregister_message_bus()

View File

@ -12,17 +12,24 @@ addon_keymaps: List[Tuple[bpy.types.KeyMap, bpy.types.KeyMapItem]] = []
@persistent @persistent
def copy_play_anim(dummy): def copy_play_anim(dummy):
wm = bpy.context.window_manager wm = bpy.context.window_manager
km = wm.keyconfigs.addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER") km = wm.keyconfigs.addon.keymaps.new(
name="File Browser Main", space_type="FILE_BROWSER"
)
km_frames = wm.keyconfigs.user.keymaps.get('Frames') km_frames = wm.keyconfigs.user.keymaps.get("Frames")
if km_frames: if km_frames:
play = km_frames.keymap_items.get('screen.animation_play') play = km_frames.keymap_items.get("screen.animation_play")
if play: if play:
kmi = km.keymap_items.new( kmi = km.keymap_items.new(
"assetlib.play_preview", "assetlib.play_preview",
play.type, play.value, play.type,
any=play.any, shift=play.shift, ctrl=play.ctrl, alt=play.alt, play.value,
oskey=play.oskey, key_modifier=play.key_modifier, any=play.any,
shift=play.shift,
ctrl=play.ctrl,
alt=play.alt,
oskey=play.oskey,
key_modifier=play.key_modifier,
) )
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
@ -33,10 +40,12 @@ def register() -> None:
# This happens when Blender is running in the background. # This happens when Blender is running in the background.
return return
km = wm.keyconfigs.addon.keymaps.new(name="File Browser Main", space_type="FILE_BROWSER") km = wm.keyconfigs.addon.keymaps.new(
name="File Browser Main", space_type="FILE_BROWSER"
)
kmi = km.keymap_items.new("wm.call_menu", "RIGHTMOUSE", "PRESS") kmi = km.keymap_items.new("wm.call_menu", "RIGHTMOUSE", "PRESS")
kmi.properties.name = 'ASSETLIB_MT_context_menu' kmi.properties.name = "ASSETLIB_MT_context_menu"
addon_keymaps.append((km, kmi)) addon_keymaps.append((km, kmi))
kmi = km.keymap_items.new("assetlib.play_preview", "SPACE", "PRESS") kmi = km.keymap_items.new("assetlib.play_preview", "SPACE", "PRESS")
@ -45,12 +54,13 @@ def register() -> None:
# km = addon.keymaps.new(name = "Grease Pencil Stroke Paint Mode", space_type = "EMPTY") # km = addon.keymaps.new(name = "Grease Pencil Stroke Paint Mode", space_type = "EMPTY")
# kmi = km.keymap_items.new('wm.call_panel', type='F2', value='PRESS') # kmi = km.keymap_items.new('wm.call_panel', type='F2', value='PRESS')
if 'copy_play_anim' not in [hand.__name__ for hand in bpy.app.handlers.load_post]: if "copy_play_anim" not in [hand.__name__ for hand in bpy.app.handlers.load_post]:
bpy.app.handlers.load_post.append(copy_play_anim) bpy.app.handlers.load_post.append(copy_play_anim)
def unregister() -> None: def unregister() -> None:
# Clear shortcuts from the keymap. # Clear shortcuts from the keymap.
if 'copy_play_anim' in [hand.__name__ for hand in bpy.app.handlers.load_post]: if "copy_play_anim" in [hand.__name__ for hand in bpy.app.handlers.load_post]:
bpy.app.handlers.load_post.remove(copy_play_anim) bpy.app.handlers.load_post.remove(copy_play_anim)
for km, kmi in addon_keymaps: for km, kmi in addon_keymaps:

View File

@ -1,9 +1,8 @@
from asset_library.library_types import library_type from asset_library.library_types import library_type
from asset_library.library_types import copy_folder from asset_library.library_types import copy_folder
from asset_library.library_types import scan_folder from asset_library.library_types import scan_folder
if 'bpy' in locals(): if "bpy" in locals():
import importlib import importlib
importlib.reload(library_type) importlib.reload(library_type)

View File

@ -1,15 +1,13 @@
""" """
Plugin for making an asset library of all blender file found in a folder Plugin for making an asset library of all blender file found in a folder
""" """
from asset_library.library_types.scan_folder import ScanFolder from asset_library.library_types.scan_folder import ScanFolder
from asset_library.common.bl_utils import load_datablocks from asset_library.common.bl_utils import load_datablocks
from asset_library.common.template import Template from asset_library.common.template import Template
import bpy import bpy
from bpy.props import (StringProperty, IntProperty, BoolProperty) from bpy.props import StringProperty, IntProperty, BoolProperty
import re import re
from pathlib import Path from pathlib import Path
from itertools import groupby from itertools import groupby
@ -24,7 +22,7 @@ from pprint import pprint
class Conform(ScanFolder): class Conform(ScanFolder):
name = "Conform" name = "Conform"
source_directory : StringProperty(subtype='DIR_PATH') source_directory: StringProperty(subtype="DIR_PATH")
target_template_file: StringProperty() target_template_file: StringProperty()
target_template_info: StringProperty() target_template_info: StringProperty()
@ -35,16 +33,18 @@ class Conform(ScanFolder):
layout.prop(self, "source_directory", text="Source : Directory") layout.prop(self, "source_directory", text="Source : Directory")
col = layout.column(align=True) col = layout.column(align=True)
col.prop(self, "source_template_file", icon='COPY_ID', text='Template file') col.prop(self, "source_template_file", icon="COPY_ID", text="Template file")
col.prop(self, "source_template_image", icon='COPY_ID', text='Template image') col.prop(self, "source_template_image", icon="COPY_ID", text="Template image")
col.prop(self, "source_template_video", icon='COPY_ID', text='Template video') col.prop(self, "source_template_video", icon="COPY_ID", text="Template video")
col.prop(self, "source_template_info", icon='COPY_ID', text='Template info') col.prop(self, "source_template_info", icon="COPY_ID", text="Template info")
col = layout.column(align=True) col = layout.column(align=True)
col.prop(self, "target_template_file", icon='COPY_ID', text='Target : Template file') col.prop(
col.prop(self, "target_template_image", icon='COPY_ID', text='Template image') self, "target_template_file", icon="COPY_ID", text="Target : Template file"
col.prop(self, "target_template_video", icon='COPY_ID', text='Template video') )
col.prop(self, "target_template_info", icon='COPY_ID', text='Template info') col.prop(self, "target_template_image", icon="COPY_ID", text="Template image")
col.prop(self, "target_template_video", icon="COPY_ID", text="Template video")
col.prop(self, "target_template_info", icon="COPY_ID", text="Template info")
def get_asset_bundle_path(self, asset_data): def get_asset_bundle_path(self, asset_data):
"""Template file are relative""" """Template file are relative"""
@ -52,26 +52,30 @@ class Conform(ScanFolder):
src_directory = Path(self.source_directory).resolve() src_directory = Path(self.source_directory).resolve()
src_template_file = Template(self.source_template_file) src_template_file = Template(self.source_template_file)
asset_path = Path(asset_data['filepath']).as_posix() asset_path = Path(asset_data["filepath"]).as_posix()
asset_path = self.format_path(asset_path) asset_path = self.format_path(asset_path)
rel_path = asset_path.relative_to(src_directory).as_posix() rel_path = asset_path.relative_to(src_directory).as_posix()
field_data = src_template_file.parse(rel_path) field_data = src_template_file.parse(rel_path)
# field_data = {f"catalog_{k}": v for k, v in field_data.items()} # field_data = {f"catalog_{k}": v for k, v in field_data.items()}
# Change the int in the template by string to allow format # Change the int in the template by string to allow format
# target_template_file = re.sub(r'{(\d+)}', r'{cat\1}', self.target_template_file) # target_template_file = re.sub(r'{(\d+)}', r'{cat\1}', self.target_template_file)
format_data = self.format_asset_data(asset_data) format_data = self.format_asset_data(asset_data)
# format_data['asset_name'] = format_data['asset_name'].lower().replace(' ', '_') # format_data['asset_name'] = format_data['asset_name'].lower().replace(' ', '_')
path = Template(self.target_template_file).format(format_data, **field_data).with_suffix('.blend') path = (
Template(self.target_template_file)
.format(format_data, **field_data)
.with_suffix(".blend")
)
path = Path(self.bundle_directory, path).resolve() path = Path(self.bundle_directory, path).resolve()
return path return path
def set_asset_preview(self, asset, asset_data): def set_asset_preview(self, asset, asset_data):
'''Load an externalize image as preview for an asset using the target template''' """Load an externalize image as preview for an asset using the target template"""
image_template = self.target_template_image image_template = self.target_template_image
if not image_template: if not image_template:
@ -82,18 +86,16 @@ class Conform(ScanFolder):
if image_path: if image_path:
with bpy.context.temp_override(id=asset): with bpy.context.temp_override(id=asset):
bpy.ops.ed.lib_id_load_custom_preview( bpy.ops.ed.lib_id_load_custom_preview(filepath=str(image_path))
filepath=str(image_path)
)
else: else:
print(f'No image found for {image_template} on {asset.name}') print(f"No image found for {image_template} on {asset.name}")
if asset.preview: if asset.preview:
return asset.preview return asset.preview
def generate_previews(self, cache_diff): def generate_previews(self, cache_diff):
print('Generate previews...') print("Generate previews...")
# if cache in (None, ''): # if cache in (None, ''):
# cache = self.fetch() # cache = self.fetch()
@ -103,12 +105,10 @@ class Conform(ScanFolder):
if isinstance(cache, (Path, str)): if isinstance(cache, (Path, str)):
cache_diff = LibraryCacheDiff(cache_diff) cache_diff = LibraryCacheDiff(cache_diff)
# TODO Support all multiple data_type # TODO Support all multiple data_type
for asset_info in cache: for asset_info in cache:
if asset_info.get('type', self.data_type) == 'FILE': if asset_info.get("type", self.data_type) == "FILE":
self.generate_blend_preview(asset_info) self.generate_blend_preview(asset_info)
else: else:
self.generate_asset_preview(asset_info) self.generate_asset_preview(asset_info)
@ -124,41 +124,47 @@ class Conform(ScanFolder):
# camera = scn.camera # camera = scn.camera
data_type = self.data_type # asset_info['data_type'] data_type = self.data_type # asset_info['data_type']
asset_path = self.format_path(asset_info['filepath']) asset_path = self.format_path(asset_info["filepath"])
# Check if a source video exists and if so copying it in the new directory # Check if a source video exists and if so copying it in the new directory
if self.source_template_video and self.target_template_video: if self.source_template_video and self.target_template_video:
for asset_data in asset_info['assets']: for asset_data in asset_info["assets"]:
asset_data = dict(asset_data, filepath=asset_path) asset_data = dict(asset_data, filepath=asset_path)
dst_asset_path = self.get_asset_bundle_path(asset_data) dst_asset_path = self.get_asset_bundle_path(asset_data)
dst_video_path = self.format_path(self.target_template_video, asset_data, filepath=dst_asset_path) dst_video_path = self.format_path(
self.target_template_video, asset_data, filepath=dst_asset_path
)
if dst_video_path.exists(): if dst_video_path.exists():
print(f'The dest video {dst_video_path} already exist') print(f"The dest video {dst_video_path} already exist")
continue continue
src_video_path = self.find_path(self.source_template_video, asset_data) src_video_path = self.find_path(self.source_template_video, asset_data)
if src_video_path: if src_video_path:
print(f'Copy video from {src_video_path} to {dst_video_path}') print(f"Copy video from {src_video_path} to {dst_video_path}")
self.copy_file(src_video_path, dst_video_path) self.copy_file(src_video_path, dst_video_path)
# Check if asset as a preview image or need it to be generated # Check if asset as a preview image or need it to be generated
asset_data_names = {} asset_data_names = {}
if self.target_template_image: if self.target_template_image:
for asset_data in asset_info['assets']: for asset_data in asset_info["assets"]:
asset_data = dict(asset_data, filepath=asset_path) asset_data = dict(asset_data, filepath=asset_path)
name = asset_data['name'] name = asset_data["name"]
dst_asset_path = self.get_asset_bundle_path(asset_data) dst_asset_path = self.get_asset_bundle_path(asset_data)
dst_image_path = self.format_path(self.target_template_image, asset_data, filepath=dst_asset_path) dst_image_path = self.format_path(
self.target_template_image, asset_data, filepath=dst_asset_path
)
if dst_image_path.exists(): if dst_image_path.exists():
print(f'The dest image {dst_image_path} already exist') print(f"The dest image {dst_image_path} already exist")
continue continue
# Check if a source image exists and if so copying it in the new directory # Check if a source image exists and if so copying it in the new directory
if self.source_template_image: if self.source_template_image:
src_image_path = self.find_path(self.source_template_image, asset_data) src_image_path = self.find_path(
self.source_template_image, asset_data
)
if src_image_path: if src_image_path:
if src_image_path.suffix == dst_image_path.suffix: if src_image_path.suffix == dst_image_path.suffix:
@ -173,28 +179,29 @@ class Conform(ScanFolder):
# Store in a dict all asset_data that does not have preview # Store in a dict all asset_data that does not have preview
asset_data_names[name] = dict(asset_data, image_path=dst_image_path) asset_data_names[name] = dict(asset_data, image_path=dst_image_path)
if not asset_data_names: # No preview to generate if not asset_data_names: # No preview to generate
return return
print('Making Preview for', list(asset_data_names.keys())) print("Making Preview for", list(asset_data_names.keys()))
asset_names = list(asset_data_names.keys()) asset_names = list(asset_data_names.keys())
assets = self.load_datablocks(asset_path, names=asset_names, link=True, type=data_type) assets = self.load_datablocks(
asset_path, names=asset_names, link=True, type=data_type
)
for asset in assets: for asset in assets:
if not asset: if not asset:
continue continue
asset_data = asset_data_names[asset.name] asset_data = asset_data_names[asset.name]
image_path = asset_data['image_path'] image_path = asset_data["image_path"]
if asset.preview: if asset.preview:
print(f'Writing asset preview to {image_path}') print(f"Writing asset preview to {image_path}")
self.write_preview(asset.preview, image_path) self.write_preview(asset.preview, image_path)
continue continue
if data_type == 'COLLECTION': if data_type == "COLLECTION":
bpy.ops.object.collection_instance_add(name=asset.name) bpy.ops.object.collection_instance_add(name=asset.name)
@ -205,7 +212,7 @@ class Conform(ScanFolder):
scn.render.filepath = str(image_path) scn.render.filepath = str(image_path)
print(f'Render asset {asset.name} to {image_path}') print(f"Render asset {asset.name} to {image_path}")
bpy.ops.render.render(write_still=True) bpy.ops.render.render(write_still=True)
# instance.user_clear() # instance.user_clear()
@ -213,4 +220,6 @@ class Conform(ScanFolder):
bpy.data.objects.remove(instance) bpy.data.objects.remove(instance)
bpy.ops.outliner.orphans_purge(do_local_ids=True, do_linked_ids=True, do_recursive=True) bpy.ops.outliner.orphans_purge(
do_local_ids=True, do_linked_ids=True, do_recursive=True
)

View File

@ -1,9 +1,7 @@
""" """
Adapter for making an asset library of all blender file found in a folder Adapter for making an asset library of all blender file found in a folder
""" """
from asset_library.library_types.library_type import LibraryType from asset_library.library_types.library_type import LibraryType
from asset_library.common.file_utils import copy_dir from asset_library.common.file_utils import copy_dir
from bpy.props import StringProperty from bpy.props import StringProperty
@ -24,17 +22,14 @@ class CopyFolder(LibraryType):
src = expandvars(self.source_directory) src = expandvars(self.source_directory)
dst = expandvars(self.bundle_directory) dst = expandvars(self.bundle_directory)
includes = [inc.strip() for inc in self.includes.split(',')] includes = [inc.strip() for inc in self.includes.split(",")]
excludes = [ex.strip() for ex in self.excludes.split(',')] excludes = [ex.strip() for ex in self.excludes.split(",")]
print(f'Copy Folder from {src} to {dst}...') print(f"Copy Folder from {src} to {dst}...")
copy_dir( copy_dir(src, dst, only_recent=True, excludes=excludes, includes=includes)
src, dst, only_recent=True,
excludes=excludes, includes=includes
)
def filter_prop(self, prop): def filter_prop(self, prop):
if prop in ('template_info', 'template_video', 'template_image', 'blend_depth'): if prop in ("template_info", "template_video", "template_image", "blend_depth"):
return False return False
return True return True

View File

@ -1,15 +1,13 @@
""" """
Plugin for making an asset library of all blender file found in a folder Plugin for making an asset library of all blender file found in a folder
""" """
from asset_library.library_types.library_type import LibraryType from asset_library.library_types.library_type import LibraryType
from asset_library.common.template import Template from asset_library.common.template import Template
from asset_library.common.file_utils import install_module from asset_library.common.file_utils import install_module
import bpy import bpy
from bpy.props import (StringProperty, IntProperty, BoolProperty) from bpy.props import StringProperty, IntProperty, BoolProperty
import re import re
from pathlib import Path from pathlib import Path
from itertools import groupby from itertools import groupby
@ -27,43 +25,43 @@ class Kitsu(LibraryType):
name = "Kitsu" name = "Kitsu"
template_name: StringProperty() template_name: StringProperty()
template_file: StringProperty() template_file: StringProperty()
source_directory : StringProperty(subtype='DIR_PATH') source_directory: StringProperty(subtype="DIR_PATH")
# blend_depth: IntProperty(default=1) # blend_depth: IntProperty(default=1)
source_template_image: StringProperty() source_template_image: StringProperty()
target_template_image: StringProperty() target_template_image: StringProperty()
url: StringProperty() url: StringProperty()
login: StringProperty() login: StringProperty()
password: StringProperty(subtype='PASSWORD') password: StringProperty(subtype="PASSWORD")
project_name: StringProperty() project_name: StringProperty()
def connect(self, url=None, login=None, password=None): def connect(self, url=None, login=None, password=None):
'''Connect to kitsu api using provided url, login and password''' """Connect to kitsu api using provided url, login and password"""
gazu = install_module('gazu') gazu = install_module("gazu")
urllib3.disable_warnings() urllib3.disable_warnings()
if not self.url: if not self.url:
print(f'Kitsu Url: {self.url} is empty') print(f"Kitsu Url: {self.url} is empty")
return return
url = self.url url = self.url
if not url.endswith('/api'): if not url.endswith("/api"):
url += '/api' url += "/api"
print(f'Info: Setting Host for kitsu {url}') print(f"Info: Setting Host for kitsu {url}")
gazu.client.set_host(url) gazu.client.set_host(url)
if not gazu.client.host_is_up(): if not gazu.client.host_is_up():
print('Error: Kitsu Host is down') print("Error: Kitsu Host is down")
try: try:
print(f'Info: Log in to kitsu as {self.login}') print(f"Info: Log in to kitsu as {self.login}")
res = gazu.log_in(self.login, self.password) res = gazu.log_in(self.login, self.password)
print(f'Info: Sucessfully login to Kitsu as {res["user"]["full_name"]}') print(f'Info: Sucessfully login to Kitsu as {res["user"]["full_name"]}')
return res['user'] return res["user"]
except Exception as e: except Exception as e:
print(f'Error: {traceback.format_exc()}') print(f"Error: {traceback.format_exc()}")
def get_asset_path(self, name, catalog, directory=None): def get_asset_path(self, name, catalog, directory=None):
directory = directory or self.source_directory directory = directory or self.source_directory
@ -72,24 +70,26 @@ class Kitsu(LibraryType):
def get_asset_info(self, data, asset_path): def get_asset_info(self, data, asset_path):
modified = time.time_ns() modified = time.time_ns()
catalog = data['entity_type_name'].title() catalog = data["entity_type_name"].title()
asset_path = self.prop_rel_path(asset_path, 'source_directory') asset_path = self.prop_rel_path(asset_path, "source_directory")
# asset_name = self.norm_file_name(data['name']) # asset_name = self.norm_file_name(data['name'])
asset_info = dict( asset_info = dict(
filepath=asset_path, filepath=asset_path,
modified=modified, modified=modified,
library_id=self.library.id, library_id=self.library.id,
assets=[dict( assets=[
dict(
catalog=catalog, catalog=catalog,
metadata=data.get('data', {}), metadata=data.get("data", {}),
description=data['description'], description=data["description"],
tags=[], tags=[],
type=self.data_type, type=self.data_type,
# image=self.library.template_image, # image=self.library.template_image,
# video=self.library.template_video, # video=self.library.template_video,
name=data['name']) name=data["name"],
] )
],
) )
return asset_info return asset_info
@ -100,29 +100,28 @@ class Kitsu(LibraryType):
# return super().bundle(cache_diff=cache_diff) # return super().bundle(cache_diff=cache_diff)
def set_asset_preview(self, asset, asset_data): def set_asset_preview(self, asset, asset_data):
'''Load an externalize image as preview for an asset using the source template''' """Load an externalize image as preview for an asset using the source template"""
asset_path = self.format_path(Path(asset_data['filepath']).as_posix()) asset_path = self.format_path(Path(asset_data["filepath"]).as_posix())
image_path = self.find_path(self.target_template_image, asset_data, filepath=asset_path) image_path = self.find_path(
self.target_template_image, asset_data, filepath=asset_path
)
if image_path: if image_path:
with bpy.context.temp_override(id=asset): with bpy.context.temp_override(id=asset):
bpy.ops.ed.lib_id_load_custom_preview( bpy.ops.ed.lib_id_load_custom_preview(filepath=str(image_path))
filepath=str(image_path)
)
else: else:
print(f'No image found for {self.target_template_image} on {asset.name}') print(f"No image found for {self.target_template_image} on {asset.name}")
if asset.preview: if asset.preview:
return asset.preview return asset.preview
def generate_previews(self, cache=None): def generate_previews(self, cache=None):
print('Generate previews...') print("Generate previews...")
if cache in (None, ''): if cache in (None, ""):
cache = self.fetch() cache = self.fetch()
elif isinstance(cache, (Path, str)): elif isinstance(cache, (Path, str)):
cache = self.read_cache(cache) cache = self.read_cache(cache)
@ -130,7 +129,7 @@ class Kitsu(LibraryType):
# TODO Support all multiple data_type # TODO Support all multiple data_type
for asset_info in cache: for asset_info in cache:
if asset_info.get('type', self.data_type) == 'FILE': if asset_info.get("type", self.data_type) == "FILE":
self.generate_blend_preview(asset_info) self.generate_blend_preview(asset_info)
else: else:
self.generate_asset_preview(asset_info) self.generate_asset_preview(asset_info)
@ -141,21 +140,22 @@ class Kitsu(LibraryType):
scn = bpy.context.scene scn = bpy.context.scene
vl = bpy.context.view_layer vl = bpy.context.view_layer
asset_path = self.format_path(asset_info['filepath']) asset_path = self.format_path(asset_info["filepath"])
lens = 85 lens = 85
if not asset_path.exists(): if not asset_path.exists():
print(f'Blend file {asset_path} not exit') print(f"Blend file {asset_path} not exit")
return return
asset_data_names = {} asset_data_names = {}
# First check wich assets need a preview # First check wich assets need a preview
for asset_data in asset_info['assets']: for asset_data in asset_info["assets"]:
name = asset_data['name'] name = asset_data["name"]
image_path = self.format_path(self.target_template_image, asset_data, filepath=asset_path) image_path = self.format_path(
self.target_template_image, asset_data, filepath=asset_path
)
if image_path.exists(): if image_path.exists():
continue continue
@ -164,12 +164,14 @@ class Kitsu(LibraryType):
asset_data_names[name] = dict(asset_data, image_path=image_path) asset_data_names[name] = dict(asset_data, image_path=image_path)
if not asset_data_names: if not asset_data_names:
print(f'All previews already existing for {asset_path}') print(f"All previews already existing for {asset_path}")
return return
# asset_names = [a['name'] for a in asset_info['assets']] # asset_names = [a['name'] for a in asset_info['assets']]
asset_names = list(asset_data_names.keys()) asset_names = list(asset_data_names.keys())
assets = self.load_datablocks(asset_path, names=asset_names, link=True, type=data_type) assets = self.load_datablocks(
asset_path, names=asset_names, link=True, type=data_type
)
print(asset_names) print(asset_names)
print(assets) print(assets)
@ -178,12 +180,14 @@ class Kitsu(LibraryType):
if not asset: if not asset:
continue continue
print(f'Generate Preview for asset {asset.name}') print(f"Generate Preview for asset {asset.name}")
asset_data = asset_data_names[asset.name] asset_data = asset_data_names[asset.name]
# print(self.target_template_image, asset_path) # print(self.target_template_image, asset_path)
image_path = self.format_path(self.target_template_image, asset_data, filepath=asset_path) image_path = self.format_path(
self.target_template_image, asset_data, filepath=asset_path
)
# Force redo preview # Force redo preview
# if asset.preview: # if asset.preview:
@ -191,7 +195,7 @@ class Kitsu(LibraryType):
# self.write_preview(asset.preview, image_path) # self.write_preview(asset.preview, image_path)
# continue # continue
if data_type == 'COLLECTION': if data_type == "COLLECTION":
bpy.ops.object.collection_instance_add(name=asset.name) bpy.ops.object.collection_instance_add(name=asset.name)
@ -204,12 +208,13 @@ class Kitsu(LibraryType):
# scn.collection.children.link(asset) # scn.collection.children.link(asset)
scn.render.filepath = str(image_path) scn.render.filepath = str(image_path)
scn.render.image_settings.file_format = self.format_from_ext(image_path.suffix) scn.render.image_settings.file_format = self.format_from_ext(
scn.render.image_settings.color_mode = 'RGBA' image_path.suffix
)
scn.render.image_settings.color_mode = "RGBA"
scn.render.image_settings.quality = 90 scn.render.image_settings.quality = 90
print(f"Render asset {asset.name} to {image_path}")
print(f'Render asset {asset.name} to {image_path}')
bpy.ops.render.render(write_still=True) bpy.ops.render.render(write_still=True)
# instance.user_clear() # instance.user_clear()
@ -217,53 +222,63 @@ class Kitsu(LibraryType):
bpy.data.objects.remove(instance) bpy.data.objects.remove(instance)
bpy.ops.outliner.orphans_purge(do_local_ids=True, do_linked_ids=True, do_recursive=True) bpy.ops.outliner.orphans_purge(
do_local_ids=True, do_linked_ids=True, do_recursive=True
)
def fetch(self): def fetch(self):
"""Gather in a list all assets found in the folder""" """Gather in a list all assets found in the folder"""
print(f'Fetch Assets for {self.library.name}') print(f"Fetch Assets for {self.library.name}")
gazu = install_module('gazu') gazu = install_module("gazu")
self.connect() self.connect()
template_file = Template(self.template_file) template_file = Template(self.template_file)
template_name = Template(self.template_name) template_name = Template(self.template_name)
project = gazu.client.fetch_first('projects', {'name': self.project_name}) project = gazu.client.fetch_first("projects", {"name": self.project_name})
entity_types = gazu.client.fetch_all('entity-types') entity_types = gazu.client.fetch_all("entity-types")
entity_types_ids = {e['id']: e['name'] for e in entity_types} entity_types_ids = {e["id"]: e["name"] for e in entity_types}
cache = self.read_cache() cache = self.read_cache()
for asset_data in gazu.asset.all_assets_for_project(project): for asset_data in gazu.asset.all_assets_for_project(project):
asset_data['entity_type_name'] = entity_types_ids[asset_data.pop('entity_type_id')] asset_data["entity_type_name"] = entity_types_ids[
asset_name = asset_data['name'] asset_data.pop("entity_type_id")
]
asset_name = asset_data["name"]
asset_field_data = dict(asset_name=asset_name, type=asset_data['entity_type_name'], source_directory=self.source_directory) asset_field_data = dict(
asset_name=asset_name,
type=asset_data["entity_type_name"],
source_directory=self.source_directory,
)
try: try:
asset_field_data.update(template_name.parse(asset_name)) asset_field_data.update(template_name.parse(asset_name))
except Exception: except Exception:
print(f'Warning: Could not parse {asset_name} with template {template_name}') print(
f"Warning: Could not parse {asset_name} with template {template_name}"
)
asset_path = template_file.find(asset_field_data) asset_path = template_file.find(asset_field_data)
if not asset_path: if not asset_path:
print(f'Warning: Could not find file for {template_file.format(asset_field_data)}') print(
f"Warning: Could not find file for {template_file.format(asset_field_data)}"
)
continue continue
asset_path = self.prop_rel_path(asset_path, "source_directory")
asset_path = self.prop_rel_path(asset_path, 'source_directory')
asset_cache_data = dict( asset_cache_data = dict(
catalog=asset_data['entity_type_name'].title(), catalog=asset_data["entity_type_name"].title(),
metadata=asset_data.get('data', {}), metadata=asset_data.get("data", {}),
description=asset_data['description'], description=asset_data["description"],
tags=[], tags=[],
type=self.data_type, type=self.data_type,
name=asset_data['name'] name=asset_data["name"],
) )
cache.add_asset_cache(asset_cache_data, filepath=asset_path) cache.add_asset_cache(asset_cache_data, filepath=asset_path)
return cache return cache

View File

@ -1,11 +1,10 @@
# from asset_library.common.functions import (norm_asset_datas,) # from asset_library.common.functions import (norm_asset_datas,)
from asset_library.common.bl_utils import get_addon_prefs, load_datablocks from asset_library.common.bl_utils import get_addon_prefs, load_datablocks
from asset_library.common.file_utils import read_file, write_file from asset_library.common.file_utils import read_file, write_file
from asset_library.common.template import Template from asset_library.common.template import Template
from asset_library.constants import (MODULE_DIR, RESOURCES_DIR) from asset_library.constants import MODULE_DIR, RESOURCES_DIR
from asset_library import (action, collection, file) from asset_library import action, collection, file
from asset_library.common.library_cache import LibraryCacheDiff from asset_library.common.library_cache import LibraryCacheDiff
from bpy.types import PropertyGroup from bpy.types import PropertyGroup
@ -82,20 +81,28 @@ class LibraryType(PropertyGroup):
@property @property
def module_type(self): def module_type(self):
lib_type = self.library.data_type lib_type = self.library.data_type
if lib_type == 'ACTION': if lib_type == "ACTION":
return action return action
elif lib_type == 'FILE': elif lib_type == "FILE":
return file return file
elif lib_type == 'COLLECTION': elif lib_type == "COLLECTION":
return collection return collection
@property @property
def format_data(self): def format_data(self):
"""Dict for formating template""" """Dict for formating template"""
return dict(self.to_dict(), bundle_dir=self.library.bundle_dir, parent=self.library.parent) return dict(
self.to_dict(),
bundle_dir=self.library.bundle_dir,
parent=self.library.parent,
)
def to_dict(self): def to_dict(self):
return {p: getattr(self, p) for p in self.bl_rna.properties.keys() if p !='rna_type'} return {
p: getattr(self, p)
for p in self.bl_rna.properties.keys()
if p != "rna_type"
}
def read_catalog(self): def read_catalog(self):
return self.library.read_catalog() return self.library.read_catalog()
@ -104,10 +111,10 @@ class LibraryType(PropertyGroup):
return self.library.read_cache(filepath=filepath) return self.library.read_cache(filepath=filepath)
def fetch(self): def fetch(self):
raise Exception('This method need to be define in the library_type') raise Exception("This method need to be define in the library_type")
def norm_file_name(self, name): def norm_file_name(self, name):
return name.replace(' ', '_') return name.replace(" ", "_")
def read_file(self, file): def read_file(self, file):
return read_file(file) return read_file(file)
@ -120,25 +127,29 @@ class LibraryType(PropertyGroup):
dst = Path(destination) dst = Path(destination)
if not src.exists(): if not src.exists():
print(f'Cannot copy file {src}: file not exist') print(f"Cannot copy file {src}: file not exist")
return return
dst.parent.mkdir(exist_ok=True, parents=True) dst.parent.mkdir(exist_ok=True, parents=True)
if src == dst: if src == dst:
print(f'Cannot copy file {src}: source and destination are the same') print(f"Cannot copy file {src}: source and destination are the same")
return return
print(f'Copy file from {src} to {dst}') print(f"Copy file from {src} to {dst}")
shutil.copy2(str(src), str(dst)) shutil.copy2(str(src), str(dst))
def load_datablocks(self, src, names=None, type='objects', link=True, expr=None, assets_only=False): def load_datablocks(
self, src, names=None, type="objects", link=True, expr=None, assets_only=False
):
"""Link or append a datablock from a blendfile""" """Link or append a datablock from a blendfile"""
if type.isupper(): if type.isupper():
type = f'{type.lower()}s' type = f"{type.lower()}s"
return load_datablocks(src, names=names, type=type, link=link, expr=expr, assets_only=assets_only) return load_datablocks(
src, names=names, type=type, link=link, expr=expr, assets_only=assets_only
)
def get_asset_data(self, asset): def get_asset_data(self, asset):
"""Extract asset information on a datablock""" """Extract asset information on a datablock"""
@ -153,9 +164,9 @@ class LibraryType(PropertyGroup):
) )
def get_asset_relative_path(self, name, catalog): def get_asset_relative_path(self, name, catalog):
'''Get a relative path for the asset''' """Get a relative path for the asset"""
name = self.norm_file_name(name) name = self.norm_file_name(name)
return Path(catalog, name, name).with_suffix('.blend') return Path(catalog, name, name).with_suffix(".blend")
def get_active_asset_library(self): def get_active_asset_library(self):
prefs = get_addon_prefs() prefs = get_addon_prefs()
@ -165,8 +176,8 @@ class LibraryType(PropertyGroup):
return self return self
lib = None lib = None
if '.library_id' in asset_handle.asset_data: if ".library_id" in asset_handle.asset_data:
lib_id = asset_handle.asset_data['.library_id'] lib_id = asset_handle.asset_data[".library_id"]
lib = next((l for l in prefs.libraries if l.id == lib_id), None) lib = next((l for l in prefs.libraries if l.id == lib_id), None)
if not lib: if not lib:
@ -178,14 +189,14 @@ class LibraryType(PropertyGroup):
return lib return lib
def get_active_asset_path(self): def get_active_asset_path(self):
'''Get the full path of the active asset_handle from the asset brower''' """Get the full path of the active asset_handle from the asset brower"""
prefs = get_addon_prefs() prefs = get_addon_prefs()
asset_handle = bpy.context.asset_file_handle asset_handle = bpy.context.asset_file_handle
lib = self.get_active_asset_library() lib = self.get_active_asset_library()
if 'filepath' in asset_handle.asset_data: if "filepath" in asset_handle.asset_data:
asset_path = asset_handle.asset_data['filepath'] asset_path = asset_handle.asset_data["filepath"]
asset_path = lib.library_type.format_path(asset_path) asset_path = lib.library_type.format_path(asset_path)
else: else:
asset_path = bpy.types.AssetHandle.get_full_library_path( asset_path = bpy.types.AssetHandle.get_full_library_path(
@ -195,30 +206,30 @@ class LibraryType(PropertyGroup):
return asset_path return asset_path
def generate_previews(self): def generate_previews(self):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def get_image_path(self, name, catalog, filepath): def get_image_path(self, name, catalog, filepath):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def get_video_path(self, name, catalog, filepath): def get_video_path(self, name, catalog, filepath):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def new_asset(self, asset, asset_cache): def new_asset(self, asset, asset_cache):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def remove_asset(self, asset, asset_cache): def remove_asset(self, asset, asset_cache):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def set_asset_preview(self, asset, asset_cache): def set_asset_preview(self, asset, asset_cache):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def format_asset_data(self, data): def format_asset_data(self, data):
"""Get a dict for use in template fields""" """Get a dict for use in template fields"""
return { return {
'asset_name': data['name'], "asset_name": data["name"],
'asset_path': Path(data['filepath']), "asset_path": Path(data["filepath"]),
'catalog': data['catalog'], "catalog": data["catalog"],
'catalog_name': data['catalog'].replace('/', '_'), "catalog_name": data["catalog"].replace("/", "_"),
} }
def format_path(self, template, data={}, **kargs): def format_path(self, template, data={}, **kargs):
@ -230,8 +241,8 @@ class LibraryType(PropertyGroup):
else: else:
data = kargs data = kargs
if template.startswith('.'): #the template is relative if template.startswith("."): # the template is relative
template = Path(data['asset_path'], template).as_posix() template = Path(data["asset_path"], template).as_posix()
params = dict( params = dict(
**data, **data,
@ -261,11 +272,7 @@ class LibraryType(PropertyGroup):
Path(asset_path).parent.mkdir(exist_ok=True, parents=True) Path(asset_path).parent.mkdir(exist_ok=True, parents=True)
bpy.data.libraries.write( bpy.data.libraries.write(
str(asset_path), str(asset_path), {asset}, path_remap="NONE", fake_user=True, compress=True
{asset},
path_remap="NONE",
fake_user=True,
compress=True
) )
# def read_catalog(self, directory=None): # def read_catalog(self, directory=None):
@ -321,8 +328,8 @@ class LibraryType(PropertyGroup):
# return write_file(cache_path, list(asset_infos)) # return write_file(cache_path, list(asset_infos))
def prop_rel_path(self, path, prop): def prop_rel_path(self, path, prop):
'''Get a filepath relative to a property of the library_type''' """Get a filepath relative to a property of the library_type"""
field_prop = '{%s}/'%prop field_prop = "{%s}/" % prop
prop_value = getattr(self, prop) prop_value = getattr(self, prop)
prop_value = Path(os.path.expandvars(prop_value)).resolve() prop_value = Path(os.path.expandvars(prop_value)).resolve()
@ -332,15 +339,15 @@ class LibraryType(PropertyGroup):
return field_prop + rel_path return field_prop + rel_path
def format_from_ext(self, ext): def format_from_ext(self, ext):
if ext.startswith('.'): if ext.startswith("."):
ext = ext[1:] ext = ext[1:]
file_format = ext.upper() file_format = ext.upper()
if file_format == 'JPG': if file_format == "JPG":
file_format = 'JPEG' file_format = "JPEG"
elif file_format == 'EXR': elif file_format == "EXR":
file_format = 'OPEN_EXR' file_format = "OPEN_EXR"
return file_format return file_format
@ -373,12 +380,17 @@ class LibraryType(PropertyGroup):
px = [0] * img_size[0] * img_size[1] * 4 px = [0] * img_size[0] * img_size[1] * 4
preview.image_pixels_float.foreach_get(px) preview.image_pixels_float.foreach_get(px)
img = bpy.data.images.new(name=filepath.name, width=img_size[0], height=img_size[1], is_data=True, alpha=True) img = bpy.data.images.new(
name=filepath.name,
width=img_size[0],
height=img_size[1],
is_data=True,
alpha=True,
)
img.pixels.foreach_set(px) img.pixels.foreach_set(px)
self.save_image(img, filepath, remove=True) self.save_image(img, filepath, remove=True)
def draw_header(self, layout): def draw_header(self, layout):
"""Draw the header of the Asset Browser Window""" """Draw the header of the Asset Browser Window"""
# layout.separator() # layout.separator()
@ -390,25 +402,27 @@ class LibraryType(PropertyGroup):
self.module_type.gui.draw_context_menu(layout) self.module_type.gui.draw_context_menu(layout)
def generate_blend_preview(self, asset_info): def generate_blend_preview(self, asset_info):
asset_name = asset_info['name'] asset_name = asset_info["name"]
catalog = asset_info['catalog'] catalog = asset_info["catalog"]
asset_path = self.format_path(asset_info['filepath']) asset_path = self.format_path(asset_info["filepath"])
dst_image_path = self.get_image_path(asset_name, asset_path, catalog) dst_image_path = self.get_image_path(asset_name, asset_path, catalog)
if dst_image_path.exists(): if dst_image_path.exists():
return return
# Check if a source image exists and if so copying it in the new directory # Check if a source image exists and if so copying it in the new directory
src_image_path = asset_info.get('image') src_image_path = asset_info.get("image")
if src_image_path: if src_image_path:
src_image_path = self.get_template_path(src_image_path, asset_name, asset_path, catalog) src_image_path = self.get_template_path(
src_image_path, asset_name, asset_path, catalog
)
if src_image_path and src_image_path.exists(): if src_image_path and src_image_path.exists():
self.copy_file(src_image_path, dst_image_path) self.copy_file(src_image_path, dst_image_path)
return return
print(f'Thumbnailing {asset_path} to {dst_image_path}') print(f"Thumbnailing {asset_path} to {dst_image_path}")
blender_thumbnailer = Path(bpy.app.binary_path).parent / 'blender-thumbnailer' blender_thumbnailer = Path(bpy.app.binary_path).parent / "blender-thumbnailer"
dst_image_path.parent.mkdir(exist_ok=True, parents=True) dst_image_path.parent.mkdir(exist_ok=True, parents=True)
@ -417,7 +431,7 @@ class LibraryType(PropertyGroup):
success = dst_image_path.exists() success = dst_image_path.exists()
if not success: if not success:
empty_preview = RESOURCES_DIR / 'empty_preview.png' empty_preview = RESOURCES_DIR / "empty_preview.png"
self.copy_file(str(empty_preview), str(dst_image_path)) self.copy_file(str(empty_preview), str(dst_image_path))
return success return success
@ -532,14 +546,12 @@ class LibraryType(PropertyGroup):
# def set_asset_catalog(self, asset, asset_data, catalog_data): # def set_asset_catalog(self, asset, asset_data, catalog_data):
# """Find the catalog if already exist or create it""" # """Find the catalog if already exist or create it"""
# catalog_name = asset_data['catalog'] # catalog_name = asset_data['catalog']
# catalog = catalog_data.get(catalog_name) # catalog = catalog_data.get(catalog_name)
# catalog_item = self.catalog.add(asset_data['catalog']) # catalog_item = self.catalog.add(asset_data['catalog'])
# asset.asset_data.catalog_id = catalog_item.id # asset.asset_data.catalog_id = catalog_item.id
# if not catalog: # if not catalog:
# catalog = {'id': str(uuid.uuid4()), 'name': catalog_name} # catalog = {'id': str(uuid.uuid4()), 'name': catalog_name}
# catalog_data[catalog_name] = catalog # catalog_data[catalog_name] = catalog
@ -572,16 +584,20 @@ class LibraryType(PropertyGroup):
blend_name = asset_cache.norm_name blend_name = asset_cache.norm_name
path_parts = catalog_parts[: self.library.blend_depth] path_parts = catalog_parts[: self.library.blend_depth]
return Path(self.bundle_directory, *path_parts, blend_name, blend_name).with_suffix('.blend') return Path(
self.bundle_directory, *path_parts, blend_name, blend_name
).with_suffix(".blend")
def bundle(self, cache_diff=None): def bundle(self, cache_diff=None):
"""Group all new assets in one or multiple blends for the asset browser""" """Group all new assets in one or multiple blends for the asset browser"""
supported_types = ('FILE', 'ACTION', 'COLLECTION') supported_types = ("FILE", "ACTION", "COLLECTION")
supported_operations = ('ADD', 'REMOVE', 'MODIFY') supported_operations = ("ADD", "REMOVE", "MODIFY")
if self.data_type not in supported_types: if self.data_type not in supported_types:
print(f'{self.data_type} is not supported yet supported types are {supported_types}') print(
f"{self.data_type} is not supported yet supported types are {supported_types}"
)
return return
catalog = self.read_catalog() catalog = self.read_catalog()
@ -595,55 +611,64 @@ class LibraryType(PropertyGroup):
# Write the cache in a temporary file for the generate preview script # Write the cache in a temporary file for the generate preview script
tmp_cache_file = cache.write(tmp=True) tmp_cache_file = cache.write(tmp=True)
bpy.ops.assetlib.generate_previews(name=self.library.name, cache=str(tmp_cache_file)) bpy.ops.assetlib.generate_previews(
name=self.library.name, cache=str(tmp_cache_file)
)
elif isinstance(cache_diff, (Path, str)): elif isinstance(cache_diff, (Path, str)):
cache_diff = LibraryCacheDiff(cache_diff).read()#json.loads(Path(cache_diff).read_text(encoding='utf-8')) cache_diff = LibraryCacheDiff(
cache_diff
).read() # json.loads(Path(cache_diff).read_text(encoding='utf-8'))
total_diffs = len(cache_diff) total_diffs = len(cache_diff)
print(f'Total Diffs={total_diffs}') print(f"Total Diffs={total_diffs}")
if total_diffs == 0: if total_diffs == 0:
print('No assets found') print("No assets found")
return return
i = 0 i = 0
for bundle_path, asset_diffs in cache_diff.group_by(self.get_asset_bundle_path): for bundle_path, asset_diffs in cache_diff.group_by(self.get_asset_bundle_path):
if bundle_path.exists(): if bundle_path.exists():
print(f'Opening existing bundle blend: {bundle_path}') print(f"Opening existing bundle blend: {bundle_path}")
bpy.ops.wm.open_mainfile(filepath=str(bundle_path)) bpy.ops.wm.open_mainfile(filepath=str(bundle_path))
else: else:
print(f'Create new bundle blend to: {bundle_path}') print(f"Create new bundle blend to: {bundle_path}")
bpy.ops.wm.read_homefile(use_empty=True) bpy.ops.wm.read_homefile(use_empty=True)
for asset_diff in asset_diffs: for asset_diff in asset_diffs:
if total_diffs <= 100 or i % int(total_diffs / 10) == 0: if total_diffs <= 100 or i % int(total_diffs / 10) == 0:
print(f'Progress: {int(i / total_diffs * 100)+1}') print(f"Progress: {int(i / total_diffs * 100)+1}")
operation = asset_diff.operation operation = asset_diff.operation
asset_cache = asset_diff.asset_cache asset_cache = asset_diff.asset_cache
asset = getattr(bpy.data, self.data_types).get(asset_cache.name) asset = getattr(bpy.data, self.data_types).get(asset_cache.name)
if operation == 'REMOVE': if operation == "REMOVE":
if asset: if asset:
getattr(bpy.data, self.data_types).remove(asset) getattr(bpy.data, self.data_types).remove(asset)
else: else:
print(f'ERROR : Remove Asset: {asset_cache.name} not found in {bundle_path}') print(
f"ERROR : Remove Asset: {asset_cache.name} not found in {bundle_path}"
)
continue continue
elif operation == 'MODIFY': elif operation == "MODIFY":
if not asset: if not asset:
print(f'WARNING: Modifiy Asset: {asset_cache.name} not found in {bundle_path} it will be created') print(
f"WARNING: Modifiy Asset: {asset_cache.name} not found in {bundle_path} it will be created"
)
if operation == 'ADD' or not asset: if operation == "ADD" or not asset:
if asset: if asset:
# raise Exception(f"Asset {asset_data['name']} Already in Blend") # raise Exception(f"Asset {asset_data['name']} Already in Blend")
print(f"Asset {asset_cache.name} Already in Blend") print(f"Asset {asset_cache.name} Already in Blend")
getattr(bpy.data, self.data_types).remove(asset) getattr(bpy.data, self.data_types).remove(asset)
# print(f"INFO: Add new asset: {asset_data['name']}") # print(f"INFO: Add new asset: {asset_data['name']}")
asset = getattr(bpy.data, self.data_types).new(name=asset_cache.name) asset = getattr(bpy.data, self.data_types).new(
name=asset_cache.name
)
asset.asset_mark() asset.asset_mark()
@ -662,12 +687,11 @@ class LibraryType(PropertyGroup):
self.set_asset_tags(asset, asset_cache) self.set_asset_tags(asset, asset_cache)
self.set_asset_info(asset, asset_cache) self.set_asset_info(asset, asset_cache)
i += 1 i += 1
# self.write_asset_preview_file() # self.write_asset_preview_file()
print(f'Saving Blend to {bundle_path}') print(f"Saving Blend to {bundle_path}")
bundle_path.parent.mkdir(exist_ok=True, parents=True) bundle_path.parent.mkdir(exist_ok=True, parents=True)
bpy.ops.wm.save_as_mainfile(filepath=str(bundle_path), compress=True) bpy.ops.wm.save_as_mainfile(filepath=str(bundle_path), compress=True)
@ -678,7 +702,6 @@ class LibraryType(PropertyGroup):
# self.write_catalog(catalog_data) # self.write_catalog(catalog_data)
catalog.write() catalog.write()
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
# def unflatten_cache(self, cache): # def unflatten_cache(self, cache):
@ -769,4 +792,3 @@ class LibraryType(PropertyGroup):
annotations = self.__class__.__annotations__ annotations = self.__class__.__annotations__
for k, v in annotations.items(): for k, v in annotations.items():
layout.prop(self, k, text=bpy.path.display_name(k)) layout.prop(self, k, text=bpy.path.display_name(k))

View File

@ -1,15 +1,13 @@
""" """
Plugin for making an asset library of all blender file found in a folder Plugin for making an asset library of all blender file found in a folder
""" """
from asset_library.library_types.library_type import LibraryType from asset_library.library_types.library_type import LibraryType
from asset_library.common.template import Template from asset_library.common.template import Template
from asset_library.common.file_utils import install_module from asset_library.common.file_utils import install_module
import bpy import bpy
from bpy.props import (StringProperty, IntProperty, BoolProperty, EnumProperty) from bpy.props import StringProperty, IntProperty, BoolProperty, EnumProperty
import re import re
from pathlib import Path from pathlib import Path
from itertools import groupby from itertools import groupby
@ -27,18 +25,25 @@ from pprint import pprint as pp
REQ_HEADERS = requests.utils.default_headers() REQ_HEADERS = requests.utils.default_headers()
REQ_HEADERS.update({"User-Agent": "Blender: PH Assets"}) REQ_HEADERS.update({"User-Agent": "Blender: PH Assets"})
class PolyHaven(LibraryType): class PolyHaven(LibraryType):
name = "Poly Haven" name = "Poly Haven"
# template_name : StringProperty() # template_name : StringProperty()
# template_file : StringProperty() # template_file : StringProperty()
directory : StringProperty(subtype='DIR_PATH') directory: StringProperty(subtype="DIR_PATH")
asset_type : EnumProperty(items=[(i.replace(' ', '_').upper(), i, '') for i in ('HDRIs', 'Models', 'Textures')], default='HDRIS') asset_type: EnumProperty(
items=[
(i.replace(" ", "_").upper(), i, "")
for i in ("HDRIs", "Models", "Textures")
],
default="HDRIS",
)
main_category: StringProperty( main_category: StringProperty(
default='artificial light, natural light, nature, studio, skies, urban' default="artificial light, natural light, nature, studio, skies, urban"
) )
secondary_category: StringProperty( secondary_category: StringProperty(
default='high constrast, low constrast, medium constrast, midday, morning-afternoon, night, sunrise-sunset' default="high constrast, low constrast, medium constrast, midday, morning-afternoon, night, sunrise-sunset"
) )
# blend_depth: IntProperty(default=1) # blend_depth: IntProperty(default=1)
@ -64,32 +69,35 @@ class PolyHaven(LibraryType):
def format_asset_info(self, asset_info, asset_path): def format_asset_info(self, asset_info, asset_path):
# prend un asset info et output un asset description # prend un asset info et output un asset description
asset_path = self.prop_rel_path(asset_path, 'source_directory') asset_path = self.prop_rel_path(asset_path, "source_directory")
modified = asset_info.get('modified', time.time_ns()) modified = asset_info.get("modified", time.time_ns())
return dict( return dict(
filepath=asset_path, filepath=asset_path,
modified=modified, modified=modified,
library_id=self.library.id, library_id=self.library.id,
assets=[dict( assets=[
catalog=asset_data.get('catalog', asset_info['catalog']), dict(
author=asset_data.get('author'), catalog=asset_data.get("catalog", asset_info["catalog"]),
metadata=asset_data.get('metadata', {}), author=asset_data.get("author"),
description=asset_data.get('description', ''), metadata=asset_data.get("metadata", {}),
tags=asset_data.get('tags', []), description=asset_data.get("description", ""),
tags=asset_data.get("tags", []),
type=self.data_type, type=self.data_type,
image=self.template_image, image=self.template_image,
video=self.template_video, video=self.template_video,
name=asset_data['name']) for asset_data in asset_info['assets'] name=asset_data["name"],
] )
for asset_data in asset_info["assets"]
],
) )
def fetch(self): def fetch(self):
"""Gather in a list all assets found in the folder""" """Gather in a list all assets found in the folder"""
print(f'Fetch Assets for {self.library.name}') print(f"Fetch Assets for {self.library.name}")
print('self.asset_type: ', self.asset_type) print("self.asset_type: ", self.asset_type)
url = f"https://api.polyhaven.com/assets?t={self.asset_type.lower()}" url = f"https://api.polyhaven.com/assets?t={self.asset_type.lower()}"
# url2 = f"https://polyhaven.com/{self.asset_type.lower()}" # url2 = f"https://polyhaven.com/{self.asset_type.lower()}"
# url += "&future=true" if early_access else "" # url += "&future=true" if early_access else ""
@ -115,27 +123,26 @@ class PolyHaven(LibraryType):
for asset_info in res.json().values(): for asset_info in res.json().values():
main_category = None main_category = None
secondary_category = None secondary_category = None
for category in asset_info['categories']: for category in asset_info["categories"]:
if category in self.main_category and not main_category: if category in self.main_category and not main_category:
main_category = category main_category = category
if category in self.secondary_category and not secondary_category: if category in self.secondary_category and not secondary_category:
secondary_category = category secondary_category = category
if main_category and secondary_category: if main_category and secondary_category:
catalog = f'{main_category}_{secondary_category}' catalog = f"{main_category}_{secondary_category}"
if not catalog: if not catalog:
return return
asset_path = self.get_asset_path(asset_info['name'], catalog) asset_path = self.get_asset_path(asset_info["name"], catalog)
print('asset_path: ', asset_path) print("asset_path: ", asset_path)
asset_info = self.format_asset_info(asset_info, asset_path) asset_info = self.format_asset_info(asset_info, asset_path)
print('asset_info: ', asset_info) print("asset_info: ", asset_info)
# return self.format_asset_info([asset['name'], self.get_asset_path(asset['name'], catalog) for asset, asset_infos in res.json().items()]) # return self.format_asset_info([asset['name'], self.get_asset_path(asset['name'], catalog) for asset, asset_infos in res.json().items()])
# pp(res.json()) # pp(res.json())
# pp(res2.json()) # pp(res2.json())
# print(res2) # print(res2)
# return asset_infos # return asset_infos

View File

@ -1,15 +1,13 @@
""" """
Plugin for making an asset library of all blender file found in a folder Plugin for making an asset library of all blender file found in a folder
""" """
from asset_library.library_types.library_type import LibraryType from asset_library.library_types.library_type import LibraryType
from asset_library.common.bl_utils import load_datablocks from asset_library.common.bl_utils import load_datablocks
from asset_library.common.template import Template from asset_library.common.template import Template
import bpy import bpy
from bpy.props import (StringProperty, IntProperty, BoolProperty) from bpy.props import StringProperty, IntProperty, BoolProperty
import re import re
from pathlib import Path from pathlib import Path
from itertools import groupby from itertools import groupby
@ -23,7 +21,7 @@ import time
class ScanFolder(LibraryType): class ScanFolder(LibraryType):
name = "Scan Folder" name = "Scan Folder"
source_directory : StringProperty(subtype='DIR_PATH') source_directory: StringProperty(subtype="DIR_PATH")
source_template_file: StringProperty() source_template_file: StringProperty()
source_template_image: StringProperty() source_template_image: StringProperty()
@ -34,10 +32,10 @@ class ScanFolder(LibraryType):
layout.prop(self, "source_directory", text="Source: Directory") layout.prop(self, "source_directory", text="Source: Directory")
col = layout.column(align=True) col = layout.column(align=True)
col.prop(self, "source_template_file", icon='COPY_ID', text='Template file') col.prop(self, "source_template_file", icon="COPY_ID", text="Template file")
col.prop(self, "source_template_image", icon='COPY_ID', text='Template image') col.prop(self, "source_template_image", icon="COPY_ID", text="Template image")
col.prop(self, "source_template_video", icon='COPY_ID', text='Template video') col.prop(self, "source_template_video", icon="COPY_ID", text="Template video")
col.prop(self, "source_template_info", icon='COPY_ID', text='Template info') col.prop(self, "source_template_info", icon="COPY_ID", text="Template info")
def get_asset_path(self, name, catalog, directory=None): def get_asset_path(self, name, catalog, directory=None):
directory = directory or self.source_directory directory = directory or self.source_directory
@ -49,20 +47,26 @@ class ScanFolder(LibraryType):
def get_image_path(self, name, catalog, filepath): def get_image_path(self, name, catalog, filepath):
catalog = self.norm_file_name(catalog) catalog = self.norm_file_name(catalog)
name = self.norm_file_name(name) name = self.norm_file_name(name)
return self.format_path(self.source_template_image, dict(name=name, catalog=catalog, filepath=filepath)) return self.format_path(
self.source_template_image,
dict(name=name, catalog=catalog, filepath=filepath),
)
def get_video_path(self, name, catalog, filepath): def get_video_path(self, name, catalog, filepath):
catalog = self.norm_file_name(catalog) catalog = self.norm_file_name(catalog)
name = self.norm_file_name(name) name = self.norm_file_name(name)
return self.format_path(self.source_template_video, dict(name=name, catalog=catalog, filepath=filepath)) return self.format_path(
self.source_template_video,
dict(name=name, catalog=catalog, filepath=filepath),
)
def new_asset(self, asset, asset_data): def new_asset(self, asset, asset_data):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
def remove_asset(self, asset, asset_data): def remove_asset(self, asset, asset_data):
raise Exception('Need to be defined in the library_type') raise Exception("Need to be defined in the library_type")
''' """
def format_asset_info(self, asset_datas, asset_path, modified=None): def format_asset_info(self, asset_datas, asset_path, modified=None):
asset_path = self.prop_rel_path(asset_path, 'source_directory') asset_path = self.prop_rel_path(asset_path, 'source_directory')
@ -97,10 +101,10 @@ class ScanFolder(LibraryType):
name=asset_data['name']) for asset_data in asset_datas name=asset_data['name']) for asset_data in asset_datas
] ]
) )
''' """
def set_asset_preview(self, asset, asset_cache): def set_asset_preview(self, asset, asset_cache):
'''Load an externalize image as preview for an asset using the source template''' """Load an externalize image as preview for an asset using the source template"""
asset_path = self.format_path(asset_cache.filepath) asset_path = self.format_path(asset_cache.filepath)
@ -108,15 +112,15 @@ class ScanFolder(LibraryType):
if not image_template: if not image_template:
return return
image_path = self.find_path(image_template, asset_cache.to_dict(), filepath=asset_path) image_path = self.find_path(
image_template, asset_cache.to_dict(), filepath=asset_path
)
if image_path: if image_path:
with bpy.context.temp_override(id=asset): with bpy.context.temp_override(id=asset):
bpy.ops.ed.lib_id_load_custom_preview( bpy.ops.ed.lib_id_load_custom_preview(filepath=str(image_path))
filepath=str(image_path)
)
else: else:
print(f'No image found for {image_template} on {asset.name}') print(f"No image found for {image_template} on {asset.name}")
if asset.preview: if asset.preview:
return asset.preview return asset.preview
@ -124,8 +128,8 @@ class ScanFolder(LibraryType):
def bundle(self, cache_diff=None): def bundle(self, cache_diff=None):
"""Group all new assets in one or multiple blends for the asset browser""" """Group all new assets in one or multiple blends for the asset browser"""
if self.data_type not in ('FILE', 'ACTION', 'COLLECTION'): if self.data_type not in ("FILE", "ACTION", "COLLECTION"):
print(f'{self.data_type} is not supported yet') print(f"{self.data_type} is not supported yet")
return return
# catalog_data = self.read_catalog() # catalog_data = self.read_catalog()
@ -140,50 +144,58 @@ class ScanFolder(LibraryType):
# Write the cache in a temporary file for the generate preview script # Write the cache in a temporary file for the generate preview script
tmp_cache_file = cache.write(tmp=True) tmp_cache_file = cache.write(tmp=True)
bpy.ops.assetlib.generate_previews(name=self.library.name, cache=str(tmp_cache_file)) bpy.ops.assetlib.generate_previews(
name=self.library.name, cache=str(tmp_cache_file)
)
elif isinstance(cache_diff, (Path, str)): elif isinstance(cache_diff, (Path, str)):
cache_diff = json.loads(Path(cache_diff).read_text(encoding='utf-8')) cache_diff = json.loads(Path(cache_diff).read_text(encoding="utf-8"))
if self.library.blend_depth == 0: if self.library.blend_depth == 0:
raise Exception('Blender depth must be 1 at min') raise Exception("Blender depth must be 1 at min")
total_assets = len(cache_diff) total_assets = len(cache_diff)
print(f'total_assets={total_assets}') print(f"total_assets={total_assets}")
if total_assets == 0: if total_assets == 0:
print('No assets found') print("No assets found")
return return
i = 0 i = 0
for blend_path, asset_cache_diffs in cache_diff.group_by(key=self.get_asset_bundle_path): for blend_path, asset_cache_diffs in cache_diff.group_by(
key=self.get_asset_bundle_path
):
if blend_path.exists(): if blend_path.exists():
print(f'Opening existing bundle blend: {blend_path}') print(f"Opening existing bundle blend: {blend_path}")
bpy.ops.wm.open_mainfile(filepath=str(blend_path)) bpy.ops.wm.open_mainfile(filepath=str(blend_path))
else: else:
print(f'Create new bundle blend to: {blend_path}') print(f"Create new bundle blend to: {blend_path}")
bpy.ops.wm.read_homefile(use_empty=True) bpy.ops.wm.read_homefile(use_empty=True)
for asset_cache_diff in asset_cache_diffs: for asset_cache_diff in asset_cache_diffs:
if total_assets <= 100 or i % int(total_assets / 10) == 0: if total_assets <= 100 or i % int(total_assets / 10) == 0:
print(f'Progress: {int(i / total_assets * 100)+1}') print(f"Progress: {int(i / total_assets * 100)+1}")
operation = asset_cache_diff.operation operation = asset_cache_diff.operation
asset_cache = asset_cache_diff.asset_cache asset_cache = asset_cache_diff.asset_cache
asset_name = asset_cache.name asset_name = asset_cache.name
asset = getattr(bpy.data, self.data_types).get(asset_name) asset = getattr(bpy.data, self.data_types).get(asset_name)
if operation == 'REMOVE': if operation == "REMOVE":
if asset: if asset:
getattr(bpy.data, self.data_types).remove(asset) getattr(bpy.data, self.data_types).remove(asset)
else: else:
print(f'ERROR : Remove Asset: {asset_name} not found in {blend_path}') print(
f"ERROR : Remove Asset: {asset_name} not found in {blend_path}"
)
continue continue
if asset_cache_diff.operation == 'MODIFY' and not asset: if asset_cache_diff.operation == "MODIFY" and not asset:
print(f'WARNING: Modifiy Asset: {asset_name} not found in {blend_path} it will be created') print(
f"WARNING: Modifiy Asset: {asset_name} not found in {blend_path} it will be created"
)
if operation == 'ADD' or not asset: if operation == "ADD" or not asset:
if asset: if asset:
# raise Exception(f"Asset {asset_name} Already in Blend") # raise Exception(f"Asset {asset_name} Already in Blend")
print(f"Asset {asset_name} Already in Blend") print(f"Asset {asset_name} Already in Blend")
@ -192,7 +204,9 @@ class ScanFolder(LibraryType):
# print(f"INFO: Add new asset: {asset_name}") # print(f"INFO: Add new asset: {asset_name}")
asset = getattr(bpy.data, self.data_types).new(name=asset_name) asset = getattr(bpy.data, self.data_types).new(name=asset_name)
else: else:
print(f'operation {operation} not supported should be in (ADD, REMOVE, MODIFY)') print(
f"operation {operation} not supported should be in (ADD, REMOVE, MODIFY)"
)
continue continue
asset.asset_mark() asset.asset_mark()
@ -205,12 +219,11 @@ class ScanFolder(LibraryType):
i += 1 i += 1
print(f'Saving Blend to {blend_path}') print(f"Saving Blend to {blend_path}")
blend_path.parent.mkdir(exist_ok=True, parents=True) blend_path.parent.mkdir(exist_ok=True, parents=True)
bpy.ops.wm.save_as_mainfile(filepath=str(blend_path), compress=True) bpy.ops.wm.save_as_mainfile(filepath=str(blend_path), compress=True)
# If the variable cache_diff was given we need to update the cache with the diff # If the variable cache_diff was given we need to update the cache with the diff
if cache is None: if cache is None:
cache = self.read_cache() cache = self.read_cache()
@ -226,7 +239,7 @@ class ScanFolder(LibraryType):
def fetch(self): def fetch(self):
"""Gather in a list all assets found in the folder""" """Gather in a list all assets found in the folder"""
print(f'Fetch Assets for {self.library.name}') print(f"Fetch Assets for {self.library.name}")
source_directory = Path(self.source_directory) source_directory = Path(self.source_directory)
template_file = Template(self.source_template_file) template_file = Template(self.source_template_file)
@ -237,21 +250,23 @@ class ScanFolder(LibraryType):
cache = self.read_cache() cache = self.read_cache()
print(f'Search for blend using glob template: {template_file.glob_pattern}') print(f"Search for blend using glob template: {template_file.glob_pattern}")
print(f'Scanning Folder {source_directory}...') print(f"Scanning Folder {source_directory}...")
# new_cache = LibraryCache() # new_cache = LibraryCache()
for asset_path in template_file.glob(source_directory): for asset_path in template_file.glob(source_directory):
source_rel_path = self.prop_rel_path(asset_path, 'source_directory') source_rel_path = self.prop_rel_path(asset_path, "source_directory")
modified = asset_path.stat().st_mtime_ns modified = asset_path.stat().st_mtime_ns
# Check if the asset description as already been cached # Check if the asset description as already been cached
file_cache = next((a for a in cache if a.filepath == source_rel_path), None) file_cache = next((a for a in cache if a.filepath == source_rel_path), None)
if file_cache: if file_cache:
if file_cache.modified >= modified: #print(asset_path, 'is skipped because not modified') if (
file_cache.modified >= modified
): # print(asset_path, 'is skipped because not modified')
continue continue
else: else:
file_cache = cache.add(filepath=source_rel_path) file_cache = cache.add(filepath=source_rel_path)
@ -260,17 +275,16 @@ class ScanFolder(LibraryType):
field_data = template_file.parse(rel_path) field_data = template_file.parse(rel_path)
# Create the catalog path from the actual path of the asset # Create the catalog path from the actual path of the asset
catalog = [v for k,v in sorted(field_data.items()) if re.findall('cat[0-9]+', k)] catalog = [
v for k, v in sorted(field_data.items()) if re.findall("cat[0-9]+", k)
]
# catalogs = [c.replace('_', ' ').title() for c in catalogs] # catalogs = [c.replace('_', ' ').title() for c in catalogs]
asset_name = field_data.get('asset_name', asset_path.stem) asset_name = field_data.get("asset_name", asset_path.stem)
if self.data_type == 'FILE': if self.data_type == "FILE":
file_cache.set_data( file_cache.set_data(
name=asset_name, name=asset_name, type="FILE", catalog=catalog, modified=modified
type='FILE',
catalog=catalog,
modified=modified
) )
continue continue
@ -282,9 +296,11 @@ class ScanFolder(LibraryType):
# continue # continue
# Scan the blend file for assets inside # Scan the blend file for assets inside
print(f'Scanning blendfile {asset_path}...') print(f"Scanning blendfile {asset_path}...")
assets = self.load_datablocks(asset_path, type=self.data_types, link=True, assets_only=True) assets = self.load_datablocks(
print(f'Found {len(assets)} {self.data_types} inside') asset_path, type=self.data_types, link=True, assets_only=True
)
print(f"Found {len(assets)} {self.data_types} inside")
for asset in assets: for asset in assets:
# catalog_path = catalog_ids.get(asset.asset_data.catalog_id) # catalog_path = catalog_ids.get(asset.asset_data.catalog_id)
@ -298,4 +314,3 @@ class ScanFolder(LibraryType):
getattr(bpy.data, self.data_types).remove(asset) getattr(bpy.data, self.data_types).remove(asset)
return cache return cache

View File

@ -1,6 +1,5 @@
from typing import Set from typing import Set
# import shutil # import shutil
from pathlib import Path from pathlib import Path
import subprocess import subprocess
@ -11,11 +10,7 @@ import json
import bpy import bpy
from bpy_extras import asset_utils from bpy_extras import asset_utils
from bpy.types import Context, Operator from bpy.types import Context, Operator
from bpy.props import ( from bpy.props import BoolProperty, EnumProperty, StringProperty, IntProperty
BoolProperty,
EnumProperty,
StringProperty,
IntProperty)
# from asset_library.constants import (DATA_TYPES, DATA_TYPE_ITEMS, MODULE_DIR) # from asset_library.constants import (DATA_TYPES, DATA_TYPE_ITEMS, MODULE_DIR)
import asset_library import asset_library
@ -26,7 +21,8 @@ from asset_library.common.bl_utils import (
get_view3d_persp, get_view3d_persp,
# suitable_areas, # suitable_areas,
refresh_asset_browsers, refresh_asset_browsers,
load_datablocks) load_datablocks,
)
from asset_library.common.file_utils import open_blender_file, synchronize from asset_library.common.file_utils import open_blender_file, synchronize
from asset_library.common.functions import get_active_library, asset_warning_callback from asset_library.common.functions import get_active_library, asset_warning_callback
@ -42,8 +38,8 @@ import bgl
class ASSETLIB_OT_remove_assets(Operator): class ASSETLIB_OT_remove_assets(Operator):
bl_idname = "assetlib.remove_assets" bl_idname = "assetlib.remove_assets"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Remove Assets' bl_label = "Remove Assets"
bl_description = 'Remove Selected Assets' bl_description = "Remove Selected Assets"
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
@ -51,7 +47,7 @@ class ASSETLIB_OT_remove_assets(Operator):
return False return False
sp = context.space_data sp = context.space_data
if sp.params.asset_library_ref == 'LOCAL': if sp.params.asset_library_ref == "LOCAL":
return False return False
return True return True
@ -65,15 +61,19 @@ class ASSETLIB_OT_remove_assets(Operator):
catalog = lib.read_catalog() catalog = lib.read_catalog()
if not catalog.context.item: if not catalog.context.item:
self.report({'ERROR'}, 'The active asset is not in the catalog') self.report({"ERROR"}, "The active asset is not in the catalog")
return {'CANCELLED'} return {"CANCELLED"}
asset_name = context.asset_file_handle.name asset_name = context.asset_file_handle.name
asset_path = lib_type.format_path(asset.asset_data['filepath']) asset_path = lib_type.format_path(asset.asset_data["filepath"])
asset_catalog = catalog.context.path asset_catalog = catalog.context.path
img_path = lib_type.get_image_path(name=asset_name, catalog=asset_catalog, filepath=asset_path) img_path = lib_type.get_image_path(
video_path = lib_type.get_video_path(name=asset_name, catalog=asset_catalog, filepath=asset_path) name=asset_name, catalog=asset_catalog, filepath=asset_path
)
video_path = lib_type.get_video_path(
name=asset_name, catalog=asset_catalog, filepath=asset_path
)
if asset_path and asset_path.exists(): if asset_path and asset_path.exists():
asset_path.unlink() asset_path.unlink()
@ -90,7 +90,7 @@ class ASSETLIB_OT_remove_assets(Operator):
bpy.ops.assetlib.bundle(name=lib.name, blocking=True) bpy.ops.assetlib.bundle(name=lib.name, blocking=True)
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_edit_data(Operator): class ASSETLIB_OT_edit_data(Operator):
@ -99,12 +99,18 @@ class ASSETLIB_OT_edit_data(Operator):
bl_description = "Edit Current Asset Data" bl_description = "Edit Current Asset Data"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
warning: StringProperty(name='') warning: StringProperty(name="")
path: StringProperty(name='Path') path: StringProperty(name="Path")
catalog: StringProperty(name='Catalog', update=asset_warning_callback, options={'TEXTEDIT_UPDATE'}) catalog: StringProperty(
name: StringProperty(name='Name', update=asset_warning_callback, options={'TEXTEDIT_UPDATE'}) name="Catalog", update=asset_warning_callback, options={"TEXTEDIT_UPDATE"}
tags: StringProperty(name='Tags', description='Tags need to separate with a comma (,)') )
description: StringProperty(name='Description') name: StringProperty(
name="Name", update=asset_warning_callback, options={"TEXTEDIT_UPDATE"}
)
tags: StringProperty(
name="Tags", description="Tags need to separate with a comma (,)"
)
description: StringProperty(name="Description")
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
@ -120,11 +126,13 @@ class ASSETLIB_OT_edit_data(Operator):
lib = prefs.libraries[lib.store_library] lib = prefs.libraries[lib.store_library]
new_name = lib.library_type.norm_file_name(self.name) new_name = lib.library_type.norm_file_name(self.name)
new_asset_path = lib.library_type.get_asset_path(name=new_name, catalog=self.catalog) new_asset_path = lib.library_type.get_asset_path(
name=new_name, catalog=self.catalog
)
# asset_data = lib.library_type.get_asset_data(self.asset) # asset_data = lib.library_type.get_asset_data(self.asset)
asset_data = dict( asset_data = dict(
tags=[t.strip() for t in self.tags.split(',') if t], tags=[t.strip() for t in self.tags.split(",") if t],
description=self.description, description=self.description,
) )
@ -137,11 +145,15 @@ class ASSETLIB_OT_edit_data(Operator):
lib.library_type.write_asset(asset=self.asset, asset_path=new_asset_path) lib.library_type.write_asset(asset=self.asset, asset_path=new_asset_path)
if self.old_image_path.exists(): if self.old_image_path.exists():
new_img_path = lib.library_type.get_image_path(new_name, self.catalog, new_asset_path) new_img_path = lib.library_type.get_image_path(
new_name, self.catalog, new_asset_path
)
self.old_image_path.rename(new_img_path) self.old_image_path.rename(new_img_path)
if self.old_video_path.exists(): if self.old_video_path.exists():
new_video_path = lib.library_type.get_video_path(new_name, self.catalog, new_asset_path) new_video_path = lib.library_type.get_video_path(
new_name, self.catalog, new_asset_path
)
self.old_video_path.rename(new_video_path) self.old_video_path.rename(new_video_path)
# if self.old_description_path.exists(): # if self.old_description_path.exists():
@ -152,22 +164,32 @@ class ASSETLIB_OT_edit_data(Operator):
except Exception: # The folder is not empty except Exception: # The folder is not empty
pass pass
diff_path = Path(bpy.app.tempdir, 'diff.json') diff_path = Path(bpy.app.tempdir, "diff.json")
diff = [dict(name=self.old_asset_name, catalog=self.old_catalog, filepath=str(self.old_asset_path), operation='REMOVE')] diff = [
dict(
name=self.old_asset_name,
catalog=self.old_catalog,
filepath=str(self.old_asset_path),
operation="REMOVE",
)
]
asset_data = lib.library_type.get_asset_data(self.asset) asset_data = lib.library_type.get_asset_data(self.asset)
diff += [dict(asset_data, diff += [
dict(
asset_data,
image=str(new_img_path), image=str(new_img_path),
filepath=str(new_asset_path), filepath=str(new_asset_path),
type=lib.data_type, type=lib.data_type,
library_id=lib.id, library_id=lib.id,
catalog=self.catalog, catalog=self.catalog,
operation='ADD' operation="ADD",
)] )
]
print(diff) print(diff)
diff_path.write_text(json.dumps(diff, indent=4), encoding='utf-8') diff_path.write_text(json.dumps(diff, indent=4), encoding="utf-8")
bpy.ops.assetlib.bundle(name=lib.name, diff=str(diff_path), blocking=True) bpy.ops.assetlib.bundle(name=lib.name, diff=str(diff_path), blocking=True)
@ -182,12 +204,12 @@ class ASSETLIB_OT_edit_data(Operator):
lib = get_active_library() lib = get_active_library()
if lib.merge_libraries: if lib.merge_libraries:
layout.prop(lib, 'store_library', expand=False) layout.prop(lib, "store_library", expand=False)
layout.prop(self, "catalog", text="Catalog") layout.prop(self, "catalog", text="Catalog")
layout.prop(self, "name", text="Name") layout.prop(self, "name", text="Name")
layout.prop(self, 'tags') layout.prop(self, "tags")
layout.prop(self, 'description') layout.prop(self, "description")
# layout.prop() # layout.prop()
@ -200,7 +222,7 @@ class ASSETLIB_OT_edit_data(Operator):
col.label(text=self.path) col.label(text=self.path)
if self.warning: if self.warning:
col.label(icon='ERROR', text=self.warning) col.label(icon="ERROR", text=self.warning)
def invoke(self, context, event): def invoke(self, context, event):
@ -213,53 +235,58 @@ class ASSETLIB_OT_edit_data(Operator):
asset_handle = context.asset_file_handle asset_handle = context.asset_file_handle
catalog_file = lib.library_type.read_catalog() catalog_file = lib.library_type.read_catalog()
catalog_ids = {v['id']: {'path': k, 'name': v['name']} for k,v in catalog_file.items()} catalog_ids = {
v["id"]: {"path": k, "name": v["name"]} for k, v in catalog_file.items()
}
# asset_handle = context.asset_file_handle # asset_handle = context.asset_file_handle
self.old_asset_name = asset_handle.name self.old_asset_name = asset_handle.name
self.old_asset_path = lib.library_type.get_active_asset_path() self.old_asset_path = lib.library_type.get_active_asset_path()
self.asset = load_datablocks(self.old_asset_path, self.old_asset_name, type=lib.data_types) self.asset = load_datablocks(
self.old_asset_path, self.old_asset_name, type=lib.data_types
)
if not self.asset: if not self.asset:
self.report({'ERROR'}, 'No asset found') self.report({"ERROR"}, "No asset found")
self.name = self.old_asset_name self.name = self.old_asset_name
self.description = asset_handle.asset_data.description self.description = asset_handle.asset_data.description
tags = [t.strip() for t in self.asset.asset_data.tags.keys() if t] tags = [t.strip() for t in self.asset.asset_data.tags.keys() if t]
self.tags = ', '.join(tags) self.tags = ", ".join(tags)
# asset_path # asset_path
self.old_catalog = catalog_ids[asset_handle.asset_data.catalog_id]['path'] self.old_catalog = catalog_ids[asset_handle.asset_data.catalog_id]["path"]
self.catalog = self.old_catalog self.catalog = self.old_catalog
self.old_image_path = lib.library_type.get_image_path(name=self.name, catalog=self.catalog, filepath=self.old_asset_path) self.old_image_path = lib.library_type.get_image_path(
self.old_video_path = lib.library_type.get_video_path(name=self.name, catalog=self.catalog, filepath=self.old_asset_path) name=self.name, catalog=self.catalog, filepath=self.old_asset_path
)
self.old_video_path = lib.library_type.get_video_path(
name=self.name, catalog=self.catalog, filepath=self.old_asset_path
)
# self.old_description_path = lib.library_type.get_description_path(self.old_asset_path) # self.old_description_path = lib.library_type.get_description_path(self.old_asset_path)
# self.old_asset_info = lib.library_type.read_asset_info_file(self.old_asset_path) # self.old_asset_info = lib.library_type.read_asset_info_file(self.old_asset_path)
# self.old_asset_info = lib.library_type.norm_asset_datas([self.old_asset_info])[0] # self.old_asset_info = lib.library_type.norm_asset_datas([self.old_asset_info])[0]
return context.window_manager.invoke_props_dialog(self, width=450) return context.window_manager.invoke_props_dialog(self, width=450)
def cancel(self, context): def cancel(self, context):
print('Cancel Edit Data, removing the asset') print("Cancel Edit Data, removing the asset")
lib = get_active_library() lib = get_active_library()
active_lib = lib.library_type.get_active_asset_library() active_lib = lib.library_type.get_active_asset_library()
getattr(bpy.data, active_lib.data_types).remove(self.asset) getattr(bpy.data, active_lib.data_types).remove(self.asset)
class ASSETLIB_OT_remove_user_library(Operator): class ASSETLIB_OT_remove_user_library(Operator):
bl_idname = "assetlib.remove_user_library" bl_idname = "assetlib.remove_user_library"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Remove User Library' bl_label = "Remove User Library"
bl_description = 'Remove User Library' bl_description = "Remove User Library"
index: IntProperty(default=-1) index: IntProperty(default=-1)
@ -268,15 +295,14 @@ class ASSETLIB_OT_remove_user_library(Operator):
prefs.user_libraries.remove(self.index) prefs.user_libraries.remove(self.index)
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_add_user_library(Operator): class ASSETLIB_OT_add_user_library(Operator):
bl_idname = "assetlib.add_user_library" bl_idname = "assetlib.add_user_library"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Add User Library' bl_label = "Add User Library"
bl_description = 'Add User Library' bl_description = "Add User Library"
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
prefs = get_addon_prefs() prefs = get_addon_prefs()
@ -284,14 +310,14 @@ class ASSETLIB_OT_add_user_library(Operator):
lib = prefs.user_libraries.add() lib = prefs.user_libraries.add()
lib.expand = True lib.expand = True
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_open_blend(Operator): class ASSETLIB_OT_open_blend(Operator):
bl_idname = "assetlib.open_blend" bl_idname = "assetlib.open_blend"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Open Blender File' bl_label = "Open Blender File"
bl_description = 'Open blender file' bl_description = "Open blender file"
# filepath : StringProperty(subtype='FILE_PATH') # filepath : StringProperty(subtype='FILE_PATH')
@ -307,21 +333,21 @@ class ASSETLIB_OT_open_blend(Operator):
open_blender_file(filepath) open_blender_file(filepath)
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_set_paths(Operator): class ASSETLIB_OT_set_paths(Operator):
bl_idname = "assetlib.set_paths" bl_idname = "assetlib.set_paths"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Set Paths' bl_label = "Set Paths"
bl_description = 'Set Library Paths' bl_description = "Set Library Paths"
name: StringProperty() name: StringProperty()
all: BoolProperty(default=False) all: BoolProperty(default=False)
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
prefs = get_addon_prefs() prefs = get_addon_prefs()
print('Set Paths') print("Set Paths")
if self.all: if self.all:
libs = prefs.libraries libs = prefs.libraries
else: else:
@ -331,20 +357,25 @@ class ASSETLIB_OT_set_paths(Operator):
lib.clear_library_path() lib.clear_library_path()
lib.set_library_path() lib.set_library_path()
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_bundle_library(Operator): class ASSETLIB_OT_bundle_library(Operator):
bl_idname = "assetlib.bundle" bl_idname = "assetlib.bundle"
bl_options = {"INTERNAL"} bl_options = {"INTERNAL"}
bl_label = 'Bundle Library' bl_label = "Bundle Library"
bl_description = 'Bundle all matching asset found inside one blend' bl_description = "Bundle all matching asset found inside one blend"
name: StringProperty() name: StringProperty()
diff: StringProperty() diff: StringProperty()
blocking: BoolProperty(default=False) blocking: BoolProperty(default=False)
mode : EnumProperty(items=[(i.replace(' ', '_').upper(), i, '') for i in ('None', 'All', 'Auto Bundle')], default='NONE') mode: EnumProperty(
directory : StringProperty(subtype='DIR_PATH') items=[
(i.replace(" ", "_").upper(), i, "") for i in ("None", "All", "Auto Bundle")
],
default="NONE",
)
directory: StringProperty(subtype="DIR_PATH")
# conform : BoolProperty(default=False) # conform : BoolProperty(default=False)
# def refresh(self): # def refresh(self):
# for area in suitable_areas(bpy.context.screen): # for area in suitable_areas(bpy.context.screen):
@ -358,9 +389,9 @@ class ASSETLIB_OT_bundle_library(Operator):
if self.name: if self.name:
libs += [prefs.libraries[self.name]] libs += [prefs.libraries[self.name]]
if self.mode == 'ALL': if self.mode == "ALL":
libs += prefs.libraries.values() libs += prefs.libraries.values()
elif self.mode == 'AUTO_BUNDLE': elif self.mode == "AUTO_BUNDLE":
libs += [l for l in prefs.libraries if l.auto_bundle] libs += [l for l in prefs.libraries if l.auto_bundle]
if not libs: if not libs:
@ -368,9 +399,10 @@ class ASSETLIB_OT_bundle_library(Operator):
lib_datas = [l.to_dict() for l in libs] lib_datas = [l.to_dict() for l in libs]
print(f'Bundle Libraries: {[l.name for l in libs]}') print(f"Bundle Libraries: {[l.name for l in libs]}")
script_code = dedent(f""" script_code = dedent(
f"""
import bpy import bpy
prefs = bpy.context.preferences.addons["asset_library"].preferences prefs = bpy.context.preferences.addons["asset_library"].preferences
@ -380,9 +412,10 @@ class ASSETLIB_OT_bundle_library(Operator):
lib.library_type.bundle(cache_diff='{self.diff}') lib.library_type.bundle(cache_diff='{self.diff}')
bpy.ops.wm.quit_blender() bpy.ops.wm.quit_blender()
""") """
)
script_path = Path(bpy.app.tempdir) / 'bundle_library.py' script_path = Path(bpy.app.tempdir) / "bundle_library.py"
script_path.write_text(script_code) script_path.write_text(script_code)
print(script_code) print(script_code)
@ -398,31 +431,31 @@ class ASSETLIB_OT_bundle_library(Operator):
else: else:
subprocess.Popen(cmd) subprocess.Popen(cmd)
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_reload_addon(Operator): class ASSETLIB_OT_reload_addon(Operator):
bl_idname = "assetlib.reload_addon" bl_idname = "assetlib.reload_addon"
bl_options = {"UNDO"} bl_options = {"UNDO"}
bl_label = 'Reload Asset Library Addon' bl_label = "Reload Asset Library Addon"
bl_description = 'Reload The Asset Library Addon and the addapters' bl_description = "Reload The Asset Library Addon and the addapters"
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
print('Execute reload') print("Execute reload")
asset_library.unregister() asset_library.unregister()
importlib.reload(asset_library) importlib.reload(asset_library)
asset_library.register() asset_library.register()
return {'FINISHED'} return {"FINISHED"}
class ASSETLIB_OT_diff(Operator): class ASSETLIB_OT_diff(Operator):
bl_idname = "assetlib.diff" bl_idname = "assetlib.diff"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Synchronize' bl_label = "Synchronize"
bl_description = 'Synchronize Action Lib to Local Directory' bl_description = "Synchronize Action Lib to Local Directory"
name: StringProperty() name: StringProperty()
conform: BoolProperty(default=False) conform: BoolProperty(default=False)
@ -433,7 +466,8 @@ class ASSETLIB_OT_diff(Operator):
lib = prefs.libraries.get(self.name) lib = prefs.libraries.get(self.name)
lib.library_type.diff() lib.library_type.diff()
return {'FINISHED'} return {"FINISHED"}
''' '''
class ASSETLIB_OT_conform_library(Operator): class ASSETLIB_OT_conform_library(Operator):
@ -482,6 +516,7 @@ class ASSETLIB_OT_conform_library(Operator):
return {'RUNNING_MODAL'} return {'RUNNING_MODAL'}
''' '''
class ASSETLIB_OT_make_custom_preview(Operator): class ASSETLIB_OT_make_custom_preview(Operator):
bl_idname = "assetlib.make_custom_preview" bl_idname = "assetlib.make_custom_preview"
bl_label = "Custom Preview" bl_label = "Custom Preview"
@ -491,15 +526,15 @@ class ASSETLIB_OT_make_custom_preview(Operator):
modal: BoolProperty(default=False) modal: BoolProperty(default=False)
def modal(self, context, event): def modal(self, context, event):
if event.type in {'ESC'}: # Cancel if event.type in {"ESC"}: # Cancel
self.restore() self.restore()
return {'CANCELLED'} return {"CANCELLED"}
elif event.type in {'RET', 'NUMPAD_ENTER'}: # Cancel elif event.type in {"RET", "NUMPAD_ENTER"}: # Cancel
return self.execute(context) return self.execute(context)
# return {'FINISHED'} # return {'FINISHED'}
return {'PASS_THROUGH'} return {"PASS_THROUGH"}
def execute(self, context): def execute(self, context):
@ -509,12 +544,11 @@ class ASSETLIB_OT_make_custom_preview(Operator):
img_path = context.scene.render.filepath img_path = context.scene.render.filepath
# print('Load Image to previews') # print('Load Image to previews')
prefs.previews.load(Path(img_path).stem, img_path, 'IMAGE') prefs.previews.load(Path(img_path).stem, img_path, "IMAGE")
# img = bpy.data.images.load(context.scene.render.filepath) # img = bpy.data.images.load(context.scene.render.filepath)
# img.update() # img.update()
# img.preview_ensure() # img.preview_ensure()
# Copy the image with a new name # Copy the image with a new name
# render = bpy.data.images['Render Result'] # render = bpy.data.images['Render Result']
@ -534,8 +568,6 @@ class ASSETLIB_OT_make_custom_preview(Operator):
# image.preview.image_size = preview_size # image.preview.image_size = preview_size
# image.preview.image_pixels_float.foreach_set(pixels) # image.preview.image_pixels_float.foreach_set(pixels)
self.restore() self.restore()
# self.is_running = False # self.is_running = False
@ -544,11 +576,11 @@ class ASSETLIB_OT_make_custom_preview(Operator):
return {"FINISHED"} return {"FINISHED"}
def restore(self): def restore(self):
print('RESTORE') print("RESTORE")
try: try:
bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW') bpy.types.SpaceView3D.draw_handler_remove(self._handle, "WINDOW")
except: except:
print('Failed remove handler') print("Failed remove handler")
pass pass
bpy.data.objects.remove(self.camera) bpy.data.objects.remove(self.camera)
@ -562,7 +594,7 @@ class ASSETLIB_OT_make_custom_preview(Operator):
bg_color = (0.8, 0.1, 0.1, 0.5) bg_color = (0.8, 0.1, 0.1, 0.5)
font_color = (1, 1, 1, 1) font_color = (1, 1, 1, 1)
text = f'Escape: Cancel Enter: Make Preview' text = f"Escape: Cancel Enter: Make Preview"
font_id = 0 font_id = 0
dim = blf.dimensions(font_id, text) dim = blf.dimensions(font_id, text)
@ -581,7 +613,7 @@ class ASSETLIB_OT_make_custom_preview(Operator):
blf.color(font_id, *font_color) # unpack color blf.color(font_id, *font_color) # unpack color
blf.position(font_id, context.region.width / 2 - dim[0] / 2, dim[1] / 2 + 5, 0) blf.position(font_id, context.region.width / 2 - dim[0] / 2, dim[1] / 2 + 5, 0)
blf.size(font_id, 12, dpi) blf.size(font_id, 12, dpi)
blf.draw(font_id, f'Escape: Cancel Enter: Make Preview') blf.draw(font_id, f"Escape: Cancel Enter: Make Preview")
def get_image_name(self): def get_image_name(self):
prefs = get_addon_prefs() prefs = get_addon_prefs()
@ -592,12 +624,12 @@ class ASSETLIB_OT_make_custom_preview(Operator):
if preview_names: if preview_names:
index = int(preview_names[-1][-2:]) + 1 index = int(preview_names[-1][-2:]) + 1
return f'preview_{index:03d}' return f"preview_{index:03d}"
def invoke(self, context, event): def invoke(self, context, event):
prefs = get_addon_prefs() prefs = get_addon_prefs()
cam_data = bpy.data.cameras.new(name='Preview Camera') cam_data = bpy.data.cameras.new(name="Preview Camera")
self.camera = bpy.data.objects.new(name='Preview Camera', object_data=cam_data) self.camera = bpy.data.objects.new(name="Preview Camera", object_data=cam_data)
# view_3d = get_view3d_persp() # view_3d = get_view3d_persp()
@ -605,32 +637,34 @@ class ASSETLIB_OT_make_custom_preview(Operator):
space = context.space_data space = context.space_data
matrix = space.region_3d.view_matrix.inverted() matrix = space.region_3d.view_matrix.inverted()
if space.region_3d.view_perspective == 'CAMERA': if space.region_3d.view_perspective == "CAMERA":
matrix = scn.camera.matrix_world matrix = scn.camera.matrix_world
self.camera.matrix_world = matrix self.camera.matrix_world = matrix
img_name = self.get_image_name() img_name = self.get_image_name()
img_path = Path(bpy.app.tempdir, img_name).with_suffix('.webp') img_path = Path(bpy.app.tempdir, img_name).with_suffix(".webp")
self.attr_changed = attr_set([ self.attr_changed = attr_set(
(space.overlay, 'show_overlays', False), [
(space.region_3d, 'view_perspective', 'CAMERA'), (space.overlay, "show_overlays", False),
(space.region_3d, 'view_camera_offset'), (space.region_3d, "view_perspective", "CAMERA"),
(space.region_3d, 'view_camera_zoom'), (space.region_3d, "view_camera_offset"),
(space, 'lock_camera', True), (space.region_3d, "view_camera_zoom"),
(space, 'show_region_ui', False), (space, "lock_camera", True),
(scn, 'camera', self.camera), (space, "show_region_ui", False),
(scn.render, 'resolution_percentage', 100), (scn, "camera", self.camera),
(scn.render, 'resolution_x', self.image_size), (scn.render, "resolution_percentage", 100),
(scn.render, 'resolution_y', self.image_size), (scn.render, "resolution_x", self.image_size),
(scn.render, 'film_transparent', True), (scn.render, "resolution_y", self.image_size),
(scn.render.image_settings, 'file_format', 'WEBP'), (scn.render, "film_transparent", True),
(scn.render.image_settings, 'color_mode', 'RGBA'), (scn.render.image_settings, "file_format", "WEBP"),
(scn.render.image_settings, "color_mode", "RGBA"),
# (scn.render.image_settings, 'color_depth', '8'), # (scn.render.image_settings, 'color_depth', '8'),
(scn.render, 'use_overwrite', True), (scn.render, "use_overwrite", True),
(scn.render, 'filepath', str(img_path)), (scn.render, "filepath", str(img_path)),
]) ]
)
bpy.ops.view3d.view_center_camera() bpy.ops.view3d.view_center_camera()
space.region_3d.view_camera_zoom -= 6 space.region_3d.view_camera_zoom -= 6
@ -643,14 +677,17 @@ class ASSETLIB_OT_make_custom_preview(Operator):
if self.modal: if self.modal:
prefs.preview_modal = True prefs.preview_modal = True
self.shader_2d = gpu.shader.from_builtin('2D_UNIFORM_COLOR') self.shader_2d = gpu.shader.from_builtin("2D_UNIFORM_COLOR")
self.screen_framing = batch_for_shader( self.screen_framing = batch_for_shader(
self.shader_2d, 'LINE_LOOP', {"pos": [(0,0), (0,h), (w,h), (w,0)]}) self.shader_2d, "LINE_LOOP", {"pos": [(0, 0), (0, h), (w, h), (w, 0)]}
)
self._handle = bpy.types.SpaceView3D.draw_handler_add(self.draw_callback_px, (context,), 'WINDOW', 'POST_PIXEL') self._handle = bpy.types.SpaceView3D.draw_handler_add(
self.draw_callback_px, (context,), "WINDOW", "POST_PIXEL"
)
context.window_manager.modal_handler_add(self) context.window_manager.modal_handler_add(self)
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
else: else:
return self.execute(context) return self.execute(context)
@ -685,10 +722,11 @@ class ASSETLIB_OT_generate_previews(Operator):
preview_blend = self.preview_blend or lib.library_type.preview_blend preview_blend = self.preview_blend or lib.library_type.preview_blend
if not preview_blend or not Path(preview_blend).exists(): if not preview_blend or not Path(preview_blend).exists():
preview_blend = MODULE_DIR / 'common' / 'preview.blend' preview_blend = MODULE_DIR / "common" / "preview.blend"
script_path = Path(bpy.app.tempdir) / 'generate_previews.py' script_path = Path(bpy.app.tempdir) / "generate_previews.py"
script_code = dedent(f""" script_code = dedent(
f"""
import bpy import bpy
prefs = bpy.context.preferences.addons["asset_library"].preferences prefs = bpy.context.preferences.addons["asset_library"].preferences
lib = prefs.env_libraries.add() lib = prefs.env_libraries.add()
@ -696,7 +734,8 @@ class ASSETLIB_OT_generate_previews(Operator):
bpy.ops.wm.open_mainfile(filepath='{preview_blend}', load_ui=True) bpy.ops.wm.open_mainfile(filepath='{preview_blend}', load_ui=True)
lib.library_type.generate_previews(cache='{self.cache}') lib.library_type.generate_previews(cache='{self.cache}')
""") """
)
script_path.write_text(script_code) script_path.write_text(script_code)
@ -707,15 +746,14 @@ class ASSETLIB_OT_generate_previews(Operator):
else: else:
subprocess.Popen(cmd) subprocess.Popen(cmd)
return {"FINISHED"}
return {'FINISHED'}
class ASSETLIB_OT_play_preview(Operator): class ASSETLIB_OT_play_preview(Operator):
bl_idname = "assetlib.play_preview" bl_idname = "assetlib.play_preview"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
bl_label = 'Play Preview' bl_label = "Play Preview"
bl_description = 'Play Preview' bl_description = "Play Preview"
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
@ -745,33 +783,32 @@ class ASSETLIB_OT_play_preview(Operator):
asset_video = lib.library_type.get_video(asset.name, asset_path) asset_video = lib.library_type.get_video(asset.name, asset_path)
if not asset_image and not asset_video: if not asset_image and not asset_video:
self.report({'ERROR'}, f'Preview for {asset.name} not found.') self.report({"ERROR"}, f"Preview for {asset.name} not found.")
return {"CANCELLED"} return {"CANCELLED"}
if asset_video: if asset_video:
self.report({'INFO'}, f'Video found. {asset_video}.') self.report({"INFO"}, f"Video found. {asset_video}.")
if prefs.video_player: if prefs.video_player:
subprocess.Popen([prefs.video_player, asset_video]) subprocess.Popen([prefs.video_player, asset_video])
else: else:
bpy.ops.wm.path_open(filepath=str(asset_video)) bpy.ops.wm.path_open(filepath=str(asset_video))
else: else:
self.report({'INFO'}, f'Image found. {asset_image}.') self.report({"INFO"}, f"Image found. {asset_image}.")
if prefs.image_player: if prefs.image_player:
subprocess.Popen([prefs.image_player, asset_image]) subprocess.Popen([prefs.image_player, asset_image])
else: else:
bpy.ops.wm.path_open(filepath=str(asset_image)) bpy.ops.wm.path_open(filepath=str(asset_image))
return {"FINISHED"} return {"FINISHED"}
class ASSETLIB_OT_synchronize(Operator): class ASSETLIB_OT_synchronize(Operator):
bl_idname = "assetlib.synchronize" bl_idname = "assetlib.synchronize"
bl_options = {"REGISTER", "UNDO"} bl_options = {"REGISTER", "UNDO"}
bl_label = 'Synchronize' bl_label = "Synchronize"
bl_description = 'Synchronize Action Lib to Local Directory' bl_description = "Synchronize Action Lib to Local Directory"
clean: BoolProperty(default=False) clean: BoolProperty(default=False)
only_new: BoolProperty(default=False) only_new: BoolProperty(default=False)
@ -781,11 +818,11 @@ class ASSETLIB_OT_synchronize(Operator):
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
print('Not yet Implemented, have to be replace by Bundle instead') print("Not yet Implemented, have to be replace by Bundle instead")
return {'FINISHED'} return {"FINISHED"}
prefs = get_addon_prefs() prefs = get_addon_prefs()
print('Synchronize') print("Synchronize")
if self.all: if self.all:
libs = prefs.libraries libs = prefs.libraries
else: else:
@ -794,7 +831,7 @@ class ASSETLIB_OT_synchronize(Operator):
for lib in libs: for lib in libs:
if self.clean and Path(lib.path_local).exists(): if self.clean and Path(lib.path_local).exists():
pass pass
print('To check first') print("To check first")
# shutil.rmtree(path_local) # shutil.rmtree(path_local)
if not lib.path_local: if not lib.path_local:
@ -804,10 +841,11 @@ class ASSETLIB_OT_synchronize(Operator):
src=lib.path, src=lib.path,
dst=lib.path_local, dst=lib.path_local,
only_new=self.only_new, only_new=self.only_new,
only_recent=self.only_recent only_recent=self.only_recent,
) )
return {'FINISHED'} return {"FINISHED"}
classes = ( classes = (
ASSETLIB_OT_play_preview, ASSETLIB_OT_play_preview,
@ -823,15 +861,17 @@ classes = (
ASSETLIB_OT_edit_data, ASSETLIB_OT_edit_data,
# ASSETLIB_OT_conform_library, # ASSETLIB_OT_conform_library,
ASSETLIB_OT_reload_addon, ASSETLIB_OT_reload_addon,
ASSETLIB_OT_make_custom_preview ASSETLIB_OT_make_custom_preview,
) )
def register(): def register():
# bpy.types.UserAssetLibrary.is_env = False # bpy.types.UserAssetLibrary.is_env = False
for cls in classes: for cls in classes:
bpy.utils.register_class(cls) bpy.utils.register_class(cls)
def unregister(): def unregister():
for cls in reversed(classes): for cls in reversed(classes):
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(cls)

View File

@ -1,14 +1,14 @@
from asset_library.pose import operators
from asset_library.pose import ( if "bpy" in locals():
operators)
if 'bpy' in locals():
import importlib import importlib
importlib.reload(operators) importlib.reload(operators)
def register(): def register():
operators.register() operators.register()
def unregister(): def unregister():
operators.unregister() operators.unregister()

View File

@ -39,7 +39,7 @@ def convert_old_poselib(old_poselib: Action) -> Collection[Action]:
# appropriate frame in the scene (to set up things like the background # appropriate frame in the scene (to set up things like the background
# colour), but the old-style poselib doesn't contain such information. All # colour), but the old-style poselib doesn't contain such information. All
# we can do is just render on the current frame. # we can do is just render on the current frame.
bpy.ops.asset.mark({'selected_ids': pose_assets}) bpy.ops.asset.mark({"selected_ids": pose_assets})
return pose_assets return pose_assets

View File

@ -22,7 +22,13 @@ import subprocess
import uuid import uuid
import time import time
from bpy.props import BoolProperty, CollectionProperty, EnumProperty, PointerProperty, StringProperty from bpy.props import (
BoolProperty,
CollectionProperty,
EnumProperty,
PointerProperty,
StringProperty,
)
from bpy.types import ( from bpy.types import (
Action, Action,
Context, Context,
@ -40,11 +46,7 @@ from asset_library.action.functions import (
get_keyframes, get_keyframes,
) )
from asset_library.common.bl_utils import ( from asset_library.common.bl_utils import get_view3d_persp, load_assets_from, split_path
get_view3d_persp,
load_assets_from,
split_path
)
class POSELIB_OT_create_pose_asset(Operator): class POSELIB_OT_create_pose_asset(Operator):
@ -59,19 +61,22 @@ class POSELIB_OT_create_pose_asset(Operator):
pose_name: StringProperty(name="Pose Name") # type: ignore pose_name: StringProperty(name="Pose Name") # type: ignore
activate_new_action: BoolProperty(name="Activate New Action", default=True) # type: ignore activate_new_action: BoolProperty(name="Activate New Action", default=True) # type: ignore
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
# Make sure that if there is an asset browser open, the artist can see the newly created pose asset. # Make sure that if there is an asset browser open, the artist can see the newly created pose asset.
asset_browse_area: Optional[bpy.types.Area] = asset_browser.area_from_context(context) asset_browse_area: Optional[bpy.types.Area] = asset_browser.area_from_context(
context
)
if not asset_browse_area: if not asset_browse_area:
# No asset browser is visible, so there also aren't any expectations # No asset browser is visible, so there also aren't any expectations
# that this asset will be visible. # that this asset will be visible.
return True return True
asset_space_params = asset_browser.params(asset_browse_area) asset_space_params = asset_browser.params(asset_browse_area)
if asset_space_params.asset_library_ref != 'LOCAL': if asset_space_params.asset_library_ref != "LOCAL":
cls.poll_message_set("Asset Browser must be set to the Current File library") cls.poll_message_set(
"Asset Browser must be set to the Current File library"
)
return False return False
return True return True
@ -85,28 +90,28 @@ class POSELIB_OT_create_pose_asset(Operator):
if pose_name: if pose_name:
prefix = True prefix = True
asset_name = Path(bpy.data.filepath).stem.split('_')[0] asset_name = Path(bpy.data.filepath).stem.split("_")[0]
action_asset_name = re.search(f'^{asset_name}.', pose_name) action_asset_name = re.search(f"^{asset_name}.", pose_name)
if action_asset_name: if action_asset_name:
pose_name = pose_name.replace(action_asset_name.group(0), '') pose_name = pose_name.replace(action_asset_name.group(0), "")
side = re.search('_\w$', pose_name) side = re.search("_\w$", pose_name)
if side: if side:
pose_name = pose_name.replace(side.group(0), '') pose_name = pose_name.replace(side.group(0), "")
if 'hands' in context.object.animation_data.action.name.lower(): if "hands" in context.object.animation_data.action.name.lower():
pose_name = f'hand_{pose_name}' pose_name = f"hand_{pose_name}"
if pose_name.startswith('lips_'): if pose_name.startswith("lips_"):
pose_name.replace('lips_', '') pose_name.replace("lips_", "")
split = pose_name.split('_') split = pose_name.split("_")
pose_name = '-'.join([s for s in split if s.isupper()]) pose_name = "-".join([s for s in split if s.isupper()])
pose_name = f'{pose_name}_{split[-1]}' pose_name = f"{pose_name}_{split[-1]}"
prefix = False prefix = False
if prefix and not pose_name.startswith(asset_name): if prefix and not pose_name.startswith(asset_name):
pose_name = f'{asset_name}_{pose_name}' pose_name = f"{asset_name}_{pose_name}"
else: else:
pose_name = self.pose_name or context.object.name pose_name = self.pose_name or context.object.name
@ -126,7 +131,6 @@ class POSELIB_OT_create_pose_asset(Operator):
if context.scene.camera: if context.scene.camera:
data_dict.update(dict(camera=context.scene.camera.name)) data_dict.update(dict(camera=context.scene.camera.name))
for k, v in data_dict.items(): for k, v in data_dict.items():
data[k] = v data[k] = v
### ###
@ -134,7 +138,7 @@ class POSELIB_OT_create_pose_asset(Operator):
if self.activate_new_action: if self.activate_new_action:
self._set_active_action(context, asset) self._set_active_action(context, asset)
self._activate_asset_in_browser(context, asset) self._activate_asset_in_browser(context, asset)
return {'FINISHED'} return {"FINISHED"}
def _set_active_action(self, context: Context, asset: Action) -> None: def _set_active_action(self, context: Context, asset: Action) -> None:
self._prevent_action_loss(context.object) self._prevent_action_loss(context.object)
@ -149,7 +153,9 @@ class POSELIB_OT_create_pose_asset(Operator):
This makes it possible to immediately check & edit the created pose asset. This makes it possible to immediately check & edit the created pose asset.
""" """
asset_browse_area: Optional[bpy.types.Area] = asset_browser.area_from_context(context) asset_browse_area: Optional[bpy.types.Area] = asset_browser.area_from_context(
context
)
if not asset_browse_area: if not asset_browse_area:
return return
@ -181,7 +187,9 @@ class POSELIB_OT_create_pose_asset(Operator):
return return
action.use_fake_user = True action.use_fake_user = True
self.report({'WARNING'}, "Action %s marked Fake User to prevent loss" % action.name) self.report(
{"WARNING"}, "Action %s marked Fake User to prevent loss" % action.name
)
class POSELIB_OT_restore_previous_action(Operator): class POSELIB_OT_restore_previous_action(Operator):
@ -215,17 +223,17 @@ class POSELIB_OT_restore_previous_action(Operator):
self._timer = wm.event_timer_add(0.001, window=context.window) self._timer = wm.event_timer_add(0.001, window=context.window)
wm.modal_handler_add(self) wm.modal_handler_add(self)
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
def modal(self, context, event): def modal(self, context, event):
if event.type != 'TIMER': if event.type != "TIMER":
return {'RUNNING_MODAL'} return {"RUNNING_MODAL"}
wm = context.window_manager wm = context.window_manager
wm.event_timer_remove(self._timer) wm.event_timer_remove(self._timer)
context.object.pose.apply_pose_from_action(self.pose_action) context.object.pose.apply_pose_from_action(self.pose_action)
return {'FINISHED'} return {"FINISHED"}
class ASSET_OT_assign_action(Operator): class ASSET_OT_assign_action(Operator):
@ -257,7 +265,9 @@ class ASSET_OT_assign_action(Operator):
class POSELIB_OT_copy_as_asset(Operator): class POSELIB_OT_copy_as_asset(Operator):
bl_idname = "poselib.copy_as_asset" bl_idname = "poselib.copy_as_asset"
bl_label = "Copy Pose As Asset" bl_label = "Copy Pose As Asset"
bl_description = "Create a new pose asset on the clipboard, to be pasted into an Asset Browser" bl_description = (
"Create a new pose asset on the clipboard, to be pasted into an Asset Browser"
)
bl_options = {"REGISTER"} bl_options = {"REGISTER"}
CLIPBOARD_ASSET_MARKER = "ASSET-BLEND=" CLIPBOARD_ASSET_MARKER = "ASSET-BLEND="
@ -289,7 +299,10 @@ class POSELIB_OT_copy_as_asset(Operator):
filepath, filepath,
) )
asset_browser.tag_redraw(context.screen) asset_browser.tag_redraw(context.screen)
self.report({"INFO"}, "Pose Asset copied, use Paste As New Asset in any Asset Browser to paste") self.report(
{"INFO"},
"Pose Asset copied, use Paste As New Asset in any Asset Browser to paste",
)
# The asset has been saved to disk, so to clean up it has to loose its asset & fake user status. # The asset has been saved to disk, so to clean up it has to loose its asset & fake user status.
asset.asset_clear() asset.asset_clear()
@ -300,7 +313,10 @@ class POSELIB_OT_copy_as_asset(Operator):
if asset.users > 0: if asset.users > 0:
# This should never happen, and indicates a bug in the code. Having a warning about it is nice, # This should never happen, and indicates a bug in the code. Having a warning about it is nice,
# but it shouldn't stand in the way of actually cleaning up the meant-to-be-temporary datablock. # but it shouldn't stand in the way of actually cleaning up the meant-to-be-temporary datablock.
self.report({"WARNING"}, "Unexpected non-zero user count for the asset, please report this as a bug") self.report(
{"WARNING"},
"Unexpected non-zero user count for the asset, please report this as a bug",
)
bpy.data.actions.remove(asset) bpy.data.actions.remove(asset)
return {"FINISHED"} return {"FINISHED"}
@ -331,8 +347,10 @@ class POSELIB_OT_paste_asset(Operator):
return False return False
asset_lib_ref = context.space_data.params.asset_library_ref asset_lib_ref = context.space_data.params.asset_library_ref
if asset_lib_ref != 'LOCAL': if asset_lib_ref != "LOCAL":
cls.poll_message_set("Asset Browser must be set to the Current File library") cls.poll_message_set(
"Asset Browser must be set to the Current File library"
)
return False return False
# Delay checking the clipboard as much as possible, as it's CPU-heavier than the other checks. # Delay checking the clipboard as much as possible, as it's CPU-heavier than the other checks.
@ -348,7 +366,6 @@ class POSELIB_OT_paste_asset(Operator):
return True return True
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
clipboard = context.window_manager.clipboard clipboard = context.window_manager.clipboard
marker_len = len(POSELIB_OT_copy_as_asset.CLIPBOARD_ASSET_MARKER) marker_len = len(POSELIB_OT_copy_as_asset.CLIPBOARD_ASSET_MARKER)
@ -385,12 +402,12 @@ class POSELIB_OT_pose_asset_select_bones(Operator):
# bl_property = "selected_side" # bl_property = "selected_side"
selected_side: EnumProperty( selected_side: EnumProperty(
name='Selected Side', name="Selected Side",
items=( items=(
('CURRENT', "Current", ""), ("CURRENT", "Current", ""),
('FLIPPED', "Flipped", ""), ("FLIPPED", "Flipped", ""),
('BOTH', "Both", ""), ("BOTH", "Both", ""),
) ),
) )
@classmethod @classmethod
@ -402,7 +419,7 @@ class POSELIB_OT_pose_asset_select_bones(Operator):
and context.asset_file_handle and context.asset_file_handle
): ):
return False return False
return context.asset_file_handle.id_type == 'ACTION' return context.asset_file_handle.id_type == "ACTION"
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
asset: FileSelectEntry = context.asset_file_handle asset: FileSelectEntry = context.asset_file_handle
@ -417,7 +434,9 @@ class POSELIB_OT_pose_asset_select_bones(Operator):
def _load_and_use_pose(self, context: Context) -> Set[str]: def _load_and_use_pose(self, context: Context) -> Set[str]:
asset_library_ref = context.asset_library_ref asset_library_ref = context.asset_library_ref
asset = context.asset_file_handle asset = context.asset_file_handle
asset_lib_path = bpy.types.AssetHandle.get_full_library_path(asset, asset_library_ref) asset_lib_path = bpy.types.AssetHandle.get_full_library_path(
asset, asset_library_ref
)
if not asset_lib_path: if not asset_lib_path:
self.report( # type: ignore self.report( # type: ignore
@ -426,7 +445,7 @@ class POSELIB_OT_pose_asset_select_bones(Operator):
f"Selected asset {asset.name} could not be located inside the asset library", f"Selected asset {asset.name} could not be located inside the asset library",
) )
return {"CANCELLED"} return {"CANCELLED"}
if asset.id_type != 'ACTION': if asset.id_type != "ACTION":
self.report( # type: ignore self.report( # type: ignore
{"ERROR"}, {"ERROR"},
f"Selected asset {asset.name} is not an Action", f"Selected asset {asset.name} is not an Action",
@ -443,9 +462,12 @@ class POSELIB_OT_pose_asset_select_bones(Operator):
def use_pose(self, context: Context, pose_asset: Action) -> Set[str]: def use_pose(self, context: Context, pose_asset: Action) -> Set[str]:
arm_object: Object = context.object arm_object: Object = context.object
# pose_usage.select_bones(arm_object, pose_asset, select=self.select, flipped=self.flipped) # pose_usage.select_bones(arm_object, pose_asset, select=self.select, flipped=self.flipped)
pose_usage.select_bones(arm_object, pose_asset, selected_side=self.selected_side) pose_usage.select_bones(
arm_object, pose_asset, selected_side=self.selected_side
)
return {"FINISHED"} return {"FINISHED"}
# This operator takes the Window Manager's `actionlib_flipped` property, and # This operator takes the Window Manager's `actionlib_flipped` property, and
# passes it to the `POSELIB_OT_blend_pose_asset` operator. This makes it # passes it to the `POSELIB_OT_blend_pose_asset` operator. This makes it
# possible to bind a key to the operator and still have it respect the global # possible to bind a key to the operator and still have it respect the global
@ -478,10 +500,14 @@ class POSELIB_OT_blend_pose_asset_for_keymap(Operator):
""" """
def invoke(self, context: Context, event: Event) -> Set[str]: def invoke(self, context: Context, event: Event) -> Set[str]:
return bpy.ops.poselib.blend_pose_asset(context.copy(), 'INVOKE_DEFAULT', flipped=self.flipped) return bpy.ops.poselib.blend_pose_asset(
context.copy(), "INVOKE_DEFAULT", flipped=self.flipped
)
def execute(self, context: Context) -> Set[str]: def execute(self, context: Context) -> Set[str]:
return bpy.ops.poselib.blend_pose_asset(context.copy(), 'EXEC_DEFAULT', flipped=self.flipped) return bpy.ops.poselib.blend_pose_asset(
context.copy(), "EXEC_DEFAULT", flipped=self.flipped
)
# This operator takes the Window Manager's `actionlib_flipped` property, and # This operator takes the Window Manager's `actionlib_flipped` property, and
@ -489,6 +515,7 @@ class POSELIB_OT_blend_pose_asset_for_keymap(Operator):
# possible to bind a key to the operator and still have it respect the global # possible to bind a key to the operator and still have it respect the global
# "Flip Pose" checkbox. # "Flip Pose" checkbox.
class POSELIB_OT_apply_pose_asset_for_keymap(Operator): class POSELIB_OT_apply_pose_asset_for_keymap(Operator):
bl_idname = "poselib.apply_pose_asset_for_keymap" bl_idname = "poselib.apply_pose_asset_for_keymap"
bl_options = {"REGISTER", "UNDO", "INTERNAL"} bl_options = {"REGISTER", "UNDO", "INTERNAL"}
@ -496,7 +523,7 @@ class POSELIB_OT_apply_pose_asset_for_keymap(Operator):
_rna = bpy.ops.poselib.apply_pose_asset.get_rna_type() _rna = bpy.ops.poselib.apply_pose_asset.get_rna_type()
bl_label = _rna.name bl_label = _rna.name
# bl_description = _rna.description # bl_description = _rna.description
bl_description = 'Apply Pose to Bones' bl_description = "Apply Pose to Bones"
del _rna del _rna
flipped: BoolProperty(name="Flipped", default=False) # type: ignore flipped: BoolProperty(name="Flipped", default=False) # type: ignore
@ -514,19 +541,34 @@ class POSELIB_OT_apply_pose_asset_for_keymap(Operator):
store_bones = {} store_bones = {}
bones = [ bones = [
'blendshape-eyes', 'blendshape-eye.L', 'blendshape-eye.R', "blendshape-eyes",
'blendshape-corner-mouth', 'blendshape-corner-mouth.L', "blendshape-eye.L",
'blendshape-corner-down-mouth.L', 'blendshape-corner-up-mouth.L', "blendshape-eye.R",
'blendshape-corner-mouth-add.L','blendshape-corner-mouth.R', "blendshape-corner-mouth",
'blendshape-corner-down-mouth.R', 'blendshape-corner-up-mouth.R', "blendshape-corner-mouth.L",
'blendshape-corner-mouth-add.R', 'blendshape-center-up-mouth', "blendshape-corner-down-mouth.L",
'blendshape-center-down-mouth', "blendshape-corner-up-mouth.L",
'hat1.R', 'hat2.R', 'hat3.R', 'hat1.L', 'hat2.L', 'hat3.L', "blendshape-corner-mouth-add.L",
"blendshape-corner-mouth.R",
"blendshape-corner-down-mouth.R",
"blendshape-corner-up-mouth.R",
"blendshape-corner-mouth-add.R",
"blendshape-center-up-mouth",
"blendshape-center-down-mouth",
"hat1.R",
"hat2.R",
"hat3.R",
"hat1.L",
"hat2.L",
"hat3.L",
] ]
attributes = [ attributes = [
'location', 'rotation_quaternion', "location",
'rotation_euler', 'rotation_axis_angle', 'scale' "rotation_quaternion",
"rotation_euler",
"rotation_axis_angle",
"scale",
] ]
if action: if action:
@ -543,30 +585,40 @@ class POSELIB_OT_apply_pose_asset_for_keymap(Operator):
if not prop_name in store_bones[bone_name].keys(): if not prop_name in store_bones[bone_name].keys():
store_bones[bone_name][prop_name] = [] store_bones[bone_name][prop_name] = []
val = getattr(context.object.pose.bones[bone_name], prop_name) val = getattr(
context.object.pose.bones[bone_name], prop_name
)
store_bones[bone_name][prop_name].append(fc.evaluate(context.scene.frame_current)) store_bones[bone_name][prop_name].append(
fc.evaluate(context.scene.frame_current)
)
bpy.ops.poselib.apply_pose_asset(context.copy(), 'EXEC_DEFAULT', flipped=True) bpy.ops.poselib.apply_pose_asset(
context.copy(), "EXEC_DEFAULT", flipped=True
)
for bone, v in store_bones.items(): for bone, v in store_bones.items():
for attr, attr_val in v.items(): for attr, attr_val in v.items():
flipped_vector = 1 flipped_vector = 1
### TODO FAIRE ÇA PROPREMENT AVEC UNE COMPREHENSION LIST OU AUTRE ### TODO FAIRE ÇA PROPREMENT AVEC UNE COMPREHENSION LIST OU AUTRE
if re.search(r'\.[RL]$', bone): if re.search(r"\.[RL]$", bone):
flipped_bone = pose_usage.flip_side_name(bone) flipped_bone = pose_usage.flip_side_name(bone)
if attr == 'location': if attr == "location":
flipped_vector = Vector((-1, 1, 1)) flipped_vector = Vector((-1, 1, 1))
# print('-----', store_bones.get(flipped_bone)[attr]) # print('-----', store_bones.get(flipped_bone)[attr])
attr_val = Vector(store_bones.get(flipped_bone)[attr]) * flipped_vector attr_val = (
Vector(store_bones.get(flipped_bone)[attr]) * flipped_vector
)
setattr(context.object.pose.bones[bone], attr, attr_val) setattr(context.object.pose.bones[bone], attr, attr_val)
return {'FINISHED'} return {"FINISHED"}
else: else:
return bpy.ops.poselib.apply_pose_asset(context.copy(), 'EXEC_DEFAULT', flipped=False) return bpy.ops.poselib.apply_pose_asset(
context.copy(), "EXEC_DEFAULT", flipped=False
)
class POSELIB_OT_convert_old_poselib(Operator): class POSELIB_OT_convert_old_poselib(Operator):
@ -577,12 +629,18 @@ class POSELIB_OT_convert_old_poselib(Operator):
@classmethod @classmethod
def poll(cls, context: Context) -> bool: def poll(cls, context: Context) -> bool:
action = context.object and context.object.animation_data and context.object.animation_data.action action = (
context.object
and context.object.animation_data
and context.object.animation_data.action
)
if not action: if not action:
cls.poll_message_set("Active object has no Action") cls.poll_message_set("Active object has no Action")
return False return False
if not action.pose_markers: if not action.pose_markers:
cls.poll_message_set("Action %r is not a old-style pose library" % action.name) cls.poll_message_set(
"Action %r is not a old-style pose library" % action.name
)
return False return False
return True return True
@ -593,12 +651,11 @@ class POSELIB_OT_convert_old_poselib(Operator):
new_actions = conversion.convert_old_poselib(old_poselib) new_actions = conversion.convert_old_poselib(old_poselib)
if not new_actions: if not new_actions:
self.report({'ERROR'}, "Unable to convert to pose assets") self.report({"ERROR"}, "Unable to convert to pose assets")
return {'CANCELLED'} return {"CANCELLED"}
self.report({'INFO'}, "Converted %d poses to pose assets" % len(new_actions))
return {'FINISHED'}
self.report({"INFO"}, "Converted %d poses to pose assets" % len(new_actions))
return {"FINISHED"}
classes = ( classes = (
@ -609,7 +666,7 @@ classes = (
POSELIB_OT_create_pose_asset, POSELIB_OT_create_pose_asset,
POSELIB_OT_paste_asset, POSELIB_OT_paste_asset,
POSELIB_OT_pose_asset_select_bones, POSELIB_OT_pose_asset_select_bones,
POSELIB_OT_restore_previous_action POSELIB_OT_restore_previous_action,
) )
register, unregister = bpy.utils.register_classes_factory(classes) register, unregister = bpy.utils.register_classes_factory(classes)

View File

@ -129,7 +129,9 @@ class PoseActionCreator:
continue continue
try: try:
value = self._current_value(armature_ob, fcurve.data_path, fcurve.array_index) value = self._current_value(
armature_ob, fcurve.data_path, fcurve.array_index
)
except UnresolvablePathError: except UnresolvablePathError:
# A once-animated property no longer exists. # A once-animated property no longer exists.
continue continue
@ -197,7 +199,9 @@ class PoseActionCreator:
fcurve: Optional[FCurve] = dst_action.fcurves.find(rna_path, index=array_index) fcurve: Optional[FCurve] = dst_action.fcurves.find(rna_path, index=array_index)
if fcurve is None: if fcurve is None:
fcurve = dst_action.fcurves.new(rna_path, index=array_index, action_group=bone_name) fcurve = dst_action.fcurves.new(
rna_path, index=array_index, action_group=bone_name
)
fcurve.keyframe_points.insert(self.params.src_frame_nr, value=value) fcurve.keyframe_points.insert(self.params.src_frame_nr, value=value)
fcurve.update() fcurve.update()
@ -296,8 +300,11 @@ def create_pose_asset(
pose_action.asset_generate_preview() pose_action.asset_generate_preview()
return pose_action return pose_action
# def create_pose_asset_from_context(context: Context, new_asset_name: str, selection=True) -> Optional[Action]: # def create_pose_asset_from_context(context: Context, new_asset_name: str, selection=True) -> Optional[Action]:
def create_pose_asset_from_context(context: Context, new_asset_name: str) -> Optional[Action]: def create_pose_asset_from_context(
context: Context, new_asset_name: str
) -> Optional[Action]:
"""Create Action asset from active object & selected bones.""" """Create Action asset from active object & selected bones."""
bones = context.selected_pose_bones_from_active_object bones = context.selected_pose_bones_from_active_object
@ -369,7 +376,10 @@ def copy_keyframe(dst_fcurve: FCurve, src_keyframe: Keyframe) -> Keyframe:
"""Copy a keyframe from one FCurve to the other.""" """Copy a keyframe from one FCurve to the other."""
dst_keyframe = dst_fcurve.keyframe_points.insert( dst_keyframe = dst_fcurve.keyframe_points.insert(
src_keyframe.co.x, src_keyframe.co.y, options={'FAST'}, keyframe_type=src_keyframe.type src_keyframe.co.x,
src_keyframe.co.y,
options={"FAST"},
keyframe_type=src_keyframe.type,
) )
for propname in { for propname in {
@ -412,7 +422,9 @@ def find_keyframe(fcurve: FCurve, frame: float) -> Optional[Keyframe]:
return None return None
def assign_from_asset_browser(asset: Action, asset_browser_area: bpy.types.Area) -> None: def assign_from_asset_browser(
asset: Action, asset_browser_area: bpy.types.Area
) -> None:
"""Assign some things from the asset browser to the asset. """Assign some things from the asset browser to the asset.
This sets the current catalog ID, and in the future could include tags This sets the current catalog ID, and in the future could include tags

View File

@ -35,15 +35,14 @@ def select_bones(arm_object: Object, action: Action, *, selected_side, toggle=Tr
continue continue
seen_bone_names.add(bone_name) seen_bone_names.add(bone_name)
if selected_side == 'FLIPPED': if selected_side == "FLIPPED":
bones_to_select.add(bone_name_flip) bones_to_select.add(bone_name_flip)
elif selected_side == 'BOTH': elif selected_side == "BOTH":
bones_to_select.add(bone_name_flip) bones_to_select.add(bone_name_flip)
bones_to_select.add(bone_name) bones_to_select.add(bone_name)
elif selected_side == 'CURRENT': elif selected_side == "CURRENT":
bones_to_select.add(bone_name) bones_to_select.add(bone_name)
for bone in bones_to_select: for bone in bones_to_select:
pose_bone = pose.bones.get(bone) pose_bone = pose.bones.get(bone)
if pose_bone: if pose_bone:
@ -174,7 +173,7 @@ def flip_side_name(to_flip: str) -> str:
return prefix + replace + suffix + number return prefix + replace + suffix + number
if __name__ == '__main__': if __name__ == "__main__":
import doctest import doctest
print(f"Test result: {doctest.testmod()}") print(f"Test result: {doctest.testmod()}")

View File

@ -1,19 +1,31 @@
import bpy import bpy
import os import os
from os.path import abspath, join from os.path import abspath, join
from bpy.types import (AddonPreferences, PointerProperty, PropertyGroup) from bpy.types import AddonPreferences, PointerProperty, PropertyGroup
from bpy.props import (BoolProperty, StringProperty, CollectionProperty, from bpy.props import (
EnumProperty, IntProperty) BoolProperty,
StringProperty,
CollectionProperty,
EnumProperty,
IntProperty,
)
from asset_library.constants import (DATA_TYPES, DATA_TYPE_ITEMS, from asset_library.constants import (
ICONS, RESOURCES_DIR, LIBRARY_TYPE_DIR, LIBRARY_TYPES, ADAPTERS) DATA_TYPES,
DATA_TYPE_ITEMS,
ICONS,
RESOURCES_DIR,
LIBRARY_TYPE_DIR,
LIBRARY_TYPES,
ADAPTERS,
)
from asset_library.common.file_utils import import_module_from_path, norm_str from asset_library.common.file_utils import import_module_from_path, norm_str
from asset_library.common.bl_utils import get_addon_prefs from asset_library.common.bl_utils import get_addon_prefs
from asset_library.common.library_cache import LibraryCache from asset_library.common.library_cache import LibraryCache
from asset_library.common.catalog import Catalog from asset_library.common.catalog import Catalog
# from asset_library.common.functions import get_catalog_path # from asset_library.common.functions import get_catalog_path
from pathlib import Path from pathlib import Path
@ -22,19 +34,20 @@ import inspect
def update_library_config(self, context): def update_library_config(self, context):
print('update_library_config not yet implemented') print("update_library_config not yet implemented")
def update_library_path(self, context): def update_library_path(self, context):
prefs = get_addon_prefs() prefs = get_addon_prefs()
self['bundle_directory'] = str(self.library_path) self["bundle_directory"] = str(self.library_path)
if not self.custom_bundle_name: if not self.custom_bundle_name:
self['custom_bundle_name'] = self.name self["custom_bundle_name"] = self.name
if not self.custom_bundle_directory: if not self.custom_bundle_directory:
custom_bundle_dir = Path(prefs.bundle_directory, self.library_name).resolve() custom_bundle_dir = Path(prefs.bundle_directory, self.library_name).resolve()
self['custom_bundle_directory'] = str(custom_bundle_dir) self["custom_bundle_directory"] = str(custom_bundle_dir)
# if self.custom_bundle_directory: # if self.custom_bundle_directory:
# self['custom_bundle_directory'] = abspath(bpy.path.abspath(self.custom_bundle_directory)) # self['custom_bundle_directory'] = abspath(bpy.path.abspath(self.custom_bundle_directory))
@ -44,6 +57,7 @@ def update_library_path(self, context):
self.set_library_path() self.set_library_path()
def update_all_library_path(self, context): def update_all_library_path(self, context):
# print('update_all_assetlib_paths') # print('update_all_assetlib_paths')
@ -56,77 +70,97 @@ def update_all_library_path(self, context):
update_library_path(lib, context) update_library_path(lib, context)
# lib.set_library_path() # lib.set_library_path()
def get_library_type_items(self, context): def get_library_type_items(self, context):
# prefs = get_addon_prefs() # prefs = get_addon_prefs()
items = [('NONE', 'None', '', 0)] items = [("NONE", "None", "", 0)]
items += [(norm_str(a.name, format=str.upper), a.name, "", i+1) for i, a in enumerate(LIBRARY_TYPES)] items += [
(norm_str(a.name, format=str.upper), a.name, "", i + 1)
for i, a in enumerate(LIBRARY_TYPES)
]
return items return items
def get_adapters_items(self, context): def get_adapters_items(self, context):
# prefs = get_addon_prefs() # prefs = get_addon_prefs()
items = [('NONE', 'None', '', 0)] items = [("NONE", "None", "", 0)]
items += [(norm_str(a.name, format=str.upper), a.name, "", i+1) for i, a in enumerate(ADAPTERS)] items += [
(norm_str(a.name, format=str.upper), a.name, "", i + 1)
for i, a in enumerate(ADAPTERS)
]
return items return items
def get_library_items(self, context): def get_library_items(self, context):
prefs = get_addon_prefs() prefs = get_addon_prefs()
items = [('NONE', 'None', '', 0)] items = [("NONE", "None", "", 0)]
items += [(l.name, l.name, "", i+1) for i, l in enumerate(prefs.libraries) if l != self] items += [
(l.name, l.name, "", i + 1) for i, l in enumerate(prefs.libraries) if l != self
]
return items return items
def get_store_library_items(self, context): def get_store_library_items(self, context):
# prefs = get_addon_prefs() # prefs = get_addon_prefs()
# libraries = [l for l in prefs.libraries if l.merge_library == self.name] # libraries = [l for l in prefs.libraries if l.merge_library == self.name]
return [(l.name, l.name, "", i) for i, l in enumerate([self] + self.merge_libraries)] return [
(l.name, l.name, "", i) for i, l in enumerate([self] + self.merge_libraries)
]
class LibraryTypes(PropertyGroup): class LibraryTypes(PropertyGroup):
def __iter__(self): def __iter__(self):
return (getattr(self, p) for p in self.bl_rna.properties.keys() if p not in ('rna_type', 'name')) return (
getattr(self, p)
for p in self.bl_rna.properties.keys()
if p not in ("rna_type", "name")
)
class Adapters(PropertyGroup): class Adapters(PropertyGroup):
def __iter__(self): def __iter__(self):
return (getattr(self, p) for p in self.bl_rna.properties.keys() if p not in ('rna_type', 'name')) return (
getattr(self, p)
for p in self.bl_rna.properties.keys()
if p not in ("rna_type", "name")
)
class AssetLibrary(PropertyGroup): class AssetLibrary(PropertyGroup):
name : StringProperty(name='Name', default='Action Library', update=update_library_path) name: StringProperty(
name="Name", default="Action Library", update=update_library_path
)
id: StringProperty() id: StringProperty()
auto_bundle : BoolProperty(name='Auto Bundle', default=False) auto_bundle: BoolProperty(name="Auto Bundle", default=False)
expand : BoolProperty(name='Expand', default=False) expand: BoolProperty(name="Expand", default=False)
use : BoolProperty(name='Use', default=True, update=update_library_path) use: BoolProperty(name="Use", default=True, update=update_library_path)
data_type : EnumProperty(name='Type', items=DATA_TYPE_ITEMS, default='COLLECTION') data_type: EnumProperty(name="Type", items=DATA_TYPE_ITEMS, default="COLLECTION")
# template_image : StringProperty(default='', description='../{name}_image.png') # template_image : StringProperty(default='', description='../{name}_image.png')
# template_video : StringProperty(default='', description='../{name}_video.mov') # template_video : StringProperty(default='', description='../{name}_video.mov')
# template_info : StringProperty(default='', description='../{name}_asset_info.json') # template_info : StringProperty(default='', description='../{name}_asset_info.json')
bundle_directory: StringProperty( bundle_directory: StringProperty(
name="Bundle Directory", name="Bundle Directory", subtype="DIR_PATH", default=""
subtype='DIR_PATH',
default=''
) )
use_custom_bundle_directory: BoolProperty(default=False, update=update_library_path) use_custom_bundle_directory: BoolProperty(default=False, update=update_library_path)
custom_bundle_directory: StringProperty( custom_bundle_directory: StringProperty(
name="Bundle Directory", name="Bundle Directory",
subtype='DIR_PATH', subtype="DIR_PATH",
default='', default="",
update=update_library_path update=update_library_path,
) )
# use_merge : BoolProperty(default=False, update=update_library_path) # use_merge : BoolProperty(default=False, update=update_library_path)
use_custom_bundle_name: BoolProperty(default=False, update=update_library_path) use_custom_bundle_name: BoolProperty(default=False, update=update_library_path)
custom_bundle_name : StringProperty(name='Merge Name', update=update_library_path) custom_bundle_name: StringProperty(name="Merge Name", update=update_library_path)
# merge_library : EnumProperty(name='Merge Library', items=get_library_items, update=update_library_path) # merge_library : EnumProperty(name='Merge Library', items=get_library_items, update=update_library_path)
# merge_name : StringProperty(name='Merge Name', update=update_library_path) # merge_name : StringProperty(name='Merge Name', update=update_library_path)
@ -134,8 +168,8 @@ class AssetLibrary(PropertyGroup):
store_library: EnumProperty(items=get_store_library_items, name="Library") store_library: EnumProperty(items=get_store_library_items, name="Library")
template: StringProperty() template: StringProperty()
expand_extra : BoolProperty(name='Expand', default=False) expand_extra: BoolProperty(name="Expand", default=False)
blend_depth : IntProperty(name='Blend Depth', default=1) blend_depth: IntProperty(name="Blend Depth", default=1)
# source_directory : StringProperty( # source_directory : StringProperty(
# name="Path", # name="Path",
@ -144,7 +178,6 @@ class AssetLibrary(PropertyGroup):
# update=update_library_path # update=update_library_path
# ) # )
# library_type : EnumProperty(items=library_type_ITEMS) # library_type : EnumProperty(items=library_type_ITEMS)
library_types: bpy.props.PointerProperty(type=LibraryTypes) library_types: bpy.props.PointerProperty(type=LibraryTypes)
library_type_name: EnumProperty(items=get_library_type_items) library_type_name: EnumProperty(items=get_library_type_items)
@ -172,7 +205,11 @@ class AssetLibrary(PropertyGroup):
@property @property
def merge_libraries(self): def merge_libraries(self):
prefs = get_addon_prefs() prefs = get_addon_prefs()
return [l for l in prefs.libraries if l != self and (l.library_path == self.library_path)] return [
l
for l in prefs.libraries
if l != self and (l.library_path == self.library_path)
]
@property @property
def child_libraries(self): def child_libraries(self):
@ -182,9 +219,9 @@ class AssetLibrary(PropertyGroup):
@property @property
def data_types(self): def data_types(self):
data_type = self.data_type data_type = self.data_type
if data_type == 'FILE': if data_type == "FILE":
data_type = 'COLLECTION' data_type = "COLLECTION"
return f'{data_type.lower()}s' return f"{data_type.lower()}s"
@property @property
def library_type(self): def library_type(self):
@ -258,11 +295,11 @@ class AssetLibrary(PropertyGroup):
for l in reversed(libs): for l in reversed(libs):
# lib_path = Path(l.path).resolve().as_posix() # lib_path = Path(l.path).resolve().as_posix()
prev_name = self.get('asset_library') or self.library_name prev_name = self.get("asset_library") or self.library_name
# print(l.name, prev_name) # print(l.name, prev_name)
if (l.name == prev_name): if l.name == prev_name:
index = list(libs).index(l) index = list(libs).index(l)
try: try:
bpy.ops.preferences.asset_library_remove(index=index) bpy.ops.preferences.asset_library_remove(index=index)
@ -286,18 +323,18 @@ class AssetLibrary(PropertyGroup):
for key, value in data.items(): for key, value in data.items():
if isinstance(value, dict): if isinstance(value, dict):
if 'name' in value: if "name" in value:
setattr(obj, f'{key}_name', value.pop('name')) setattr(obj, f"{key}_name", value.pop("name"))
# print('Nested value', getattr(obj, key)) # print('Nested value', getattr(obj, key))
self.set_dict(value, obj=getattr(obj, key)) self.set_dict(value, obj=getattr(obj, key))
elif key in obj.bl_rna.properties.keys(): elif key in obj.bl_rna.properties.keys():
if key == 'id': if key == "id":
value = str(value) value = str(value)
elif key == 'custom_bundle_name': elif key == "custom_bundle_name":
if not 'use_custom_bundle_name' in data.values(): if not "use_custom_bundle_name" in data.values():
obj["use_custom_bundle_name"] = True obj["use_custom_bundle_name"] = True
elif isinstance(value, str): elif isinstance(value, str):
@ -309,19 +346,18 @@ class AssetLibrary(PropertyGroup):
# obj[key] = value # obj[key] = value
else: else:
print(f'Prop {key} of {obj} not exist') print(f"Prop {key} of {obj} not exist")
self['bundle_directory'] = str(self.library_path) self["bundle_directory"] = str(self.library_path)
if not self.custom_bundle_name: if not self.custom_bundle_name:
self['custom_bundle_name'] = self.name self["custom_bundle_name"] = self.name
# self.library_type_name = data['library_type'] # self.library_type_name = data['library_type']
# if not self.library_type: # if not self.library_type:
# print(f"No library_type named {data['library_type']}") # print(f"No library_type named {data['library_type']}")
# return # return
# for key, value in data.items(): # for key, value in data.items():
# if key == 'options': # if key == 'options':
# for k, v in data['options'].items(): # for k, v in data['options'].items():
@ -337,22 +373,26 @@ class AssetLibrary(PropertyGroup):
# self[key] = value # self[key] = value
def to_dict(self): def to_dict(self):
data = {p: getattr(self, p) for p in self.bl_rna.properties.keys() if p !='rna_type'} data = {
p: getattr(self, p)
for p in self.bl_rna.properties.keys()
if p != "rna_type"
}
if self.library_type: if self.library_type:
data['library_type'] = self.library_type.to_dict() data["library_type"] = self.library_type.to_dict()
data['library_type']['name'] = data.pop('library_type_name') data["library_type"]["name"] = data.pop("library_type_name")
del data['library_types'] del data["library_types"]
if self.adapter: if self.adapter:
data['adapter'] = self.adapter.to_dict() data["adapter"] = self.adapter.to_dict()
data['adapter']['name'] = data.pop('adapter_name') data["adapter"]["name"] = data.pop("adapter_name")
del data['adapters'] del data["adapters"]
return data return data
def set_library_path(self): def set_library_path(self):
'''Update the Blender Preference Filepaths tab with the addon libraries''' """Update the Blender Preference Filepaths tab with the addon libraries"""
prefs = bpy.context.preferences prefs = bpy.context.preferences
name = self.library_name name = self.library_name
@ -360,7 +400,6 @@ class AssetLibrary(PropertyGroup):
self.clear_library_path() self.clear_library_path()
if not self.use or not lib_path: if not self.use or not lib_path:
# if all(not l.use for l in self.merge_libraries): # if all(not l.use for l in self.merge_libraries):
# self.clear_library_path() # self.clear_library_path()
@ -390,7 +429,7 @@ class AssetLibrary(PropertyGroup):
lib.name = name lib.name = name
self['asset_library'] = name self["asset_library"] = name
lib.path = str(lib_path) lib.path = str(lib_path)
@property @property
@ -403,24 +442,24 @@ class AssetLibrary(PropertyGroup):
prefs = get_addon_prefs() prefs = get_addon_prefs()
return self in prefs.env_libraries.values() return self in prefs.env_libraries.values()
def add_row(
def add_row(self, layout, data=None, prop=None, label='', self, layout, data=None, prop=None, label="", boolean=None, factor=0.39
boolean=None, factor=0.39): ):
'''Act like the use_property_split but with more control''' """Act like the use_property_split but with more control"""
enabled = True enabled = True
split = layout.split(factor=factor, align=True) split = layout.split(factor=factor, align=True)
row = split.row(align=False) row = split.row(align=False)
row.use_property_split = False row.use_property_split = False
row.alignment= 'RIGHT' row.alignment = "RIGHT"
row.label(text=str(label)) row.label(text=str(label))
if boolean: if boolean:
boolean_data = self boolean_data = self
if isinstance(boolean, (list, tuple)): if isinstance(boolean, (list, tuple)):
boolean_data, boolean = boolean boolean_data, boolean = boolean
row.prop(boolean_data, boolean, text='') row.prop(boolean_data, boolean, text="")
enabled = getattr(boolean_data, boolean) enabled = getattr(boolean_data, boolean)
row = split.row(align=True) row = split.row(align=True)
@ -429,20 +468,19 @@ class AssetLibrary(PropertyGroup):
if isinstance(data, str): if isinstance(data, str):
row.label(text=data) row.label(text=data)
else: else:
row.prop(data or self, prop, text='') row.prop(data or self, prop, text="")
return split return split
def draw_operators(self, layout): def draw_operators(self, layout):
row = layout.row(align=True) row = layout.row(align=True)
row.alignment = 'RIGHT' row.alignment = "RIGHT"
row.prop(self, 'library_type_name', text='') row.prop(self, "library_type_name", text="")
row.prop(self, 'auto_bundle', text='', icon='UV_SYNC_SELECT') row.prop(self, "auto_bundle", text="", icon="UV_SYNC_SELECT")
row.operator("assetlib.diff", text='', icon='FILE_REFRESH').name = self.name row.operator("assetlib.diff", text="", icon="FILE_REFRESH").name = self.name
op = row.operator("assetlib.bundle", icon='MOD_BUILD', text='') op = row.operator("assetlib.bundle", icon="MOD_BUILD", text="")
op.name = self.name op.name = self.name
layout.separator(factor=3) layout.separator(factor=3)
@ -456,42 +494,46 @@ class AssetLibrary(PropertyGroup):
# row.alignment = 'LEFT' # row.alignment = 'LEFT'
icon = "DISCLOSURE_TRI_DOWN" if self.expand else "DISCLOSURE_TRI_RIGHT" icon = "DISCLOSURE_TRI_DOWN" if self.expand else "DISCLOSURE_TRI_RIGHT"
row.prop(self, 'expand', icon=icon, emboss=False, text='') row.prop(self, "expand", icon=icon, emboss=False, text="")
if self.is_user: if self.is_user:
row.prop(self, 'use', text='') row.prop(self, "use", text="")
row.prop(self, 'data_type', icon_only=True, emboss=False) row.prop(self, "data_type", icon_only=True, emboss=False)
row.prop(self, 'name', text='') row.prop(self, "name", text="")
self.draw_operators(row) self.draw_operators(row)
index = list(prefs.user_libraries).index(self) index = list(prefs.user_libraries).index(self)
row.operator("assetlib.remove_user_library", icon="X", text='', emboss=False).index = index row.operator(
"assetlib.remove_user_library", icon="X", text="", emboss=False
).index = index
else: else:
row.prop(self, 'use', text='') row.prop(self, "use", text="")
row.label(icon=ICONS[self.data_type]) row.label(icon=ICONS[self.data_type])
# row.label(text=self.name) # row.label(text=self.name)
subrow = row.row(align=True) subrow = row.row(align=True)
subrow.alignment = 'LEFT' subrow.alignment = "LEFT"
subrow.prop(self, 'expand', emboss=False, text=self.name) subrow.prop(self, "expand", emboss=False, text=self.name)
# row.separator_spacer() # row.separator_spacer()
self.draw_operators(row) self.draw_operators(row)
sub_row = row.row() sub_row = row.row()
sub_row.enabled = False sub_row.enabled = False
sub_row.label(icon='FAKE_USER_ON') sub_row.label(icon="FAKE_USER_ON")
if self.expand: if self.expand:
col = layout.column(align=False) col = layout.column(align=False)
col.use_property_split = True col.use_property_split = True
# row = col.row(align=True) # row = col.row(align=True)
row = self.add_row(col, row = self.add_row(
col,
prop="custom_bundle_name", prop="custom_bundle_name",
boolean="use_custom_bundle_name", boolean="use_custom_bundle_name",
label='Custom Bundle Name') label="Custom Bundle Name",
)
row.enabled = not self.use_custom_bundle_directory row.enabled = not self.use_custom_bundle_directory
@ -499,9 +541,11 @@ class AssetLibrary(PropertyGroup):
if self.use_custom_bundle_directory: if self.use_custom_bundle_directory:
prop = "custom_bundle_directory" prop = "custom_bundle_directory"
self.add_row(col, prop=prop, self.add_row(
col,
prop=prop,
boolean="use_custom_bundle_directory", boolean="use_custom_bundle_directory",
label='Custom Bundle Directory', label="Custom Bundle Directory",
) )
col.prop(self, "blend_depth") col.prop(self, "blend_depth")
@ -521,9 +565,8 @@ class AssetLibrary(PropertyGroup):
col.separator() col.separator()
class Collections: class Collections:
'''Util Class to merge multiple collections''' """Util Class to merge multiple collections"""
collections = [] collections = []
@ -533,7 +576,7 @@ class Collections:
for col in collection: for col in collection:
# print('Merge methods') # print('Merge methods')
for attr in dir(col): for attr in dir(col):
if attr.startswith('_'): if attr.startswith("_"):
continue continue
value = getattr(col, attr) value = getattr(col, attr)
@ -598,49 +641,47 @@ class AssetLibraryPrefs(AddonPreferences):
# library_types = {} # library_types = {}
author: StringProperty(default=os.getlogin()) author: StringProperty(default=os.getlogin())
image_player: StringProperty(default='') image_player: StringProperty(default="")
video_player: StringProperty(default='') video_player: StringProperty(default="")
library_type_directory : StringProperty(name="Library Type Directory", subtype='DIR_PATH') library_type_directory: StringProperty(
adapter_directory : StringProperty(name="Adapter Directory", subtype='DIR_PATH') name="Library Type Directory", subtype="DIR_PATH"
)
adapter_directory: StringProperty(name="Adapter Directory", subtype="DIR_PATH")
env_libraries: CollectionProperty(type=AssetLibrary) env_libraries: CollectionProperty(type=AssetLibrary)
user_libraries: CollectionProperty(type=AssetLibrary) user_libraries: CollectionProperty(type=AssetLibrary)
expand_settings: BoolProperty(default=False) expand_settings: BoolProperty(default=False)
bundle_directory: StringProperty( bundle_directory: StringProperty(
name="Path", name="Path", subtype="DIR_PATH", default="", update=update_all_library_path
subtype='DIR_PATH',
default='',
update=update_all_library_path
) )
config_directory: StringProperty( config_directory: StringProperty(
name="Config Path", name="Config Path",
subtype='FILE_PATH', subtype="FILE_PATH",
default=str(RESOURCES_DIR / "asset_library_config.json"), default=str(RESOURCES_DIR / "asset_library_config.json"),
update=update_library_config update=update_library_config,
) )
def load_library_types(self): def load_library_types(self):
from asset_library.library_types.library_type import LibraryType from asset_library.library_types.library_type import LibraryType
print('Asset Library: Load Library Types') print("Asset Library: Load Library Types")
LIBRARY_TYPES.clear() LIBRARY_TYPES.clear()
library_type_files = list(LIBRARY_TYPE_DIR.glob('*.py')) library_type_files = list(LIBRARY_TYPE_DIR.glob("*.py"))
if self.library_type_directory: if self.library_type_directory:
user_LIBRARY_TYPE_DIR = Path(self.library_type_directory) user_LIBRARY_TYPE_DIR = Path(self.library_type_directory)
if user_LIBRARY_TYPE_DIR.exists(): if user_LIBRARY_TYPE_DIR.exists():
library_type_files += list(user_LIBRARY_TYPE_DIR.glob('*.py')) library_type_files += list(user_LIBRARY_TYPE_DIR.glob("*.py"))
for library_type_file in library_type_files: for library_type_file in library_type_files:
if library_type_file.stem.startswith('_'): if library_type_file.stem.startswith("_"):
continue continue
mod = import_module_from_path(library_type_file) mod = import_module_from_path(library_type_file)
# print(library_type_file) # print(library_type_file)
for name, obj in inspect.getmembers(mod): for name, obj in inspect.getmembers(mod):
@ -656,13 +697,17 @@ class AssetLibraryPrefs(AddonPreferences):
continue continue
try: try:
print(f'Register Plugin {name}') print(f"Register Plugin {name}")
bpy.utils.register_class(obj) bpy.utils.register_class(obj)
setattr(LibraryTypes, norm_str(obj.name), bpy.props.PointerProperty(type=obj)) setattr(
LibraryTypes,
norm_str(obj.name),
bpy.props.PointerProperty(type=obj),
)
LIBRARY_TYPES.append(obj) LIBRARY_TYPES.append(obj)
except Exception as e: except Exception as e:
print(f'Could not register library_type {name}') print(f"Could not register library_type {name}")
print(e) print(e)
def load_adapters(self): def load_adapters(self):
@ -683,25 +728,25 @@ class AssetLibraryPrefs(AddonPreferences):
box = main_col.box() box = main_col.box()
row = box.row(align=True) row = box.row(align=True)
icon = "DISCLOSURE_TRI_DOWN" if self.expand_settings else "DISCLOSURE_TRI_RIGHT" icon = "DISCLOSURE_TRI_DOWN" if self.expand_settings else "DISCLOSURE_TRI_RIGHT"
row.prop(self, 'expand_settings', icon=icon, emboss=False, text='') row.prop(self, "expand_settings", icon=icon, emboss=False, text="")
row.label(icon='PREFERENCES') row.label(icon="PREFERENCES")
row.label(text='Settings') row.label(text="Settings")
# row.separator_spacer() # row.separator_spacer()
subrow = row.row() subrow = row.row()
subrow.alignment = 'RIGHT' subrow.alignment = "RIGHT"
subrow.operator("assetlib.reload_addon", text='Reload Addon') subrow.operator("assetlib.reload_addon", text="Reload Addon")
if prefs.expand_settings: if prefs.expand_settings:
col = box.column(align=True) col = box.column(align=True)
col.use_property_split = True col.use_property_split = True
# col.prop(self, 'use_single_path', text='Single Path') # col.prop(self, 'use_single_path', text='Single Path')
col.prop(self, 'bundle_directory', text='Bundle Directory') col.prop(self, "bundle_directory", text="Bundle Directory")
col.separator() col.separator()
col.prop(self, 'library_type_directory') col.prop(self, "library_type_directory")
col.prop(self, 'config_directory') col.prop(self, "config_directory")
col.separator() col.separator()
@ -710,18 +755,26 @@ class AssetLibraryPrefs(AddonPreferences):
# col.separator() # col.separator()
# col.prop(self, 'template_image', text='Template Image', icon='COPY_ID') # col.prop(self, 'template_image', text='Template Image', icon='COPY_ID')
col.prop(self, 'image_player', text='Image Player') #icon='OUTLINER_OB_IMAGE' col.prop(
self, "image_player", text="Image Player"
) # icon='OUTLINER_OB_IMAGE'
# col.separator() # col.separator()
# col.prop(self, 'template_video', text='Template Video', icon='COPY_ID') # col.prop(self, 'template_video', text='Template Video', icon='COPY_ID')
col.prop(self, 'video_player', text='Video Player') #icon='FILE_MOVIE' col.prop(self, "video_player", text="Video Player") # icon='FILE_MOVIE'
col.separator() col.separator()
col.operator("assetlib.add_user_library", text='Bundle All Libraries', icon='MOD_BUILD') col.operator(
"assetlib.add_user_library",
text="Bundle All Libraries",
icon="MOD_BUILD",
)
for lib in self.libraries:# list(self.env_libraries) + list(self.user_libraries): for (
lib
) in self.libraries: # list(self.env_libraries) + list(self.user_libraries):
if lib.parent: if lib.parent:
continue continue
@ -729,8 +782,8 @@ class AssetLibraryPrefs(AddonPreferences):
lib.draw(box) lib.draw(box)
row = main_col.row() row = main_col.row()
row.alignment = 'RIGHT' row.alignment = "RIGHT"
row.operator("assetlib.add_user_library", icon="ADD", text='', emboss=False) row.operator("assetlib.add_user_library", icon="ADD", text="", emboss=False)
classes = [ classes = [
@ -741,6 +794,7 @@ classes = [
AssetLibraryPrefs, AssetLibraryPrefs,
] ]
def register(): def register():
for cls in classes: for cls in classes:
bpy.utils.register_class(cls) bpy.utils.register_class(cls)
@ -748,25 +802,26 @@ def register():
prefs = get_addon_prefs() prefs = get_addon_prefs()
# Read Env and override preferences # Read Env and override preferences
bundle_dir = os.getenv('ASSETLIB_BUNDLE_DIR') bundle_dir = os.getenv("ASSETLIB_BUNDLE_DIR")
if bundle_dir: if bundle_dir:
prefs['bundle_directory'] = os.path.expandvars(bundle_dir) prefs["bundle_directory"] = os.path.expandvars(bundle_dir)
config_dir = os.getenv('ASSETLIB_CONFIG_DIR') config_dir = os.getenv("ASSETLIB_CONFIG_DIR")
if config_dir: if config_dir:
prefs['config_directory'] = os.path.expandvars(config_dir) prefs["config_directory"] = os.path.expandvars(config_dir)
LIBRARY_TYPE_DIR = os.getenv('ASSETLIB_LIBRARY_TYPE_DIR') LIBRARY_TYPE_DIR = os.getenv("ASSETLIB_LIBRARY_TYPE_DIR")
if LIBRARY_TYPE_DIR: if LIBRARY_TYPE_DIR:
prefs['library_type_directory'] = os.path.expandvars(LIBRARY_TYPE_DIR) prefs["library_type_directory"] = os.path.expandvars(LIBRARY_TYPE_DIR)
ADAPTER_DIR = os.getenv('ASSETLIB_ADAPTER_DIR') ADAPTER_DIR = os.getenv("ASSETLIB_ADAPTER_DIR")
if ADAPTER_DIR: if ADAPTER_DIR:
prefs['adapter_directory'] = os.path.expandvars(ADAPTER_DIR) prefs["adapter_directory"] = os.path.expandvars(ADAPTER_DIR)
prefs.load_library_types() prefs.load_library_types()
prefs.load_adapters() prefs.load_adapters()
def unregister(): def unregister():
for cls in reversed(classes + LIBRARY_TYPES): for cls in reversed(classes + LIBRARY_TYPES):
bpy.utils.unregister_class(cls) bpy.utils.unregister_class(cls)

View File

@ -1,4 +1,3 @@
import bpy import bpy
from pathlib import Path from pathlib import Path
from asset_library.common.file_utils import read_file, write_file from asset_library.common.file_utils import read_file, write_file
@ -14,13 +13,13 @@ class AssetCache:
self._data = data self._data = data
self.catalog = data['catalog'] self.catalog = data["catalog"]
self.author = data.get('author', '') self.author = data.get("author", "")
self.description = data.get('description', '') self.description = data.get("description", "")
self.tags = data.get('tags', []) self.tags = data.get("tags", [])
self.type = data.get('type') self.type = data.get("type")
self.name = data['name'] self.name = data["name"]
self._metadata = data.get('metadata', {}) self._metadata = data.get("metadata", {})
@property @property
def filepath(self): def filepath(self):
@ -28,10 +27,7 @@ class AssetCache:
@property @property
def metadata(self): def metadata(self):
metadata = { metadata = {".library_id": self.library.id, ".filepath": self.filepath}
'.library_id': self.library.id,
'.filepath': self.filepath
}
metadata.update(self.metadata) metadata.update(self.metadata)
@ -39,7 +35,7 @@ class AssetCache:
@property @property
def norm_name(self): def norm_name(self):
return self.name.replace(' ', '_').lower() return self.name.replace(" ", "_").lower()
def to_dict(self): def to_dict(self):
return dict( return dict(
@ -49,23 +45,23 @@ class AssetCache:
description=self.description, description=self.description,
tags=self.tags, tags=self.tags,
type=self.type, type=self.type,
name=self.name name=self.name,
) )
def __str__(self): def __str__(self):
return f'AssetCache(name={self.name}, type={self.type}, catalog={self.catalog})' return f"AssetCache(name={self.name}, type={self.type}, catalog={self.catalog})"
class FileCache: class FileCache:
def __init__(self, library_cache, data): def __init__(self, library_cache, data):
self.library_cache = library_cache self.library_cache = library_cache
self.filepath = data['filepath'] self.filepath = data["filepath"]
self.modified = data.get('modified', time.time_ns()) self.modified = data.get("modified", time.time_ns())
self._data = [] self._data = []
for asset_cache_data in data.get('assets', []): for asset_cache_data in data.get("assets", []):
self.add(asset_cache_data) self.add(asset_cache_data)
def add(self, asset_cache_data): def add(self, asset_cache_data):
@ -77,7 +73,7 @@ class FileCache:
filepath=self.filepath.as_posix(), filepath=self.filepath.as_posix(),
modified=self.modified, modified=self.modified,
library_id=self.library_cache.library.id, library_id=self.library_cache.library.id,
assets=[asset_cache.to_dict() for asset_cache in self] assets=[asset_cache.to_dict() for asset_cache in self],
) )
def __iter__(self): def __iter__(self):
@ -87,14 +83,14 @@ class FileCache:
return self._data[key] return self._data[key]
def __str__(self): def __str__(self):
return f'FileCache(filepath={self.filepath})' return f"FileCache(filepath={self.filepath})"
class AssetCacheDiff: class AssetCacheDiff:
def __init__(self, library_diff, asset_cache, operation): def __init__(self, library_diff, asset_cache, operation):
self.library_cache = library_cache self.library_cache = library_cache
self.filepath = data['filepath'] self.filepath = data["filepath"]
self.operation = operation self.operation = operation
@ -113,7 +109,7 @@ class LibraryCacheDiff:
self.add(asset_diff) self.add(asset_diff)
def read(self): def read(self):
print(f'Read cache from {self.filepath}') print(f"Read cache from {self.filepath}")
for asset_diff_data in read_file(self.filepath): for asset_diff_data in read_file(self.filepath):
self.add(asset_diff_data) self.add(asset_diff_data)
@ -121,7 +117,7 @@ class LibraryCacheDiff:
return self return self
def group_by(self, key): def group_by(self, key):
'''Return groups of file cache diff using the key provided''' """Return groups of file cache diff using the key provided"""
data = list(self).sort(key=key) data = list(self).sort(key=key)
return groupby(data, key=key) return groupby(data, key=key)
@ -158,7 +154,7 @@ class LibraryCache:
@property @property
def asset_caches(self): def asset_caches(self):
'''Return an iterator to get all asset caches''' """Return an iterator to get all asset caches"""
return (asset_cache for file_cache in self for asset_cache in file_cache) return (asset_cache for file_cache in self for asset_cache in file_cache)
@property @property
@ -166,7 +162,7 @@ class LibraryCache:
return Path(bpy.app.tempdir) / self.filename return Path(bpy.app.tempdir) / self.filename
def read(self): def read(self):
print(f'Read cache from {self.filepath}') print(f"Read cache from {self.filepath}")
for file_cache_data in read_file(self.filepath): for file_cache_data in read_file(self.filepath):
self.add(file_cache_data) self.add(file_cache_data)
@ -178,7 +174,7 @@ class LibraryCache:
if temp: if temp:
filepath = self.tmp_filepath filepath = self.tmp_filepath
print(f'Write cache file to {filepath}') print(f"Write cache file to {filepath}")
write_file(filepath, self._data) write_file(filepath, self._data)
return filepath return filepath
@ -195,10 +191,10 @@ class LibraryCache:
cache = deepcopy(cache) cache = deepcopy(cache)
cache.sort(key=lambda x : x['filepath']) cache.sort(key=lambda x: x["filepath"])
groups = groupby(cache, key=lambda x : x['filepath']) groups = groupby(cache, key=lambda x: x["filepath"])
keys = ['filepath', 'modified', 'library_id'] keys = ["filepath", "modified", "library_id"]
for _, asset_datas in groups: for _, asset_datas in groups:
asset_datas = list(asset_datas) asset_datas = list(asset_datas)
@ -206,7 +202,10 @@ class LibraryCache:
# print(asset_datas[0]) # print(asset_datas[0])
asset_info = {k: asset_datas[0][k] for k in keys} asset_info = {k: asset_datas[0][k] for k in keys}
asset_info['assets'] = [{k:v for k, v in a.items() if k not in keys+['operation']} for a in asset_datas] asset_info["assets"] = [
{k: v for k, v in a.items() if k not in keys + ["operation"]}
for a in asset_datas
]
new_cache.append(asset_info) new_cache.append(asset_info)
@ -218,24 +217,40 @@ class LibraryCache:
cache = self.read() cache = self.read()
cache_dict = {f"{a['filepath']}/{a['name']}": a for a in cache.asset_caches} cache_dict = {f"{a['filepath']}/{a['name']}": a for a in cache.asset_caches}
new_cache_dict = {f"{a['filepath']}/{a['name']}" : a for a in new_cache.asset_caches} new_cache_dict = {
f"{a['filepath']}/{a['name']}": a for a in new_cache.asset_caches
}
assets_added = [AssetCacheDiff(v, 'ADD') for k, v in new_cache.items() if k not in cache] assets_added = [
assets_removed = [AssetCacheDiff(v, 'REMOVED') for k, v in cache.items() if k not in new_cache] AssetCacheDiff(v, "ADD") for k, v in new_cache.items() if k not in cache
assets_modified = [AssetCacheDiff(v, 'MODIFIED') for k, v in cache.items() if v not in assets_removed and v!= new_cache[k]] ]
assets_removed = [
AssetCacheDiff(v, "REMOVED") for k, v in cache.items() if k not in new_cache
]
assets_modified = [
AssetCacheDiff(v, "MODIFIED")
for k, v in cache.items()
if v not in assets_removed and v != new_cache[k]
]
if assets_added: if assets_added:
print(f'{len(assets_added)} Assets Added \n{tuple(a.name for a in assets_added[:10])}...\n') print(
f"{len(assets_added)} Assets Added \n{tuple(a.name for a in assets_added[:10])}...\n"
)
if assets_removed: if assets_removed:
print(f'{len(assets_removed)} Assets Removed \n{tuple(a.name for a in assets_removed[:10])}...\n') print(
f"{len(assets_removed)} Assets Removed \n{tuple(a.name for a in assets_removed[:10])}...\n"
)
if assets_modified: if assets_modified:
print(f'{len(assets_modified)} Assets Modified \n{tuple(a.name for a in assets_modified[:10])}...\n') print(
f"{len(assets_modified)} Assets Modified \n{tuple(a.name for a in assets_modified[:10])}...\n"
)
cache_diff = LibraryCacheDiff() cache_diff = LibraryCacheDiff()
cache_diff.set(assets_added + assets_removed + assets_modified) cache_diff.set(assets_added + assets_removed + assets_modified)
if not len(LibraryCacheDiff): if not len(LibraryCacheDiff):
print('No change in the library') print("No change in the library")
return cache_diff return cache_diff
@ -249,11 +264,12 @@ class LibraryCache:
return self._data[key] return self._data[key]
def __str__(self): def __str__(self):
return f'LibraryCache(library={self.library.name})' return f"LibraryCache(library={self.library.name})"
print() print()
prefs = bpy.context.preferences.addons['asset_library'].preferences prefs = bpy.context.preferences.addons["asset_library"].preferences
library = prefs.env_libraries[0] library = prefs.env_libraries[0]