Merge pull request #66 from Yusarina/import_anything_changes

VRM Added, Import Anything chages.

core/importer.py

@@ -40,6 +40,7 @@ import_types: dict[str, typing.Callable[[str, list[dict[str,str]], str], None]]
     "x3d": (lambda directory, files, filepath : bpy.ops.import_scene.x3d(files=files, directory=directory, filepath=filepath)),
     "wrl": (lambda directory, files, filepath : bpy.ops.import_scene.x3d(files=files, directory=directory, filepath=filepath)),
     "vmd": (lambda directory, files, filepath : import_multi_files(directory=directory, files=files, filepath=filepath, method = (lambda directory, filepath: bpy.ops.tuxedo.import_mmd_animation(directory=directory, filepath=filepath)))),
+    "vrm": (lambda directory, files, filepath: bpy.ops.import_scene.vrm(filepath=filepath)),
     "pmx": (lambda directory, files, filepath : import_pmx(filepath)),
     "pmd": (lambda directory, files, filepath : import_pmd(filepath)),
 }

functions/import_anything.py

@@ -3,11 +3,13 @@ from bpy.types import Operator
 from bpy_extras.io_utils import ImportHelper
 from ..core.register import register_wrap
 from ..core.importer import imports, import_types
-from ..core.common import remove_default_objects, open_web_after_delay_multi_threaded
+from ..core.common import remove_default_objects
 from ..functions.translations import t
 import pathlib
 import os
 
+VRM_IMPORTER_URL = "https://github.com/saturday06/VRM_Addon_for_Blender"
+
 @register_wrap
 class AvatarToolKit_OT_ImportAnyModel(Operator, ImportHelper):
     bl_idname = 'avatar_toolkit.import_any_model'
@@ -16,71 +18,71 @@ class AvatarToolKit_OT_ImportAnyModel(Operator, ImportHelper):
     bl_options = {'REGISTER', 'UNDO'}
     files: bpy.props.CollectionProperty(type=bpy.types.OperatorFileListElement, options={'HIDDEN', 'SKIP_SAVE'})
     
-    filter_glob: bpy.props.StringProperty(default = imports, options={'HIDDEN','SKIP_SAVE'})
+    filter_glob: bpy.props.StringProperty(default=imports, options={'HIDDEN', 'SKIP_SAVE'})
     directory: bpy.props.StringProperty(maxlen=1024, subtype='FILE_PATH', options={'HIDDEN', 'SKIP_SAVE'})
 
-    
+    # since I wrote this myself, a bit more efficient than cats. mostly - @989onan
-    #since I wrote this myself, a bit more efficent than cats. mostly - @989onan
     def execute(self, context: bpy.types.Context):
-        file_grouping_dict: dict[str, list[dict[str,str]]] = dict()#group our files so our importers can import them together. in the case of OBJ+MTL and others that need grouped files, this is extremely important.
+        file_grouping_dict: dict[str, list[dict[str, str]]] = dict()  # group our files so our importers can import them together. in the case of OBJ+MTL and others that need grouped files, this is extremely important.
         remove_default_objects()
-        #check if we are importing multiple files
+        # check if we are importing multiple files
-        is_multi = False
+        is_multi = len(self.files) > 0
-        try:
-            for file in self.files:
-                pass
-            is_multi = True
-        except Exception as e:
-            is_multi = False
-            print(e)
-                
-            
-        #put the files together into lists of same importers
-            
-        if(is_multi):
-            for file in self.files:
-                fullpath = os.path.join(self.directory,os.path.basename(file.name))
-                name = pathlib.Path(fullpath).suffix.replace(".","")
-                #this makes sure our imports that should be grouped stay together.
-                #basically the method checks for if the first value has a lambda with the same bytecode as another lambda, then it will use that value's key (ex:"obj"<->"mtl" or "fbx"), keeping same importers together
-                try:
-                    name2 = next(key for key,value in import_types.items() if value.__code__.co_code == import_types[name].__code__.co_code)
-                    print(name +" is the same importer as "+name2+", grouping.")
-                    name = name2
-                except Exception as e:
-                    print("error when trying to find a value of the same value in the kinds of importers. May just be an import type that's a singlet:")
-                    print(e)
-                if name not in file_grouping_dict: file_grouping_dict[name] = []
-                file_grouping_dict[name].append({"name": os.path.basename(file.name)}) #emulate passing a list of files.
 
