14d74bcf5d01683f5cd1c7c81785e0ea3c87ee52
[blender.git] / source / blender / python / intern / bpy_interface.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor(s): Michel Selten, Willian P. Germano, Stephen Swaney,
19  * Chris Keith, Chris Want, Ken Hughes, Campbell Barton
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  */
23
24 /** \file blender/python/intern/bpy_interface.c
25  *  \ingroup pythonintern
26  *
27  * This file deals with embedding the python interpreter within blender,
28  * starting and stopping python and exposing blender/python modules so they can
29  * be accesses from scripts.
30  */
31
32  
33 /* grr, python redefines */
34 #ifdef _POSIX_C_SOURCE
35 #  undef _POSIX_C_SOURCE
36 #endif
37
38 #include <Python.h>
39
40 #include "MEM_guardedalloc.h"
41
42 #include "RNA_types.h"
43
44 #include "bpy.h"
45 #include "gpu.h"
46 #include "bpy_rna.h"
47 #include "bpy_util.h"
48 #include "bpy_traceback.h"
49 #include "bpy_intern_string.h"
50
51 #include "DNA_space_types.h"
52 #include "DNA_text_types.h"
53
54 #include "BLI_path_util.h"
55 #include "BLI_math_base.h"
56 #include "BLI_string.h"
57 #include "BLI_string_utf8.h"
58 #include "BLI_utildefines.h"
59
60
61 #include "BKE_context.h"
62 #include "BKE_text.h"
63 #include "BKE_main.h"
64 #include "BKE_global.h" /* only for script checking */
65
66 #include "BPY_extern.h"
67
68 #include "../generic/bpy_internal_import.h" // our own imports
69 #include "../generic/py_capi_utils.h"
70
71 /* inittab initialization functions */
72 #include "../generic/bgl.h"
73 #include "../generic/blf_py_api.h"
74 #include "../mathutils/mathutils.h"
75
76 /* for internal use, when starting and ending python scripts */
77
78 /* incase a python script triggers another python call, stop bpy_context_clear from invalidating */
79 static int py_call_level= 0;
80 BPy_StructRNA *bpy_context_module= NULL; /* for fast access */
81
82 // #define TIME_PY_RUN // simple python tests. prints on exit.
83
84 #ifdef TIME_PY_RUN
85 #include "PIL_time.h"
86 static int              bpy_timer_count= 0;
87 static double   bpy_timer; /* time since python starts */
88 static double   bpy_timer_run; /* time for each python script run */
89 static double   bpy_timer_run_tot; /* accumulate python runs */
90 #endif
91
92 /* use for updating while a python script runs - in case of file load */
93 void bpy_context_update(bContext *C)
94 {
95         BPy_SetContext(C);
96         bpy_import_main_set(CTX_data_main(C));
97         BPY_modules_update(C); /* can give really bad results if this isnt here */
98 }
99
100 void bpy_context_set(bContext *C, PyGILState_STATE *gilstate)
101 {
102         py_call_level++;
103
104         if (gilstate)
105                 *gilstate= PyGILState_Ensure();
106
107         if (py_call_level==1) {
108                 bpy_context_update(C);
109
110 #ifdef TIME_PY_RUN
111                 if (bpy_timer_count==0) {
112                         /* record time from the beginning */
113                         bpy_timer= PIL_check_seconds_timer();
114                         bpy_timer_run= bpy_timer_run_tot= 0.0;
115                 }
116                 bpy_timer_run= PIL_check_seconds_timer();
117
118
119                 bpy_timer_count++;
120 #endif
121         }
122 }
123
124 /* context should be used but not now because it causes some bugs */
125 void bpy_context_clear(bContext *UNUSED(C), PyGILState_STATE *gilstate)
126 {
127         py_call_level--;
128
129         if (gilstate)
130                 PyGILState_Release(*gilstate);
131
132         if (py_call_level < 0) {
133                 fprintf(stderr, "ERROR: Python context internal state bug. this should not happen!\n");
134         }
135         else if (py_call_level==0) {
136                 /* XXX - Calling classes currently wont store the context :\,
137                  * cant set NULL because of this. but this is very flakey still. */
138 #if 0
139                 BPy_SetContext(NULL);
140                 bpy_import_main_set(NULL);
141 #endif
142
143 #ifdef TIME_PY_RUN
144                 bpy_timer_run_tot += PIL_check_seconds_timer() - bpy_timer_run;
145                 bpy_timer_count++;
146 #endif
147
148         }
149 }
150
151 void BPY_text_free_code(Text *text)
152 {
153         if (text->compiled) {
154                 Py_DECREF((PyObject *)text->compiled);
155                 text->compiled= NULL;
156         }
157 }
158
159 void BPY_modules_update(bContext *C)
160 {
161 #if 0 // slow, this runs all the time poll, draw etc 100's of time a sec.
162         PyObject *mod= PyImport_ImportModuleLevel("bpy", NULL, NULL, NULL, 0);
163         PyModule_AddObject(mod, "data", BPY_rna_module());
164         PyModule_AddObject(mod, "types", BPY_rna_types()); // atm this does not need updating
165 #endif
166
167         /* refreshes the main struct */
168         BPY_update_rna_module();
169         bpy_context_module->ptr.data= (void *)C;
170 }
171
172 void BPY_context_set(bContext *C)
173 {
174         BPy_SetContext(C);
175 }
176
177 /* defined in AUD_C-API.cpp */
178 extern PyObject *AUD_initPython(void);
179 /* defined in cycles/blender */
180 extern PyObject *CYCLES_initPython(void);
181
182 static struct _inittab bpy_internal_modules[]= {
183         {(char *)"mathutils", PyInit_mathutils},
184 //      {(char *)"mathutils.geometry", PyInit_mathutils_geometry},
185 //      {(char *)"mathutils.noise", PyInit_mathutils_noise},
186         {(char *)"bgl", BPyInit_bgl},
187         {(char *)"blf", BPyInit_blf},
188 #ifdef WITH_AUDASPACE
189         {(char *)"aud", AUD_initPython},
190 #endif
191 #ifdef WITH_CYCLES
192         {(char *)"bcycles", CYCLES_initPython},
193 #endif
194         {(char *)"gpu", GPU_initPython},
195         {NULL, NULL}
196 };
197
198 /* call BPY_context_set first */
199 void BPY_python_start(int argc, const char **argv)
200 {
201 #ifndef WITH_PYTHON_MODULE
202         PyThreadState *py_tstate= NULL;
203
204         /* not essential but nice to set our name */
205         static wchar_t program_path_wchar[FILE_MAX]; /* python holds a reference */
206         BLI_strncpy_wchar_from_utf8(program_path_wchar, BLI_program_path(), sizeof(program_path_wchar) / sizeof(wchar_t));
207         Py_SetProgramName(program_path_wchar);
208
209         /* must run before python initializes */
210         PyImport_ExtendInittab(bpy_internal_modules);
211
212         /* allow to use our own included python */
213         PyC_SetHomePath(BLI_get_folder(BLENDER_SYSTEM_PYTHON, NULL));
214
215         /* without this the sys.stdout may be set to 'ascii'
216          * (it is on my system at least), where printing unicode values will raise
217          * an error, this is highly annoying, another stumbling block for devs,
218          * so use a more relaxed error handler and enforce utf-8 since the rest of
219          * blender is utf-8 too - campbell */
220         BLI_setenv("PYTHONIOENCODING", "utf-8:surrogateescape");
221
222         /* Python 3.2 now looks for '2.xx/python/include/python3.2d/pyconfig.h' to
223          * parse from the 'sysconfig' module which is used by 'site',
224          * so for now disable site. alternatively we could copy the file. */
225         Py_NoSiteFlag= 1;
226
227         Py_Initialize();
228
229         // PySys_SetArgv(argc, argv); // broken in py3, not a huge deal
230         /* sigh, why do python guys not have a char** version anymore? :( */
231         {
232                 int i;
233                 PyObject *py_argv= PyList_New(argc);
234                 for (i=0; i<argc; i++) {
235                         /* should fix bug #20021 - utf path name problems, by replacing
236                          * PyUnicode_FromString, with this one */
237                         PyList_SET_ITEM(py_argv, i, PyC_UnicodeFromByte(argv[i]));
238                 }
239
240                 PySys_SetObject("argv", py_argv);
241                 Py_DECREF(py_argv);
242         }
243         
244         /* Initialize thread support (also acquires lock) */
245         PyEval_InitThreads();
246 #else
247         (void)argc;
248         (void)argv;
249
250         /* must run before python initializes */
251         PyImport_ExtendInittab(bpy_internal_modules);
252 #endif
253
254         bpy_intern_string_init();
255
256         /* bpy.* and lets us import it */
257         BPy_init_modules();
258
259         bpy_import_init(PyEval_GetBuiltins());
260         
261         pyrna_alloc_types();
262
263         BPY_atexit_register(); /* this can init any time */
264
265 #ifndef WITH_PYTHON_MODULE
266         py_tstate= PyGILState_GetThisThreadState();
267         PyEval_ReleaseThread(py_tstate);
268 #endif
269 }
270
271 void BPY_python_end(void)
272 {
273         // fprintf(stderr, "Ending Python!\n");
274
275         PyGILState_Ensure(); /* finalizing, no need to grab the state */
276         
277         // free other python data.
278         pyrna_free_types();
279
280         /* clear all python data from structs */
281
282         bpy_intern_string_exit();
283
284         BPY_atexit_unregister(); /* without this we get recursive calls to WM_exit */
285
286         Py_Finalize();
287         
288 #ifdef TIME_PY_RUN
289         // measure time since py started
290         bpy_timer= PIL_check_seconds_timer() - bpy_timer;
291
292         printf("*bpy stats* - ");
293         printf("tot exec: %d,  ", bpy_timer_count);
294         printf("tot run: %.4fsec,  ", bpy_timer_run_tot);
295         if (bpy_timer_count>0)
296                 printf("average run: %.6fsec,  ", (bpy_timer_run_tot/bpy_timer_count));
297
298         if (bpy_timer>0.0)
299                 printf("tot usage %.4f%%", (bpy_timer_run_tot/bpy_timer)*100.0);
300
301         printf("\n");
302
303         // fprintf(stderr, "Ending Python Done!\n");
304
305 #endif
306
307 }
308
309 static void python_script_error_jump_text(struct Text *text)
310 {
311         int lineno;
312         int offset;
313         python_script_error_jump(text->id.name+2, &lineno, &offset);
314         if (lineno != -1) {
315                 /* select the line with the error */
316                 txt_move_to(text, lineno - 1, INT_MAX, FALSE);
317                 txt_move_to(text, lineno - 1, offset, TRUE);
318         }
319 }
320
321 /* super annoying, undo _PyModule_Clear(), bug [#23871] */
322 #define PYMODULE_CLEAR_WORKAROUND
323
324 #ifdef PYMODULE_CLEAR_WORKAROUND
325 /* bad!, we should never do this, but currently only safe way I could find to keep namespace.
326  * from being cleared. - campbell */
327 typedef struct {
328         PyObject_HEAD
329         PyObject *md_dict;
330         /* ommit other values, we only want the dict. */
331 } PyModuleObject;
332 #endif
333
334 static int python_script_exec(bContext *C, const char *fn, struct Text *text,
335                               struct ReportList *reports, const short do_jump)
336 {
337         PyObject *main_mod= NULL;
338         PyObject *py_dict= NULL, *py_result= NULL;
339         PyGILState_STATE gilstate;
340
341         BLI_assert(fn || text);
342
343         if (fn==NULL && text==NULL) {
344                 return 0;
345         }
346
347         bpy_context_set(C, &gilstate);
348
349         PyC_MainModule_Backup(&main_mod);
350
351         if (text) {
352                 char fn_dummy[FILE_MAXDIR];
353                 bpy_text_filename_get(fn_dummy, sizeof(fn_dummy), text);
354
355                 if (text->compiled == NULL) {   /* if it wasn't already compiled, do it now */
356                         char *buf= txt_to_buf(text);
357
358                         text->compiled= Py_CompileString(buf, fn_dummy, Py_file_input);
359
360                         MEM_freeN(buf);
361
362                         if (PyErr_Occurred()) {
363                                 if (do_jump) {
364                                         python_script_error_jump_text(text);
365                                 }
366                                 BPY_text_free_code(text);
367                         }
368                 }
369
370                 if (text->compiled) {
371                         py_dict= PyC_DefaultNameSpace(fn_dummy);
372                         py_result=  PyEval_EvalCode(text->compiled, py_dict, py_dict);
373                 }
374
375         }
376         else {
377                 FILE *fp= fopen(fn, "r");
378
379                 if (fp) {
380                         py_dict= PyC_DefaultNameSpace(fn);
381
382 #ifdef _WIN32
383                         /* Previously we used PyRun_File to run directly the code on a FILE
384                          * object, but as written in the Python/C API Ref Manual, chapter 2,
385                          * 'FILE structs for different C libraries can be different and
386                          * incompatible'.
387                          * So now we load the script file data to a buffer */
388                         {
389                                 char *pystring;
390
391                                 fclose(fp);
392
393                                 pystring= MEM_mallocN(strlen(fn) + 32, "pystring");
394                                 pystring[0]= '\0';
395                                 sprintf(pystring, "exec(open(r'%s').read())", fn);
396                                 py_result= PyRun_String(pystring, Py_file_input, py_dict, py_dict);
397                                 MEM_freeN(pystring);
398                         }
399 #else
400                         py_result= PyRun_File(fp, fn, Py_file_input, py_dict, py_dict);
401                         fclose(fp);
402 #endif
403                 }
404                 else {
405                         PyErr_Format(PyExc_IOError,
406                                      "Python file \"%s\" could not be opened: %s",
407                                      fn, strerror(errno));
408                         py_result= NULL;
409                 }
410         }
411
412         if (!py_result) {
413                 if (text) {
414                         if (do_jump) {
415                                 python_script_error_jump_text(text);
416                         }
417                 }
418                 BPy_errors_to_report(reports);
419         }
420         else {
421                 Py_DECREF(py_result);
422         }
423
424         if (py_dict) {
425 #ifdef PYMODULE_CLEAR_WORKAROUND
426                 PyModuleObject *mmod= (PyModuleObject *)PyDict_GetItemString(PyThreadState_GET()->interp->modules, "__main__");
427                 PyObject *dict_back= mmod->md_dict;
428                 /* freeing the module will clear the namespace,
429                  * gives problems running classes defined in this namespace being used later. */
430                 mmod->md_dict= NULL;
431                 Py_DECREF(dict_back);
432 #endif
433
434 #undef PYMODULE_CLEAR_WORKAROUND
435         }
436
437         PyC_MainModule_Restore(main_mod);
438
439         bpy_context_clear(C, &gilstate);
440
441         return (py_result != NULL);
442 }
443
444 /* Can run a file or text block */
445 int BPY_filepath_exec(bContext *C, const char *filepath, struct ReportList *reports)
446 {
447         return python_script_exec(C, filepath, NULL, reports, FALSE);
448 }
449
450
451 int BPY_text_exec(bContext *C, struct Text *text, struct ReportList *reports, const short do_jump)
452 {
453         return python_script_exec(C, NULL, text, reports, do_jump);
454 }
455
456 void BPY_DECREF(void *pyob_ptr)
457 {
458         PyGILState_STATE gilstate= PyGILState_Ensure();
459         Py_DECREF((PyObject *)pyob_ptr);
460         PyGILState_Release(gilstate);
461 }
462
463 /* return -1 on error, else 0 */
464 int BPY_button_exec(bContext *C, const char *expr, double *value, const short verbose)
465 {
466         PyGILState_STATE gilstate;
467         PyObject *py_dict, *mod, *retval;
468         int error_ret= 0;
469         PyObject *main_mod= NULL;
470         
471         if (!value || !expr) return -1;
472
473         if (expr[0]=='\0') {
474                 *value= 0.0;
475                 return error_ret;
476         }
477
478         bpy_context_set(C, &gilstate);
479
480         PyC_MainModule_Backup(&main_mod);
481
482         py_dict= PyC_DefaultNameSpace("<blender button>");
483
484         mod= PyImport_ImportModule("math");
485         if (mod) {
486                 PyDict_Merge(py_dict, PyModule_GetDict(mod), 0); /* 0 - dont overwrite existing values */
487                 Py_DECREF(mod);
488         }
489         else { /* highly unlikely but possibly */
490                 PyErr_Print();
491                 PyErr_Clear();
492         }
493         
494         retval= PyRun_String(expr, Py_eval_input, py_dict, py_dict);
495         
496         if (retval == NULL) {
497                 error_ret= -1;
498         }
499         else {
500                 double val;
501
502                 if (PyTuple_Check(retval)) {
503                         /* Users my have typed in 10km, 2m
504                          * add up all values */
505                         int i;
506                         val= 0.0;
507
508                         for (i=0; i<PyTuple_GET_SIZE(retval); i++) {
509                                 val+= PyFloat_AsDouble(PyTuple_GET_ITEM(retval, i));
510                         }
511                 }
512                 else {
513                         val= PyFloat_AsDouble(retval);
514                 }
515                 Py_DECREF(retval);
516                 
517                 if (val==-1 && PyErr_Occurred()) {
518                         error_ret= -1;
519                 }
520                 else if (!finite(val)) {
521                         *value= 0.0;
522                 }
523                 else {
524                         *value= val;
525                 }
526         }
527         
528         if (error_ret) {
529                 if (verbose) {
530                         BPy_errors_to_report(CTX_wm_reports(C));
531                 }
532                 else {
533                         PyErr_Clear();
534                 }
535         }
536
537         PyC_MainModule_Backup(&main_mod);
538         
539         bpy_context_clear(C, &gilstate);
540         
541         return error_ret;
542 }
543
544 int BPY_string_exec(bContext *C, const char *expr)
545 {
546         PyGILState_STATE gilstate;
547         PyObject *main_mod= NULL;
548         PyObject *py_dict, *retval;
549         int error_ret= 0;
550         Main *bmain_back; /* XXX, quick fix for release (Copy Settings crash), needs further investigation */
551
552         if (!expr) return -1;
553
554         if (expr[0]=='\0') {
555                 return error_ret;
556         }
557
558         bpy_context_set(C, &gilstate);
559
560         PyC_MainModule_Backup(&main_mod);
561
562         py_dict= PyC_DefaultNameSpace("<blender string>");
563
564         bmain_back= bpy_import_main_get();
565         bpy_import_main_set(CTX_data_main(C));
566
567         retval= PyRun_String(expr, Py_eval_input, py_dict, py_dict);
568
569         bpy_import_main_set(bmain_back);
570
571         if (retval == NULL) {
572                 error_ret= -1;
573
574                 BPy_errors_to_report(CTX_wm_reports(C));
575         }
576         else {
577                 Py_DECREF(retval);
578         }
579
580         PyC_MainModule_Restore(main_mod);
581
582         bpy_context_clear(C, &gilstate);
583         
584         return error_ret;
585 }
586
587
588 void BPY_modules_load_user(bContext *C)
589 {
590         PyGILState_STATE gilstate;
591         Main *bmain= CTX_data_main(C);
592         Text *text;
593
594         /* can happen on file load */
595         if (bmain==NULL)
596                 return;
597
598         /* update pointers since this can run from a nested script
599          * on file load */
600         if (py_call_level) {
601                 bpy_context_update(C);
602         }
603
604         bpy_context_set(C, &gilstate);
605
606         for (text=CTX_data_main(C)->text.first; text; text= text->id.next) {
607                 if (text->flags & TXT_ISSCRIPT && BLI_testextensie(text->id.name+2, ".py")) {
608                         if (!(G.f & G_SCRIPT_AUTOEXEC)) {
609                                 printf("scripts disabled for \"%s\", skipping '%s'\n", bmain->name, text->id.name+2);
610                         }
611                         else {
612                                 PyObject *module= bpy_text_import(text);
613
614                                 if (module==NULL) {
615                                         PyErr_Print();
616                                         PyErr_Clear();
617                                 }
618                                 else {
619                                         Py_DECREF(module);
620                                 }
621                         }
622                 }
623         }
624         bpy_context_clear(C, &gilstate);
625 }
626
627 int BPY_context_member_get(bContext *C, const char *member, bContextDataResult *result)
628 {
629         PyObject *pyctx= (PyObject *)CTX_py_dict_get(C);
630         PyObject *item= PyDict_GetItemString(pyctx, member);
631         PointerRNA *ptr= NULL;
632         int done= 0;
633
634         if (item==NULL) {
635                 /* pass */
636         }
637         else if (item==Py_None) {
638                 /* pass */
639         }
640         else if (BPy_StructRNA_Check(item)) {
641                 ptr= &(((BPy_StructRNA *)item)->ptr);
642
643                 //result->ptr= ((BPy_StructRNA *)item)->ptr;
644                 CTX_data_pointer_set(result, ptr->id.data, ptr->type, ptr->data);
645                 done= 1;
646         }
647         else if (PySequence_Check(item)) {
648                 PyObject *seq_fast= PySequence_Fast(item, "bpy_context_get sequence conversion");
649                 if (seq_fast==NULL) {
650                         PyErr_Print();
651                         PyErr_Clear();
652                 }
653                 else {
654                         int len= PySequence_Fast_GET_SIZE(seq_fast);
655                         int i;
656                         for (i= 0; i < len; i++) {
657                                 PyObject *list_item= PySequence_Fast_GET_ITEM(seq_fast, i);
658
659                                 if (BPy_StructRNA_Check(list_item)) {
660                                         /*
661                                         CollectionPointerLink *link= MEM_callocN(sizeof(CollectionPointerLink), "bpy_context_get");
662                                         link->ptr= ((BPy_StructRNA *)item)->ptr;
663                                         BLI_addtail(&result->list, link);
664                                         */
665                                         ptr= &(((BPy_StructRNA *)list_item)->ptr);
666                                         CTX_data_list_add(result, ptr->id.data, ptr->type, ptr->data);
667                                 }
668                                 else {
669                                         printf("List item not a valid type\n");
670                                 }
671
672                         }
673                         Py_DECREF(seq_fast);
674
675                         done= 1;
676                 }
677         }
678
679         if (done==0) {
680                 if (item)       printf("PyContext '%s' not a valid type\n", member);
681                 else            printf("PyContext '%s' not found\n", member);
682         }
683         else {
684                 if (G.f & G_DEBUG) {
685                         printf("PyContext '%s' found\n", member);
686                 }
687         }
688
689         return done;
690 }
691
692
693 #ifdef WITH_PYTHON_MODULE
694 #include "BLI_fileops.h"
695 /* TODO, reloading the module isnt functional at the moment. */
696
697 static void bpy_module_free(void *mod);
698 extern int main_python_enter(int argc, const char **argv);
699 extern void main_python_exit(void);
700 static struct PyModuleDef bpy_proxy_def= {
701         PyModuleDef_HEAD_INIT,
702         "bpy",  /* m_name */
703         NULL,  /* m_doc */
704         0,  /* m_size */
705         NULL,  /* m_methods */
706         NULL,  /* m_reload */
707         NULL,  /* m_traverse */
708         NULL,  /* m_clear */
709         bpy_module_free,  /* m_free */
710 };
711
712 typedef struct {
713         PyObject_HEAD
714         /* Type-specific fields go here. */
715         PyObject *mod;
716 } dealloc_obj;
717
718 /* call once __file__ is set */
719 void bpy_module_delay_init(PyObject *bpy_proxy)
720 {
721         const int argc= 1;
722         const char *argv[2];
723
724         /* updating the module dict below will loose the reference to __file__ */
725         PyObject *filename_obj= PyModule_GetFilenameObject(bpy_proxy);
726
727         const char *filename_rel= _PyUnicode_AsString(filename_obj); /* can be relative */
728         char filename_abs[1024];
729
730         BLI_strncpy(filename_abs, filename_rel, sizeof(filename_abs));
731         BLI_path_cwd(filename_abs);
732
733         argv[0]= filename_abs;
734         argv[1]= NULL;
735         
736         // printf("module found %s\n", argv[0]);
737
738         main_python_enter(argc, argv);
739
740         /* initialized in BPy_init_modules() */
741         PyDict_Update(PyModule_GetDict(bpy_proxy), PyModule_GetDict(bpy_package_py));
742 }
743
744 static void dealloc_obj_dealloc(PyObject *self);
745
746 static PyTypeObject dealloc_obj_Type= {{{0}}};
747
748 /* use our own dealloc so we can free a property if we use one */
749 static void dealloc_obj_dealloc(PyObject *self)
750 {
751         bpy_module_delay_init(((dealloc_obj *)self)->mod);
752
753         /* Note, for subclassed PyObjects we cant just call PyObject_DEL() directly or it will crash */
754         dealloc_obj_Type.tp_free(self);
755 }
756
757 PyMODINIT_FUNC
758 PyInit_bpy(void)
759 {
760         PyObject *bpy_proxy= PyModule_Create(&bpy_proxy_def);
761         
762         /* Problem:
763          * 1) this init function is expected to have a private member defined - 'md_def'
764          *    but this is only set for C defined modules (not py packages)
765          *    so we cant return 'bpy_package_py' as is.
766          *
767          * 2) there is a 'bpy' C module for python to load which is basically all of blender,
768          *    and there is scripts/bpy/__init__.py, 
769          *    we may end up having to rename this module so there is no naming conflict here eg:
770          *    'from blender import bpy'
771          *
772          * 3) we dont know the filename at this point, workaround by assigning a dummy value
773          *    which calls back when its freed so the real loading can take place.
774          */
775
776         /* assign an object which is freed after __file__ is assigned */
777         dealloc_obj *dob;
778         
779         /* assign dummy type */
780         dealloc_obj_Type.tp_name= "dealloc_obj";
781         dealloc_obj_Type.tp_basicsize= sizeof(dealloc_obj);
782         dealloc_obj_Type.tp_dealloc= dealloc_obj_dealloc;
783         dealloc_obj_Type.tp_flags= Py_TPFLAGS_DEFAULT;
784         
785         if (PyType_Ready(&dealloc_obj_Type) < 0)
786                 return NULL;
787
788         dob= (dealloc_obj *) dealloc_obj_Type.tp_alloc(&dealloc_obj_Type, 0);
789         dob->mod= bpy_proxy; /* borrow */
790         PyModule_AddObject(bpy_proxy, "__file__", (PyObject *)dob); /* borrow */
791
792         return bpy_proxy;
793 }
794
795 static void bpy_module_free(void *UNUSED(mod))
796 {
797         main_python_exit();
798 }
799
800 #endif