Initial commit.

modoboa_installer/scripts/amavis.py

@@ -0,0 +1,57 @@
+"""Amavis related functions."""
+
+import re
+
+from .. import utils
+
+from . import base
+from . import install
+
+
+class Amavis(base.Installer):
+
+    """Amavis installer."""
+
+    appname = "amavis"
+    packages = ["libdbi-perl", "amavisd-new"]
+    with_db = True
+    config_files = ["05-node_id", "15-content_filter_mode", "50-user"]
+
+    def get_packages(self):
+        """Additional packages."""
+        db_driver = "pg" if self.db_driver == "pgsql" else self.db_driver
+        return self.packages + ["libdbd-{}-perl".format(db_driver)]
+
+    @property
+    def installed_version(self):
+        """Check the installed version."""
+        name = "amavisd-new"
+        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
+
+    def get_sql_schema_path(self):
+        """Return schema path."""
+        version = self.installed_version
+        if version is None:
+            raise utils.FatalError("Amavis is not installed")
+        return self.get_file_path(
+            "amavis_{}_{}.sql".format(self.dbengine, version))
+
+    def setup_database(self):
+        """Additional config."""
+        super(Amavis, self).setup_database()
+        self.backend.grant_access(
+            self.dbname, self.config.get("modoboa", "dbuser"))
+
+    def post_run(self):
+        """Additional tasks."""
+        with open("/etc/mailname", "w") as fp:
+            fp.write("{}\n".format(self.config.get("general", "hostname")))
+        install("spamassassin", self.config)
+        install("clamav", self.config)