vrchat-project-template/Assets/avatar-example/mesh/blend~/fbx_export.py

import os
from datetime import datetime

import bpy
from bpy.props import StringProperty, BoolProperty, PointerProperty, IntProperty
from bpy.types import UIList, Operator, Panel, PropertyGroup, Context

from io_scene_fbx import ExportFBX


def message_box(message: str, title: str = "Message", icon: str = 'INFO') -> None:
    """Shows modal window with a message

    :param message: Message text
    :param title: Window title
    :param icon: Window icon
    """
    def draw(self, context):
        col = self.layout.column(align=True)
        for line in message.strip().split("\n"):
            col.label(line)
    bpy.context.window_manager.popup_menu(draw, title=title, icon=icon)


def name_as_path(name: str) -> str:
    """Converts and sanitizes an arbitrary string to a canonical path with OS-specific separator

    :param name: Some name
    :returns: Canonical path
    """
    return os.path.join(*[bpy.path.clean_name(n) for n in name.replace("\\", "/").split("/") if n]) + ".fbx"


def fbx_export(context: Context) -> None:
    """
    Export all objects Groups to ASCII FBX
    """
    print("[%s] Batch FBX export begin" % datetime.now())

    rootpath = os.path.normpath(bpy.path.abspath(context.scene.fbx_export.export_dir))

    for group in [g for g in bpy.data.groups if g.fbx_exportable]:
        filepath = os.path.join(rootpath, name_as_path(group.name))
        basepath = os.path.dirname(filepath)

        if not os.path.exists(basepath):
            os.makedirs(basepath)

        override = context.copy()
        override.update(
            {
                "selected_objects": group.objects
            }
        )

        bpy.ops.export_scene.fbx(
            override,
            version='ASCII6100',
            use_selection=True,
            global_scale=1.0,
            axis_forward='-Z',
            axis_up='Y',
            object_types=context.scene.fbx_export.object_types,
            use_mesh_modifiers=True,
            mesh_smooth_type='OFF',
            use_mesh_edges=False,
            use_tspace=False,
            use_armature_deform_only=False,
            add_leaf_bones=False,
            primary_bone_axis='Y',
            secondary_bone_axis='X',
            armature_nodetype='NULL',
            use_anim=True,
            use_default_take=False,
            path_mode='RELATIVE',
            batch_mode='OFF',
            use_batch_own_dir=False,
            filepath=filepath,
        )

    print("\n[%s] Batch FBX export end." % datetime.now())


class FBXExportProps(PropertyGroup):
    export_dir = StringProperty(
        name="FBX Export Dir",
        description="FBX Export Directory",
        subtype="DIR_PATH",
        default="//../fbx/",
    )
    on_save = BoolProperty(
        name="Export on save",
        description="Export FBX on Blender file save",
        default=False
    )
    object_types = ExportFBX.object_types
    group_index = IntProperty(default=0)


class FBX_OT_ManualExport(Operator):
    bl_idname = "fbx.manual_export"
    bl_label = "Manual Export"
    bl_description = "Export FBX manualy"

    def execute(self, context):
        try:
            fbx_export(context)
        except Exception as e:
            self.report({'ERROR'}, str(e))
            return {'CANCELLED'}
        return {'FINISHED'}


class FBX_UL_ExportItems(UIList):
    def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
        row = layout.row(align=True)
        row.prop(item, "fbx_exportable", icon='CHECKBOX_HLT' if item.fbx_exportable else 'CHECKBOX_DEHLT', text="", emboss=False)
        row.prop(item, "name", text="", emboss=False, icon='GROUP')


class SCENE_PT_fbx_autoexport(Panel):
    """Creates a Panel in the scene context of the properties editor"""
    bl_label = "FBX Auto-Export"
    bl_idname = "SCENE_PT_fbx_autoexport"
    bl_space_type = 'PROPERTIES'
    bl_region_type = 'WINDOW'
    bl_context = "scene"

    def draw(self, context):
        layout = self.layout
        scene = context.scene

        col = layout.column()

        sub = col.column(align=True)
        sub.prop(scene.fbx_export, "on_save", icon='LOAD_FACTORY')
        sub.operator(FBX_OT_ManualExport.bl_idname, icon="PASTEDOWN")
    
        col.separator()

        split = col.split()
        sub = split.column()
        sub.label("Filter groups:")
        sub.template_list("FBX_UL_ExportItems", "", bpy.data, "groups", scene.fbx_export, "group_index", rows=5)
        sub = split.column()
        sub.label("Filter object types:")
        sub.prop(scene.fbx_export, "object_types")

        col.separator()

        col.prop(scene.fbx_export, "export_dir")


def on_save(*args, **kwargs) -> None:
    """
    Runs when Blender file is being saved
    """
    if not bpy.context.scene.fbx_export.on_save:
        return

    try:
        fbx_export(bpy.context)
    except Exception as e:
        message_box(str(e), "Export error", 'ERROR')


classes = [
    FBXExportProps,
    FBX_OT_ManualExport,
    FBX_UL_ExportItems,
    SCENE_PT_fbx_autoexport,
]


def register() -> None:
    for cls in classes:
        bpy.utils.register_class(cls)
    bpy.types.Scene.fbx_export = PointerProperty(type=FBXExportProps)
    bpy.types.Group.fbx_exportable = BoolProperty(
        name="Export Group",
        description="Mark group as FBX exportable",
        default=False
    )
    bpy.app.handlers.save_pre.append(on_save)


def unregister() -> None:
    bpy.app.handlers.save_pre.remove(on_save)
    del bpy.types.Group.fbx_exportable
    del bpy.types.Scene.fbx_export
    for cls in reversed(classes):
        bpy.utils.unregister_class(cls)


register()