fix for invalid axis being allowed on import/export

1 files changed, 12 insertions, 1 deletions

diff --git a/io_scene_3ds/__init__.py b/io_scene_3ds/__init__.py
index e7934afc..2206c20d 100644
--- a/io_scene_3ds/__init__.py
+++ b/io_scene_3ds/__init__.py
@@ -43,7 +43,12 @@ if "bpy" in locals():
 
 import bpy
 from bpy.props import StringProperty, FloatProperty, BoolProperty, EnumProperty
-from bpy_extras.io_utils import ImportHelper, ExportHelper, axis_conversion
+
+from bpy_extras.io_utils import (ImportHelper,
+                                 ExportHelper,
+                                 axis_conversion,
+                                 axis_conversion_ensure,
+                                 )
 
 
 class Import3DS(bpy.types.Operator, ImportHelper):
@@ -82,6 +87,9 @@ class Import3DS(bpy.types.Operator, ImportHelper):
             default='Z',
             )
 
+    def check(self, context):
+        return axis_conversion_ensure(self, "axis_forward", "axis_up")
+
     def execute(self, context):
         from . import import_3ds
 
@@ -127,6 +135,9 @@ class Export3DS(bpy.types.Operator, ExportHelper):
             default='Z',
             )
 
+    def check(self, context):
+        return axis_conversion_ensure(self, "axis_forward", "axis_up")
+
     def execute(self, context):
         from . import export_3ds