diff options
author | Nutti <nutti.metro@gmail.com> | 2019-01-26 05:22:38 +0300 |
---|---|---|
committer | Nutti <nutti.metro@gmail.com> | 2019-01-26 05:22:38 +0300 |
commit | c034e1968465acb939efc089e63c5c51302947f5 (patch) | |
tree | 84f04a329e082ff4d7f860a926f8848cadb63428 /magic_uv/utils/bl_class_registry.py | |
parent | 2eb519ceca77a4fe2fd5f8d071767db06aa01aa5 (diff) |
Magic UV: Release v6.0
Support Blender 2.8.
Diffstat (limited to 'magic_uv/utils/bl_class_registry.py')
-rw-r--r-- | magic_uv/utils/bl_class_registry.py | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/magic_uv/utils/bl_class_registry.py b/magic_uv/utils/bl_class_registry.py new file mode 100644 index 00000000..81e4b770 --- /dev/null +++ b/magic_uv/utils/bl_class_registry.py @@ -0,0 +1,80 @@ +# <pep8-80 compliant> + +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +__author__ = "Nutti <nutti.metro@gmail.com>" +__status__ = "production" +__version__ = "6.0" +__date__ = "26 Jan 2019" + +import bpy + +from .. import common + + +class BlClassRegistry: + class_list = [] + + def __init__(self, *_, **kwargs): + self.legacy = kwargs.get('legacy', False) + + def __call__(self, cls): + if hasattr(cls, "bl_idname"): + BlClassRegistry.add_class(cls.bl_idname, cls, self.legacy) + else: + bl_idname = "{}{}{}{}".format(cls.bl_space_type, + cls.bl_region_type, + cls.bl_context, cls.bl_label) + BlClassRegistry.add_class(bl_idname, cls, self.legacy) + return cls + + @classmethod + def add_class(cls, bl_idname, op_class, legacy): + for class_ in cls.class_list: + if (class_["bl_idname"] == bl_idname) and \ + (class_["legacy"] == legacy): + raise RuntimeError("{} is already registered" + .format(bl_idname)) + + new_op = { + "bl_idname": bl_idname, + "class": op_class, + "legacy": legacy, + } + cls.class_list.append(new_op) + common.debug_print("{} is registered.".format(bl_idname)) + + @classmethod + def register(cls): + for class_ in cls.class_list: + bpy.utils.register_class(class_["class"]) + common.debug_print("{} is registered to Blender." + .format(class_["bl_idname"])) + + @classmethod + def unregister(cls): + for class_ in cls.class_list: + bpy.utils.unregister_class(class_["class"]) + common.debug_print("{} is unregistered from Blender." + .format(class_["bl_idname"])) + + @classmethod + def cleanup(cls): + cls.class_list = [] + common.debug_print("Cleanup registry.") |