summaryrefslogtreecommitdiff
path: root/python/examples/kaitai/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/examples/kaitai/__init__.py')
-rw-r--r--python/examples/kaitai/__init__.py15
1 files changed, 14 insertions, 1 deletions
diff --git a/python/examples/kaitai/__init__.py b/python/examples/kaitai/__init__.py
index 90cdbd77..7ac825d2 100644
--- a/python/examples/kaitai/__init__.py
+++ b/python/examples/kaitai/__init__.py
@@ -1 +1,14 @@
-from . import view
+import sys
+
+# invoked from python2 command line? (eg: python -m kaitai ...)
+if sys.version_info[0] == 2 and sys.argv[0:] and sys.argv[0]=='-c':
+ pass
+# invoked from python3 command line? (eg: python -m kaitai ...)
+elif sys.version_info[0] == 3 and sys.argv[0:] and sys.argv[0]=='-m':
+ pass
+# invoked from binja
+else:
+ if sys.version_info[0] == 2:
+ import view
+ else:
+ from . import view