Jython compatibility

1 files changed, 10 insertions, 6 deletions

diff --git a/cyaml.py b/cyaml.py
index 276e029..bf4e639 100644
--- a/cyaml.py
+++ b/cyaml.py
@@ -5,12 +5,16 @@ __all__ = ['CBaseLoader', 'CSafeLoader', 'CLoader',
 
 from _ruamel_yaml import CParser, CEmitter
 
-from .constructor import *
-
-from .serializer import *
-from .representer import *
-
-from .resolver import *
+try:
+    from .constructor import *
+    from .serializer import *
+    from .representer import *
+    from .resolver import *
+except (ImportError, ValueError):  # for Jython
+    from ruamel.yaml.constructor import *
+    from ruamel.yaml.serializer import *
+    from ruamel.yaml.representer import *
+    from ruamel.yaml.resolver import *
 
 
 class CBaseLoader(CParser, BaseConstructor, BaseResolver):