Add import_nuke function

florianeinfalt · Feb 28, 2017 · e6e2b32 · e6e2b32
1 parent 48a1828
commit e6e2b32
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/nukecontexts/__init__.py b/nukecontexts/__init__.py
@@ -5,6 +5,18 @@
 import logging
 import platform
 
+
+def import_nuke():
+    try:
+        import nuke
+        return nuke
+    except ImportError as e:
+        try:
+            os.environ['NON_PRODUCTION_CONTEXT']
+        except KeyError:
+            raise e
+
+
 TESTING = False
 try:
     TESTING = os.environ['NON_PRODUCTION_CONTEXT']
@@ -21,7 +33,7 @@
     match = re.search(application, sys.executable)
     if not match:
         raise RuntimeError('Import nukecontexts from within Nuke')
-    import nuke
+    nuke = import_nuke()
 
 __version__ = '0.1.4'
 __all__ = ['ctx']

diff --git a/nukecontexts/ctx.py b/nukecontexts/ctx.py
@@ -3,8 +3,9 @@
 
 from tqdm import tqdm
 
-from nukecontexts import logger
-from nukecontexts import sentry
+from nukecontexts import import_nuke, logger, sentry
+
+nuke = import_nuke()
 
 
 class NukeContextError(ValueError):