fix object bones armObj.data.bones[bone]

aiaustin · Dec 4, 2024 · fef1458 · fef1458
1 parent 6c2817c
commit fef1458
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/Onigiri/rigutils.py b/Onigiri/rigutils.py
@@ -3595,19 +3595,20 @@ def set_bone_groups(armObj):
     for bone in skel.avatar_skeleton:
         if bone not in armObj.data.bones:
             continue
+        boneObj = armObj.data.bones[bone]
         if skel.avatar_skeleton[bone]["type"] == "bone":
-            armObj.data.collections[mod_data.rig_group_mbones].assign(bone)
-            bone.palette = mod_data.rig_group_mtheme
+            armObj.data.collections[mod_data.rig_group_mbones].assign(boneObj)
+            boneObj.palette = mod_data.rig_group_mtheme
         elif skel.avatar_skeleton[bone]["type"] == "attachment":
             if " " in bone:
-                armObj.data.collections[mod_data.rig_group_nbones].assign(bone)
-                bone.palette = mod_data.rig_group_ntheme
+                armObj.data.collections[mod_data.rig_group_nbones].assign(boneObj)
+                boneObj.palette = mod_data.rig_group_ntheme
             else:
-                armObj.data.collections[mod_data.rig_group_abones].assign(bone)
-                bone.palette = mod_data.rig_group_atheme
+                armObj.data.collections[mod_data.rig_group_abones].assign(boneObj)
+                boneObj.palette = mod_data.rig_group_atheme
         else:
-            armObj.data.collections[mod_data.rig_group_vbones].assign(bone)
-            bone.palette = mod_data.rig_group_vtheme
+            armObj.data.collections[mod_data.rig_group_vbones].assign(boneObj)
+            boneObj.palette = mod_data.rig_group_vtheme
 
     bpy.ops.object.mode_set(mode="OBJECT")