blob: 081327cdba5478d5ec10f9d284651411b4883d74 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# Copyright (C) 2024 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
import shutil
# Local modules
import defaults
import vyos
def check_build_config():
if not os.path.exists(defaults.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 shutil.which(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 format_missing_dependencies(self):
msg = "E: There are missing system dependencies!\n"
if self.__missing['packages']:
msg += "E: Missing packages: " + " ".join(self.__missing['packages'])
if self.__missing['binaries']:
msg += "E: Missing binaries: " + " ".join(self.__missing['binaries'])
return msg
def check_system_dependencies(deps):
checker = DependencyChecker(deps)
missing = checker.get_missing_dependencies()
if missing:
raise OSError(checker.format_missing_dependencies())
else:
pass
def cmd(command):
res = vyos.utils.process.call(command, shell=True)
if res > 0:
raise OSError(f"Command '{command}' failed")
def rc_cmd(command):
code, out = vyos.utils.process.rc_cmd(command, shell=True)
if code > 0:
raise OSError(f"Command '{command}' failed")
else:
return out
|
