bugfix [#23065] Pickle can not dump instances of user defined classes
- __import__("__main__").__dict__ will now always match the current scripts namespace. (which is what pickle expects).
- __builtins__ as a module rather then a dict from PyEval_GetBuiltins() acts slightly differently, use the module to follow python.
This commit is contained in:
@@ -152,8 +152,27 @@ void BPY_update_modules( void )
|
|||||||
static PyObject *CreateGlobalDictionary( bContext *C, const char *filename )
|
static PyObject *CreateGlobalDictionary( bContext *C, const char *filename )
|
||||||
{
|
{
|
||||||
PyObject *item;
|
PyObject *item;
|
||||||
PyObject *dict = PyDict_New( );
|
PyObject *dict;
|
||||||
PyDict_SetItemString( dict, "__builtins__", PyEval_GetBuiltins( ) );
|
#if 1
|
||||||
|
/* important we use the dict from __main__, this is what python expects
|
||||||
|
* for 'pickle' to work as well as strings like this...
|
||||||
|
|
||||||
|
>> foo = 10
|
||||||
|
>> print(__import__("__main__").foo)
|
||||||
|
*/
|
||||||
|
dict= PyModule_GetDict(PyImport_AddModule("__main__"));
|
||||||
|
PyDict_Clear(dict);
|
||||||
|
Py_INCREF(dict);
|
||||||
|
|
||||||
|
/* using builtins rather then PyEval_GetBuiltins()
|
||||||
|
* print's many less items when printing, the modules __dict__
|
||||||
|
* this is how python works so better follow. */
|
||||||
|
PyDict_SetItemString(dict, "__builtins__", PyImport_AddModule("builtins"));
|
||||||
|
#else
|
||||||
|
/* otherwise this works for 99% of cases, from 2.4x */
|
||||||
|
dict = PyDict_New();
|
||||||
|
PyDict_SetItemString( dict, "__builtins__", PyEval_GetBuiltins());
|
||||||
|
#endif
|
||||||
|
|
||||||
item = PyUnicode_FromString( "__main__" );
|
item = PyUnicode_FromString( "__main__" );
|
||||||
PyDict_SetItemString( dict, "__name__", item );
|
PyDict_SetItemString( dict, "__name__", item );
|
||||||
@@ -389,6 +408,10 @@ int BPY_run_python_script( bContext *C, const char *fn, struct Text *text, struc
|
|||||||
Py_DECREF( py_result );
|
Py_DECREF( py_result );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* so __main__ module isnt left with an invalid __file__ variable which could be confusing */
|
||||||
|
if (PyDict_DelItemString(py_dict, "__file__"))
|
||||||
|
PyErr_Clear();
|
||||||
|
|
||||||
Py_DECREF(py_dict);
|
Py_DECREF(py_dict);
|
||||||
|
|
||||||
bpy_context_clear(C, &gilstate);
|
bpy_context_clear(C, &gilstate);
|
||||||
|
|||||||
Reference in New Issue
Block a user