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)