gather bitmask dependency tree

2 files changed, 38 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index e7173c80..3ef61df9 100644
--- a/Makefile
+++ b/Makefile
@@ -119,5 +119,8 @@ install_wheel:
 	# if it's the first time, you'll need to get_wheels first
 	pip install --pre --use-wheel --no-index --find-links=../wheelhouse -r pkg/requirements.pip
 
+gather_deps:
+	pipdeptree | pkg/scripts/filter-bitmask-deps
+
 clean :
 	$(RM) $(COMPILED_UI) $(COMPILED_RESOURCES) $(COMPILED_UI:.py=.pyc) $(COMPILED_RESOURCES:.py=.pyc)
diff --git a/pkg/scripts/filter-bitmask-deps b/pkg/scripts/filter-bitmask-deps
new file mode 100755
index 00000000..9808d394
--- /dev/null
+++ b/pkg/scripts/filter-bitmask-deps
@@ -0,0 +1,35 @@
+#!/usr/bin/env python
+
+"""
+Filter bitmask dependencies.
+
+Usage: pipdeptree | filter-bitmask-deps
+"""
+import fileinput
+
+TARGET = "leap.bitmask"
+
+
+def get_bitmask_deps(dep_lines):
+    res = []
+    begin = False
+    for dep in dep_lines:
+        if dep.startswith(TARGET):
+            begin = True
+            res.append(dep)
+            continue
+        elif dep.startswith(' ') and begin:
+            res.append(dep)
+            continue
+        if begin:
+            return res
+
+
+if __name__ == "__main__":
+    lines = []
+    for line in fileinput.input():
+        lines.append(line)
+
+    bitmask_deps = get_bitmask_deps(lines)
+    for line in bitmask_deps:
+        print line[:-1]