patch [#33331] Time To Start Moving To Stdbool
[blender.git] / source / tests / bl_load_addons.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 imp
28
29 def disable_addons():
30     # first disable all
31     addons = bpy.context.user_preferences.addons
32     for mod_name in list(addons.keys()):
33         addon_utils.disable(mod_name)
34     assert(bool(addons) is False)
35
36
37 def test_load_addons():
38     modules = addon_utils.modules({})
39     modules.sort(key=lambda mod: mod.__name__)
40
41     disable_addons()
42
43     addons = bpy.context.user_preferences.addons
44
45     addons_fail = []
46
47     for mod in modules:
48         mod_name = mod.__name__
49         print("\tenabling:", mod_name)
50         addon_utils.enable(mod_name)
51         if mod_name not in addons:
52             addons_fail.append(mod_name)
53     
54     if addons_fail:
55         print("addons failed to load (%d):" % len(addons_fail))
56         for mod_name in addons_fail:
57             print("    %s" % mod_name)
58     else:
59         print("addons all loaded without errors!")
60     print("")
61
62
63 def reload_addons(do_reload=True, do_reverse=True):
64     modules = addon_utils.modules({})
65     modules.sort(key=lambda mod: mod.__name__)
66     addons = bpy.context.user_preferences.addons
67
68     disable_addons()
69
70     # Run twice each time.
71     for i in (0, 1):
72         for mod in modules:
73             mod_name = mod.__name__
74             print("\tenabling:", mod_name)
75             addon_utils.enable(mod_name)
76             assert(mod_name in addons)
77
78         for mod in addon_utils.modules({}):
79             mod_name = mod.__name__
80             print("\tdisabling:", mod_name)
81             addon_utils.disable(mod_name)
82             assert(not (mod_name in addons))
83
84             # now test reloading
85             if do_reload:
86                 imp.reload(sys.modules[mod_name])
87
88             if do_reverse:
89                 # in case order matters when it shouldnt
90                 modules.reverse()
91
92
93 def main():
94     # first load addons, print a list of all addons that fail
95     test_load_addons()
96     
97     reload_addons(do_reload=False, do_reverse=False)
98     reload_addons(do_reload=False, do_reverse=True)
99     reload_addons(do_reload=True, do_reverse=True)
100
101
102 if __name__ == "__main__":
103
104     # So a python error exits(1)
105     try:
106         main()
107     except:
108         import traceback
109         traceback.print_exc()
110         sys.exit(1)