Bugfix [#32410] Create New Group (CTRL+G) tried to activate even when nothing is
[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
30 def source_list(path, filename_check=None):
31     from os.path import join
32     for dirpath, dirnames, filenames in os.walk(path):
33         # skip '.svn'
34         if dirpath.startswith("."):
35             continue
36
37         for filename in filenames:
38             filepath = join(dirpath, filename)
39             if filename_check is None or filename_check(filepath):
40                 yield filepath
41
42
43 def load_addons():
44     modules = addon_utils.modules({})
45     modules.sort(key=lambda mod: mod.__name__)
46     addons = bpy.context.user_preferences.addons
47
48     # first disable all
49     for mod_name in list(addons.keys()):
50         addon_utils.disable(mod_name)
51
52     assert(bool(addons) == False)
53
54     for mod in modules:
55         mod_name = mod.__name__
56         addon_utils.enable(mod_name)
57         assert(mod_name in addons)
58
59
60 def load_modules():
61     modules = []
62     module_paths = []
63
64     # paths blender stores scripts in.
65     paths = bpy.utils.script_paths()
66
67     print("Paths:")
68     for script_path in paths:
69         print("\t'%s'" % script_path)
70
71     #
72     # find all sys.path we added
73     for script_path in paths:
74         for mod_dir in sys.path:
75             if mod_dir.startswith(script_path):
76                 module_paths.append(mod_dir)
77
78     #
79     # collect modules from our paths.
80     module_names = set()
81     for mod_dir in module_paths:
82         # print("mod_dir", mod_dir)
83         for mod, mod_full in bpy.path.module_names(mod_dir):
84             if mod in module_names:
85                 raise Exception("Module found twice %r" % mod)
86
87             modules.append(__import__(mod))
88
89             module_names.add(mod)
90     del module_names
91
92     #
93     # now submodules
94     for m in modules:
95         filepath = m.__file__
96         if os.path.basename(filepath).startswith("__init__."):
97             mod_dir = os.path.dirname(filepath)
98             for submod, submod_full in bpy.path.module_names(mod_dir):
99                 # fromlist is ignored, ugh.
100                 mod_name_full = m.__name__ + "." + submod
101                 __import__(mod_name_full)
102                 mod_imp = sys.modules[mod_name_full]
103
104                 # check we load what we ask for.
105                 assert(os.path.samefile(mod_imp.__file__, submod_full))
106
107                 modules.append(mod_imp)
108
109     #
110     # check which filepaths we didnt load
111     source_files = []
112     for mod_dir in module_paths:
113         source_files.extend(source_list(mod_dir, filename_check=lambda f: f.endswith(".py")))
114
115     source_files = list(set(source_files))
116     source_files.sort()
117
118     #
119     # remove loaded files
120     loaded_files = list({m.__file__ for m in modules})
121     loaded_files.sort()
122
123     for f in loaded_files:
124         source_files.remove(f)
125
126     #
127     # test we tested all files except for presets and templates
128     ignore_paths = [
129         os.sep + "presets" + os.sep,
130         os.sep + "templates" + os.sep,
131     ]
132
133     for f in source_files:
134         ok = False
135         for ignore in ignore_paths:
136             if ignore in f:
137                 ok = True
138         if not ok:
139             raise Exception("Source file %r not loaded in test" % f)
140
141     print("loaded %d modules" % len(loaded_files))
142
143
144 def main():
145     load_addons()
146     load_modules()
147
148 if __name__ == "__main__":
149     # So a python error exits(1)
150     try:
151         main()
152     except:
153         import traceback
154         traceback.print_exc()
155         sys.exit(1)