fix #53 better lock handling

Author: Lauloque

convert_Rotation_Mode/operators.py

@@ -2,7 +2,14 @@
 import bpy
 from bpy.types import Operator
 from bpy.types import Context
-from .utils import dprint, is_pose_mode, get_fcurves, toggle_rotation_locks, jump_next_frame
+from .utils import (
+    dprint,
+    is_pose_mode,
+    get_fcurves,
+    get_rotation_locks,
+    toggle_rotation_locks,
+    jump_next_frame,
+)
 
 
 class CRM_OT_convert_rotation_mode(Operator):
@@ -53,13 +60,8 @@ def execute(self, context):
                 f" # Target Rmode will be {CRM_Properties.targetRmode}"
             )
 
-            # self.locks = []
-            self.locks.append(current_bone.lock_rotation[0])
-            self.locks.append(current_bone.lock_rotation[1])
-            self.locks.append(current_bone.lock_rotation[2])
-            self.locks.append(current_bone.lock_rotation_w)
-            self.locks.append(current_bone.lock_rotations_4d)
-            toggle_rotation_locks('OFF', current_bone)
+            locks = get_rotation_locks(current_bone)
+            toggle_rotation_locks(current_bone, 'OFF')
             dprint(" |  # Backed up and unlocked rotations")
 
             original_rmode = current_bone.rotation_mode
@@ -128,7 +130,7 @@ def execute(self, context):
                     break
 
             if CRM_Properties.preserveLocks:
-                toggle_rotation_locks('ON', current_bone)
+                toggle_rotation_locks(current_bone, 'ON', locks)
                 dprint(" |  # Reverted rotation locks")
 
             dprint(

convert_Rotation_Mode/utils.py

@@ -18,6 +18,14 @@ def get_fcurves(obj):
         return None
 
 
+def get_rotation_locks(bone):
+    """Return the current rotation lock state of the bone as a list."""
+    return list(bone.lock_rotation) + [
+        bone.lock_rotation_w,
+        bone.lock_rotations_4d,
+    ]
+
+
 def jump_next_frame(context):
     """Jump to the next frame in the timeline."""
     bpy.ops.screen.keyframe_jump(next=True)