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                 # incase 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)