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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
import subprocess
from devbot import command
class FedoraPackageManager:
def install_packages(self, packages):
args = ["yum", "install"]
args.extend(packages)
command.run_with_sudo(args)
def remove_packages(self, packages):
args = ["rpm", "-e"]
args.extend(packages)
command.run_with_sudo(args)
def update(self):
command.run_with_sudo(["yum", "update"])
def find_all(self):
query_format = "--queryformat=[%{NAME} ]"
all = subprocess.check_output(["rpm", "-qa", query_format]).strip()
return all.split(" ")
def find_with_deps(self, packages):
result = []
for package in packages:
if package not in result:
result.append(package)
self._find_deps(package, result)
return result
def _find_deps(self, package, result):
query_format = "--queryformat=[%{REQUIRENAME} ]"
capabilities = subprocess.check_output(["rpm", "-q",
query_format,
package]).strip()
for capability in capabilities.strip().split(" "):
if capability.startswith("rpmlib"):
continue
query_format = "--queryformat=[%{NAME} ]"
deps_packages = subprocess.check_output(["rpm", "-q",
query_format,
"--whatprovides",
capability]).strip()
for dep_package in deps_packages.split(" "):
if dep_package not in result:
result.append(dep_package)
self._find_deps(dep_package, result)
class UbuntuPackageManager:
def install_packages(self, packages):
args = ["apt-get", "install"]
args.extend(packages)
command.run_with_sudo(args)
def remove_packages(self, packages):
raise NotImplementedError
def update(self):
raise NotImplementedError
def find_with_deps(package_names):
raise NotImplementedError
def get_package_manager():
name, version = _get_distro_info()
if name == "fedora":
return FedoraPackageManager()
elif name == "ubuntu":
return UbuntuPackageManager()
def get_system_version():
name, version = _get_distro_info()
if (name == "ubuntu" and version == "12.10") or \
(name == "fedora" and version == "18"):
return "3.6"
else:
return "3.4"
def get_distro_name():
name, version = _get_distro_info()
return name
def _get_distro_info():
distro = "unsupported"
version = "unknown"
# Fedora
try:
fedora_release = open("/etc/fedora-release").read().strip()
if fedora_release == "Fedora release 17 (Beefy Miracle)":
distro = "fedora"
version = "17"
elif fedora_release == "Fedora release 18 (Spherical Cow)":
distro = "fedora"
version = "18"
except IOError:
pass
# Ubuntu
try:
distributor = subprocess.check_output(["lsb_release", "-si"]).strip()
release = subprocess.check_output(["lsb_release", "-sr"]).strip()
if distributor == "Ubuntu" and release == "12.10":
distro = "ubuntu"
version = "12.10"
except OSError:
pass
arch = subprocess.check_output(["uname", "-i"]).strip()
if arch not in ["i386", "i686", "x86_64"]:
distro = "unsupported"
version = "unknown"
return distro, version
|