__init__.py

bl_info = {
    "name": "MACHIN3tools(BG2CSE汉化版)",
    "author": "MACHIN3",
    "version": (0, 3, 14),
    "blender": (2, 81, 0),
    "location": "",
    "description": "简化 Blender 2.80及以上版本",
    "warning": "",
    "wiki_url": "https://machin3.io/MACHIN3tools/docs",
    "category": "Mesh"}


def reload_modules(name):
    """
    This makes sure all modules are reloaded from new files, when the addon is removed and a new version is installed in the same session,
    or when Blender's 'Reload Scripts' operator is run manually.
    It's important, that utils modules are reloaded first, as operators and menus import from them
    """

    import os
    import importlib

    # first fetch and reload all utils modules
    utils_modules = sorted([name[:-3] for name in os.listdir(os.path.join(__path__[0], "utils")) if name.endswith('.py')])

    for module in utils_modules:
        impline = "from . utils import %s" % (module)

        print("reloading %s" % (".".join([name] + ['utils'] + [module])))

        exec(impline)
        importlib.reload(eval(module))

    # then update the classes and keys dicts
    from . import registration
    importlib.reload(registration)

    # and based on that, reload the modules containing operator and menu classes
    modules = []

    for label in registration.classes:
        entries = registration.classes[label]
        for entry in entries:
            path = entry[0].split('.')
            module = path.pop(-1)

            if (path, module) not in modules:
                modules.append((path, module))

    for path, module in modules:
        if path:
            impline = "from . %s import %s" % (".".join(path), module)
        else:
            impline = "from . import %s" % (module)

        print("reloading %s" % (".".join([name] + path + [module])))

        exec(impline)
        importlib.reload(eval(module))


if 'bpy' in locals():
    reload_modules(bl_info['name'])

import bpy
from bpy.props import PointerProperty
from . properties import M3SceneProperties
from . utils.registration import get_core, get_tools, get_pie_menus, get_menus
from . utils.registration import register_classes, unregister_classes, register_keymaps, unregister_keymaps, register_icons, unregister_icons, add_object_context_menu, remove_object_context_menu
from . utils.registration import add_object_buttons
from . handlers import update_object_axes_drawing


# TODO: support translation, see https://blendermarket.com/inbox/conversations/20371


def register():
    global classes, keymaps, icons

    # CORE

    core_classes = register_classes(get_core())


    # PROPERTIES

    bpy.types.Scene.M3 = PointerProperty(type=M3SceneProperties)

    # TOOLS, PIE MENUS, KEYMAPS, MENUS

    tool_classlists, tool_keylists, tool_count = get_tools()
    pie_classlists, pie_keylists, pie_count = get_pie_menus()
    menu_classlists, menu_keylists, menu_count = get_menus()

    classes = register_classes(tool_classlists + pie_classlists + menu_classlists) + core_classes
    keymaps = register_keymaps(tool_keylists + pie_keylists + menu_keylists)

    add_object_context_menu()

    bpy.types.VIEW3D_MT_mesh_add.prepend(add_object_buttons)


    # ICONS

    icons = register_icons()


    # HANDLERS

    bpy.app.handlers.undo_pre.append(update_object_axes_drawing)
    bpy.app.handlers.redo_pre.append(update_object_axes_drawing)
    bpy.app.handlers.load_pre.append(update_object_axes_drawing)


    # REGISTRATION OUTPUT

    print("Registered %s %s with %d %s, %d pie %s and %s context %s" % (bl_info["name"], ".".join([str(i) for i in bl_info['version']]), tool_count, "tool" if tool_count == 1 else "tools", pie_count, "menu" if pie_count == 1 else "menus", menu_count, "menu" if menu_count == 1 else "menus"))


def unregister():
    global classes, keymaps, icons

    # HANDLERS

    bpy.app.handlers.undo_pre.remove(update_object_axes_drawing)
    bpy.app.handlers.redo_pre.remove(update_object_axes_drawing)
    bpy.app.handlers.load_pre.remove(update_object_axes_drawing)


    # TOOLS, PIE MENUS, KEYMAPS, MENUS

    bpy.types.VIEW3D_MT_mesh_add.remove(add_object_buttons)

    remove_object_context_menu()

    unregister_keymaps(keymaps)
    unregister_classes(classes)


    # PROPERTIES

    del bpy.types.Scene.M3


    # ICONS

    unregister_icons(icons)

    print("Unregistered %s %s." % (bl_info["name"], ".".join([str(i) for i in bl_info['version']])))