1 files changed, 55 insertions, 0 deletions
diff --git a/scripts/util.py b/scripts/util.py
index 740d4ccc..7cc33364 100644
--- a/scripts/util.py
+++ b/scripts/util.py
@@ -1,5 +1,25 @@
+# Copyright (C) 2015 VyOS maintainers and contributors
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 or later as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# File: util.py
+# Purpose:
+#   Various common functions for use in build scripts.
+
+
 import sys
 import os
+from distutils.spawn import find_executable
 
 import defaults
 
@@ -8,3 +28,38 @@ def check_build_config():
         print("Build config file ({file}) does not exist".format(file=defaults.BUILD_CONFIG))
         print("If you are running this script by hand, you should better not. Run 'make iso' instead.")
         sys.exit(1)
+
+
+class DependencyChecker(object):
+    def __init__(self, spec):
+        missing_packages = self._get_missing_packages(spec['packages'])
+        missing_binaries = self._get_missing_binaries(spec['binaries'])
+        self.__missing = {'packages': missing_packages, 'binaries': missing_binaries}
+
+
+    def _package_installed(self, name):
+        result = os.system("dpkg-query -W --showformat='${{Status}}\n' {name} 2>&1 | grep 'install ok installed' >/dev/null".format(name=name))
+        return True if result == 0 else False
+
+    def _get_missing_packages(self, packages):
+        missing_packages = []
+        for p in packages:
+            if not self._package_installed(p):
+                missing_packages.append(p)
+        return missing_packages
+
+    def _get_missing_binaries(self, binaries):
+        missing_binaries = []
+        for b in binaries:
+            if not find_executable(b):
+                missing_binaries.append(b)
+        return missing_binaries
+
+    def get_missing_dependencies(self):
+        if self.__missing['packages'] or self.__missing['binaries']:
+            return self.__missing
+        return None
+
+    def print_missing_deps(self):
+        print("Missing packages: " + " ".join(self.__missing['packages']))
+        print("Missing binaries: " + " ".join(self.__missing['binaries']))