+        if is_multi:
+            for file in self.files:
+                fullpath = os.path.join(self.directory, os.path.basename(file.name))
+                name = pathlib.Path(fullpath).suffix.replace(".", "")
+                # this makes sure our imports that should be grouped stay together.
+                # basically the method checks for if the first value has a lambda with the same bytecode as another lambda, then it will use that value's key (ex:"obj"<->"mtl" or "fbx"), keeping same importers together
+                if name not in file_grouping_dict:
+                    file_grouping_dict[name] = []
+                file_grouping_dict[name].append({"name": os.path.basename(file.name)})  # emulate passing a list of files.
         else:
-            fullpath: str = os.path.join(os.path.dirname(self.filepath),os.path.basename(self.filepath))
+            fullpath: str = os.path.join(os.path.dirname(self.filepath), os.path.basename(self.filepath))
-            name = pathlib.Path(fullpath).suffix.replace(".","")
+            name = pathlib.Path(fullpath).suffix.replace(".", "")
-            if name not in file_grouping_dict: file_grouping_dict[name] = []
+            if name not in file_grouping_dict:
-            file_grouping_dict[name].append({"name": fullpath}) #emulate passing a list of files.
+                file_grouping_dict[name] = []
-        
+            file_grouping_dict[name].append({"name": fullpath})  # emulate passing a list of files.
-        #import the files together to make sure things like obj import together. This is important
+
-        for file_group_name,files in file_grouping_dict.items():
+        # import the files together to make sure things like obj import together. This is important
+        for file_group_name, files in file_grouping_dict.items():
             try:
-                if(self.directory):
+                # Check for VRM importer availability
-                    print(files)
+                if file_group_name == "vrm" and not hasattr(bpy.ops.import_scene, "vrm"):
-                    import_types[file_group_name](self.directory,files,self.filepath)
+                    bpy.ops.wm.vrm_importer_popup('INVOKE_DEFAULT')
+                    return {'CANCELLED'}
+
+                if self.directory:
+                    import_types[file_group_name](self.directory, files, self.filepath)
                 else:
-                    import_types[file_group_name]("",files,self.filepath) #give an empty directory, works just fine for 90%
+                    import_types[file_group_name]("", files, self.filepath)  # give an empty directory, works just fine for 90%
             except AttributeError as e:
-                print("Warning, you may not have the required importer for extension type \"{extension}\"!".format(extension = file_group_name))
+                if file_group_name == "vrm":
-                
+                    bpy.ops.wm.vrm_importer_popup('INVOKE_DEFAULT')
-                open_web_after_delay_multi_threaded(delay=12, url=t('Importing.importer_search_term').format(extension = file_group_name))
+                else:
-
+                    self.report({'ERROR'}, t('Importing.need_importer').format(extension=file_group_name))
-                self.report({'ERROR'},t('Importing.need_importer').format(extension = file_group_name))
+                print("Importer error:", e)
-
+                return {'CANCELLED'}
-                print("importer error was:")
-                print(e)
 
         self.report({'INFO'}, t('Quick_Access.import_success'))
         return {'FINISHED'}
 
+@register_wrap
+class VRMImporterPopup(Operator):
+    bl_idname = "wm.vrm_importer_popup"
+    bl_label = "VRM Importer Not Installed"
 
+    def execute(self, context):
+        return {'FINISHED'}
+
+    def invoke(self, context, event):
+        return context.window_manager.invoke_props_dialog(self, width=300)
+
+    def draw(self, context):
+        layout = self.layout
+        layout.label(text="VRM importer plugin is not installed.")
+        layout.label(text="Please install it to import VRM files.")
+        layout.operator("wm.url_open", text="Get VRM Importer").url = VRM_IMPORTER_URL
 
 #TODO: This needs to be done with our own MMD importer.
 """