Merging r47662 through r47666 from trunk into soc-2011-tomato
[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
30 def reload_addons(do_reload=True, do_reverse=True):
31     modules = addon_utils.modules({})
32     modules.sort(key=lambda mod: mod.__name__)
33     addons = bpy.context.user_preferences.addons
34
35     # first disable all
36     for mod_name in list(addons.keys()):
37         addon_utils.disable(mod_name)
38
39     assert(bool(addons) == False)
40
41     # Run twice each time.
42     for i in (0, 1):
43         for mod in modules:
44             mod_name = mod.__name__
45             print("\tenabling:", mod_name)
46             addon_utils.enable(mod_name)
47             assert(mod_name in addons)
48
49         for mod in addon_utils.modules({}):
50             mod_name = mod.__name__
51             print("\tdisabling:", mod_name)
52             addon_utils.disable(mod_name)
53             assert(not (mod_name in addons))
54
55             # now test reloading
56             if do_reload:
57                 imp.reload(sys.modules[mod_name])
58
59             if do_reverse:
60                 # in case order matters when it shouldnt
61                 modules.reverse()
62
63
64 def main():
65     reload_addons(do_reload=False, do_reverse=False)
66     reload_addons(do_reload=False, do_reverse=True)
67     reload_addons(do_reload=True, do_reverse=True)
68
69
70 if __name__ == "__main__":
71
72     # So a python error exits(1)
73     try:
74         main()
75     except:
76         import traceback
77         traceback.print_exc()
78         sys.exit(1)