fix for invalid axis being allowed on import/export

1 files changed, 12 insertions, 1 deletions

diff --git a/io_scene_x3d/__init__.py b/io_scene_x3d/__init__.py
index c49a86a8..1d0841e0 100644
--- a/io_scene_x3d/__init__.py
+++ b/io_scene_x3d/__init__.py
@@ -41,7 +41,12 @@ if "bpy" in locals():
 
 import bpy
 from bpy.props import StringProperty, BoolProperty, EnumProperty
-from bpy_extras.io_utils import ImportHelper, ExportHelper, axis_conversion, path_reference_mode
+from bpy_extras.io_utils import (ImportHelper,
+                                 ExportHelper,
+                                 axis_conversion,
+                                 axis_conversion_ensure,
+                                 path_reference_mode,
+                                 )
 
 
 class ImportX3D(bpy.types.Operator, ImportHelper):
@@ -76,6 +81,9 @@ class ImportX3D(bpy.types.Operator, ImportHelper):
             default='Y',
             )
 
+    def check(self, context):
+        return axis_conversion_ensure(self, "axis_forward", "axis_up")
+
     def execute(self, context):
         from . import import_x3d
 
@@ -128,6 +136,9 @@ class ExportX3D(bpy.types.Operator, ExportHelper):
 
     path_mode = path_reference_mode
 
+    def check(self, context):
+        return axis_conversion_ensure(self, "axis_forward", "axis_up")
+
     def execute(self, context):
         from . import export_x3d