fix own regression in 2.66 [#34610] Text editor: Syntax highlighting freezes
[blender.git] / source / tests / bl_load_py_modules.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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21 # simple script to enable all addons, and disable
22
23 import bpy
24 import addon_utils
25
26 import sys
27 import os
28
29 BLACKLIST = {
30     "bl_i18n_utils",
31     "cycles",
32     "io_export_dxf",  # TODO, check on why this fails
33     }
34
35
36 def source_list(path, filename_check=None):
37     from os.path import join
38     for dirpath, dirnames, filenames in os.walk(path):
39         # skip '.svn'
40         if dirpath.startswith("."):
41             continue
42
43         for filename in filenames:
44             filepath = join(dirpath, filename)
45             if filename_check is None or filename_check(filepath):
46                 yield filepath
47
48
49 def load_addons():
50     modules = addon_utils.modules({})
51     modules.sort(key=lambda mod: mod.__name__)
52     addons = bpy.context.user_preferences.addons
53
54     # first disable all
55     for mod_name in list(addons.keys()):
56         addon_utils.disable(mod_name)
57
58     assert(bool(addons) is False)
59
60     for mod in modules:
61         mod_name = mod.__name__
62         addon_utils.enable(mod_name)
63         assert(mod_name in addons)
64
65
66 def load_modules():
67     modules = []
68     module_paths = []
69
70     # paths blender stores scripts in.
71     paths = bpy.utils.script_paths()
72
73     print("Paths:")
74     for script_path in paths:
75         print("\t'%s'" % script_path)
76
77     #
78     # find all sys.path we added
79     for script_path in paths:
80         for mod_dir in sys.path:
81             if mod_dir.startswith(script_path):
82                 if mod_dir not in module_paths:
83                     if os.path.exists(mod_dir):
84                         module_paths.append(mod_dir)
85
86     #
87     # collect modules from our paths.
88     module_names = {}
89     for mod_dir in module_paths:
90         # print("mod_dir", mod_dir)
91         for mod, mod_full in bpy.path.module_names(mod_dir):
92             if mod in BLACKLIST:
93                 continue
94             if mod in module_names:
95                 mod_dir_prev, mod_full_prev = module_names[mod]
96                 raise Exception("Module found twice %r.\n    (%r -> %r, %r -> %r)" %
97                                 (mod, mod_dir, mod_full, mod_dir_prev, mod_full_prev))
98
99             modules.append(__import__(mod))
100
101             module_names[mod] = mod_dir, mod_full
102     del module_names
103
104     #
105     # now submodules
106     for m in modules:
107         filepath = m.__file__
108         if os.path.basename(filepath).startswith("__init__."):
109             mod_dir = os.path.dirname(filepath)
110             for submod, submod_full in bpy.path.module_names(mod_dir):
111                 # fromlist is ignored, ugh.
112                 mod_name_full = m.__name__ + "." + submod
113                 __import__(mod_name_full)
114                 mod_imp = sys.modules[mod_name_full]
115
116                 # check we load what we ask for.
117                 assert(os.path.samefile(mod_imp.__file__, submod_full))
118
119                 modules.append(mod_imp)
120
121     #
122     # check which filepaths we didn't load
123     source_files = []
124     for mod_dir in module_paths:
125         source_files.extend(source_list(mod_dir, filename_check=lambda f: f.endswith(".py")))
126
127     source_files = list(set(source_files))
128     source_files.sort()
129
130     #
131     # remove loaded files
132     loaded_files = list({m.__file__ for m in modules})
133     loaded_files.sort()
134
135     for f in loaded_files:
136         source_files.remove(f)
137
138     #
139     # test we tested all files except for presets and templates
140     ignore_paths = [
141         os.sep + "presets" + os.sep,
142         os.sep + "templates" + os.sep,
143     ] + [(os.sep + f + os.sep) for f in BLACKLIST]
144
145     for f in source_files:
146         ok = False
147         for ignore in ignore_paths:
148             if ignore in f:
149                 ok = True
150         if not ok:
151             raise Exception("Source file %r not loaded in test" % f)
152
153     print("loaded %d modules" % len(loaded_files))
154
155
156 def main():
157     load_addons()
158     load_modules()
159
160 if __name__ == "__main__":
161     # So a python error exits(1)
162     try:
163         main()
164     except:
165         import traceback
166         traceback.print_exc()
167         sys.exit(1)