CentOS support.

modoboa_installer/package.py

@@ -0,0 +1,105 @@
+"""Package management related tools."""
+
+import re
+
+from . import utils
+
+
+class Package(object):
+    """Base classe."""
+
+    def __init__(self, dist_name):
+        """Constructor."""
+        self.dist_name = dist_name
+
+    def preconfigure(self, name, question, qtype, answer):
+        """Empty method."""
+        pass
+
+
+class DEBPackage(Package):
+    """DEB based operations."""
+
+    FORMAT = "deb"
+
+    def __init__(self, dist_name):
+        super(DEBPackage, self).__init__(dist_name)
+        self.index_updated = False
+
+    def update(self):
+        """Update local cache."""
+        if self.index_updated:
+            return
+        utils.exec_cmd("apt-get update --quiet")
+        self.index_updated = True
+
+    def preconfigure(self, name, question, qtype, answer):
+        """Pre-configure a package before installation."""
+        line = "{0} {0}/{1} {2} {3}".format(name, question, qtype, answer)
+        utils.exec_cmd("echo '{}' | debconf-set-selections".format(line))
+
+    def install(self, name):
+        """Install a package."""
+        self.update()
+        utils.exec_cmd("apt-get install --quiet --assume-yes {}".format(name))
+
+    def install_many(self, names):
+        """Install many packages."""
+        self.update()
+        utils.exec_cmd("apt-get install --quiet --assume-yes {}".format(
+            " ".join(names)))
+
+    def get_installed_version(self, name):
+        """Get installed package version."""
+        code, output = utils.exec_cmd(
+            "dpkg -s {} | grep Version".format(name), capture_output=True)
+        match = re.match(r"Version: \d:(.+)-\d", output.decode())
+        if match:
+            return match.group(1)
+        return None
+
+
+class RPMPackage(Package):
+    """RPM based operations."""
+
+    FORMAT = "rpm"
+
+    def __init__(self, dist_name):
+        """Initialize backend."""
+        super(RPMPackage, self).__init__(dist_name)
+        if "centos" in dist_name:
+            self.install("epel-release")
+
+    def install(self, name):
+        """Install a package."""
+        utils.exec_cmd("yum install -y --quiet {}".format(name))
+
+    def install_many(self, names):
+        """Install many packages."""
+        utils.exec_cmd("yum install -y --quiet {}".format(" ".join(names)))
+
+    def get_installed_version(self, name):
+        """Get installed package version."""
+        code, output = utils.exec_cmd(
+            "rpm -qi {} | grep Version".format(name), capture_output=True)
+        match = re.match(r"Version\s+: (.+)", output.decode())
+        if match:
+            return match.group(1)
+        return None
+
+
+def get_backend():
+    """Return the appropriate package backend."""
+    distname = utils.dist_name()
+    backend = None
+    if distname in ["debian", "ubuntu"]:
+        backend = DEBPackage
+    elif "centos" in distname:
+        backend = RPMPackage
+    else:
+        raise NotImplementedError(
+            "Sorry, this distribution is not supported yet.")
+    return backend(distname)
+
+
+backend = get_backend()