Support for the C Macro system in Python.
[blender.git] / release / scripts / modules / bpy / __init__.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # internal blender C module
20 import _bpy
21 from _bpy import types, props
22
23 data = _bpy.data
24 context = _bpy.context
25
26 # python modules
27 from bpy import utils
28
29 from bpy import ops as _ops_module
30
31 # fake operator module
32 ops = _ops_module.ops_fake_module
33
34 def load_scripts(reload_scripts=False):
35     import os
36     import sys
37     import traceback
38
39     def test_import(module_name):
40         try:
41             return __import__(module_name)
42         except:
43             traceback.print_exc()
44             return None
45
46     for base_path in utils.script_paths():
47         for path_subdir in ("ui", "op", "io"):
48             path = os.path.join(base_path, path_subdir)
49             if os.path.isdir(path):
50                 sys.path.insert(0, path)
51                 for f in sorted(os.listdir(path)):
52                     if f.endswith(".py"):
53                         # python module
54                         mod = test_import(f[0:-3])
55                     elif "." not in f:
56                         # python package
57                         mod = test_import(f)
58                     else:
59                         mod = None
60
61                     if reload_scripts and mod:
62                         print("Reloading:", mod)
63                         reload(mod)
64
65 def _main():
66
67     # a bit nasty but this prevents help() and input() from locking blender
68     # Ideally we could have some way for the console to replace sys.stdin but
69     # python would lock blender while waiting for a return value, not easy :|
70     import sys
71     sys.stdin = None
72
73     if "-d" in sys.argv and False: # Enable this to measure startup speed
74         import cProfile
75         cProfile.run('import bpy; bpy.load_scripts()', 'blender.prof')
76
77         import pstats
78         p = pstats.Stats('blender.prof')
79         p.sort_stats('cumulative').print_stats(100)
80
81     else:
82         load_scripts()
83
84 _main()
85
86