add in module register calls (first pass, batch replace, will test each addon next).

3 files changed, 12 insertions, 0 deletions

diff --git a/rigify/__init__.py b/rigify/__init__.py
index 20821f9d..cf0ed06a 100644
--- a/rigify/__init__.py
+++ b/rigify/__init__.py
@@ -127,6 +127,8 @@ for rig in rig_list:
 ##### REGISTER #####
 
 def register():
+    bpy.utils.register_module(__name__)
+
     bpy.types.PoseBone.rigify_type = bpy.props.StringProperty(name="Rigify Type", description="Rig type for this bone.")
     bpy.types.PoseBone.rigify_parameters = bpy.props.CollectionProperty(type=RigifyParameters)
 
@@ -139,6 +141,8 @@ def register():
 
 
 def unregister():
+    bpy.utils.unregister_module(__name__)
+
     del bpy.types.PoseBone.rigify_type
     del bpy.types.PoseBone.rigify_parameters
 
diff --git a/rigify/metarig_menu.py b/rigify/metarig_menu.py
index 65dbbce9..54fae2d5 100644
--- a/rigify/metarig_menu.py
+++ b/rigify/metarig_menu.py
@@ -46,11 +46,15 @@ menu_func = (lambda self, context: self.layout.operator(AddHuman.bl_idname,
 
 
 def register():
+    bpy.utils.register_module(__name__)
+
     #bpy.types.register(AddHuman)
     bpy.types.INFO_MT_armature_add.append(menu_func)
 
 
 def unregister():
+    bpy.utils.unregister_module(__name__)
+
     #bpy.types.unregister(AddHuman)
     bpy.types.INFO_MT_armature_add.remove(menu_func)
 
diff --git a/rigify/ui.py b/rigify/ui.py
index a27d319b..827b0fc2 100644
--- a/rigify/ui.py
+++ b/rigify/ui.py
@@ -233,6 +233,8 @@ class Sample(bpy.types.Operator):
 #import space_info  # ensure the menu is loaded first
 
 def register():
+    bpy.utils.register_module(__name__)
+
     #bpy.types.register(DATA_PT_rigify_buttons)
     #bpy.types.register(BONE_PT_rigify_buttons)
     #bpy.types.register(Generate)
@@ -243,6 +245,8 @@ def register():
 
 
 def unregister():
+    bpy.utils.unregister_module(__name__)
+
     #bpy.types.unregister(DATA_PT_rigify_buttons)
     #bpy.types.unregister(BONE_PT_rigify_buttons)
     #bpy.types.unregister(Generate)