Bugfix for #5000

Setup for Armature weak ref list was missing from some places where
we execute py code.  This confused the interpreter and gave random
attribute/tuple parse errors.
Changed name of weak ref list to "__arm_weakrefs" to avoid name
collision with user variables.

3 files changed, 64 insertions, 42 deletions

diff --git a/source/blender/python/api2_2x/Armature.c b/source/blender/python/api2_2x/Armature.c
index 658a06d4207..d255183e71b 100644
--- a/source/blender/python/api2_2x/Armature.c
+++ b/source/blender/python/api2_2x/Armature.c
@@ -1348,36 +1348,38 @@ PyObject *Armature_RebuildBones(PyObject *pyarmature)
 PyObject *PyArmature_FromArmature(struct bArmature *armature)
 {
 	BPy_Armature *py_armature = NULL;
-	PyObject *maindict = NULL, *armdict = NULL, *weakref = NULL;
+	PyObject *maindict = NULL, *weakref = NULL;
+	PyObject *armlist = NULL;  /* list of armature weak refs */
+	char *list_name = ARM_WEAKREF_LIST_NAME;
 
 	//create armature type
-	py_armature = (BPy_Armature*)PyObject_New(BPy_Armature, &Armature_Type); //Armature_Type.tp_alloc(&Armature_Type, 0); //*new*
-	py_armature->weaklist = NULL; //init the weaklist
-	if (!py_armature)
+	py_armature = (BPy_Armature*)Armature_Type.tp_alloc(&Armature_Type, 0); /*new*/
+	if (!py_armature){
+		printf("Oops - can't create py armature\n");
 		goto RuntimeError;
+	}
+
+	py_armature->weaklist = NULL; //init the weaklist
 	py_armature->armature = armature;
 	
 	//create armature.bones
 	py_armature->Bones = (BPy_BonesDict*)PyBonesDict_FromPyArmature(py_armature);
-	if (!py_armature->Bones)
+	if (!py_armature->Bones){
+		printf("Oops - creating armature.bones\n");
 		goto RuntimeError;
+	}
 
 	//put a weakreference in __main__
 	maindict= PyModule_GetDict(PyImport_AddModule(	"__main__"));
-	
-	armdict = PyDict_GetItemString(maindict, "armatures");
-	/*armature list doesn't exist  for pydrivers. . .so check.
-	  by the way, why is the var called armatures? what if a script author has
-	  a similar var in his script? won't they conflict?.
-	  
-	  joeedh*/
-	if (!armdict) return (PyObject *) py_armature;
-	
-	weakref = PyWeakref_NewProxy((PyObject*)py_armature, Py_None);
-	if (PyList_Append(armdict, weakref) == -1){
-		goto RuntimeError;
-	}
 
+	armlist = PyDict_GetItemString(maindict, list_name);
+	if( armlist){
+		weakref = PyWeakref_NewProxy((PyObject*)py_armature, Py_None);
+		if (PyList_Append(armlist, weakref) == -1){
+			printf("Oops - list-append failed\n");
+			goto RuntimeError;
+		}
+	}
 	return (PyObject *) py_armature;
 
 RuntimeError:
diff --git a/source/blender/python/api2_2x/Window.c b/source/blender/python/api2_2x/Window.c
index cf1c6e1688f..1fcc252f581 100644
--- a/source/blender/python/api2_2x/Window.c
+++ b/source/blender/python/api2_2x/Window.c
@@ -884,18 +884,42 @@ static PyObject *M_Window_GetPerspMatrix( PyObject * self )
 	return perspmat;
 }
 
+
+/* update_armature_weakrefs()
+ * helper function used in M_Window_EditMode.
+ * rebuilds list of Armature weakrefs in __main__
+ */
+
+static int update_armature_weakrefs()
+{
+	/* stuff for armature weak refs */
+	char *list_name = ARM_WEAKREF_LIST_NAME;
+	PyObject *maindict = NULL, *armlist = NULL;
+	PyObject *pyarmature = NULL;
+	int x;
+
+	maindict= PyModule_GetDict(PyImport_AddModule(	"__main__"));
+	armlist = PyDict_GetItemString(maindict, list_name);
+	if( !armlist){
+		printf("Oops - update_armature_weakrefs()\n");
+		return 0;
+	}
+
+	for (x = 0; x < PySequence_Size(armlist); x++){
+		pyarmature = PyWeakref_GetObject(PySequence_GetItem(	armlist, x));
+		if (pyarmature != Py_None)
+			Armature_RebuildEditbones(pyarmature);
+	}
+	return 1;
+}
+
+
 static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
 {
 	short status = -1;
 	char *undo_str = "From script";
 	int undo_str_len = 11;
 	int do_undo = 1;
-#if 0
-/* bug 5000 */
-	PyObject *maindict = NULL, *armlist = NULL;
-	PyObject *pyarmature = NULL;
-	int x;
-#endif
 
 	if( !PyArg_ParseTuple( args,
 				"|hs#i", &status, &undo_str, &undo_str_len, &do_undo ) )
@@ -905,17 +929,14 @@ static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
 	if( status >= 0 ) {
 		if( status ) {
 			if( !G.obedit ){
-#if 0
-/* bug 5000 */
+
 				//update armatures
-				maindict= PyModule_GetDict(PyImport_AddModule(	"__main__"));
-				armlist = PyDict_GetItemString(maindict, "armatures");
-				for (x = 0; x < PySequence_Size(armlist); x++){
-					pyarmature = PyWeakref_GetObject(PySequence_GetItem(	armlist, x));
-					if (pyarmature != Py_None)
-						Armature_RebuildEditbones(pyarmature);
+				if(! update_armature_weakrefs()){
+					return EXPP_ReturnPyObjError( 
+						PyExc_RuntimeError,
+						"internal error -  update_armature_weakrefs");
 				}
-#endif
+
 				//enter editmode
 				enter_editmode(0);
 			}
@@ -925,17 +946,13 @@ static PyObject *M_Window_EditMode( PyObject * self, PyObject * args )
 			BIF_undo_push( undo_str ); /* This checks user undo settings */
 			exit_editmode( EM_FREEDATA );
 
-#if 0
-/* bug 5000 */
 			//update armatures
-			maindict= PyModule_GetDict(PyImport_AddModule(	"__main__"));
-			armlist = PyDict_GetItemString(maindict, "armatures");
-			for (x = 0; x < PySequence_Size(armlist); x++){
-				pyarmature = PyWeakref_GetObject(PySequence_GetItem(	armlist, x));
-				if (pyarmature != Py_None)
-					Armature_RebuildBones(pyarmature);
+			if(! update_armature_weakrefs()){
+				return EXPP_ReturnPyObjError( 
+					PyExc_RuntimeError,
+					"internal error -  update_armature_weakrefs");
 			}
-#endif
+
 		}
 	}
 
diff --git a/source/blender/python/api2_2x/gen_utils.h b/source/blender/python/api2_2x/gen_utils.h
index 04eddebddc1..488975ce31c 100644
--- a/source/blender/python/api2_2x/gen_utils.h
+++ b/source/blender/python/api2_2x/gen_utils.h
@@ -59,6 +59,9 @@
 #define Py_RETURN_TRUE  return PyBool_FromLong(1)
 #endif
 
+/* name of list of Armature weak refs built into __main__ */
+#define ARM_WEAKREF_LIST_NAME "__arm_weakrefs"
+
 int EXPP_FloatsAreEqual(float A, float B, int floatSteps);
 int EXPP_VectorsAreEqual(float *vecA, float *vecB, int size, int floatSteps);