Check for babeltrace python package on installation

diff --git a/linuxautomaton/setup.py b/linuxautomaton/setup.py
index 488ee24d5339451fc6a95e2a4235c0a16cbe28d9..5b8ae76c6e10fee671a193406073ae08e53db85c 100755 (executable)
--- a/linuxautomaton/setup.py
+++ b/linuxautomaton/setup.py
@@ -32,6 +32,15 @@ if v.major < 3:
     sys.stderr.write('Sorry, linuxautomaton needs Python 3\n')
     sys.exit(1)
 
+# TODO: find a less hackish way to do this.
+if 'install' in sys.argv:
+    try:
+        import babeltrace
+    except ImportError:
+        sys.stderr.write('lttnganalysescli needs the babeltrace package.\n \
+        See https://www.efficios.com/babeltrace for more info.\n')
+        sys.exit(1)
+
 packages = [
     'linuxautomaton',
 ]

diff --git a/lttnganalysescli/setup.py b/lttnganalysescli/setup.py
index 5994b02b2321ccd32cc5585bf09953a70c47d319..a30e5f2c2afcc9bf29dccb0e52832eb4029e23e4 100755 (executable)
--- a/lttnganalysescli/setup.py
+++ b/lttnganalysescli/setup.py
@@ -32,6 +32,15 @@ if v.major < 3:
     sys.stderr.write('Sorry, lttnganalysescli needs Python 3\n')
     sys.exit(1)
 
+# TODO: find a less hackish way to do this.
+if 'install' in sys.argv:
+    try:
+        import babeltrace
+    except ImportError:
+        sys.stderr.write('lttnganalysescli needs the babeltrace package.\n \
+        See https://www.efficios.com/babeltrace for more info.\n')
+        sys.exit(1)
+
 packages = [
     'lttnganalysescli',
     'ascii_graph',