From af2bd9bde61faa30dc269d0322c10f825a870593 Mon Sep 17 00:00:00 2001
From: Antoine Nguyen <tonio@ngyn.org>
Date: Fri, 28 Jul 2017 16:26:40 +0200
Subject: [PATCH] Added missing db driver to automx env.

fix #147
---
 modoboa_installer/scripts/automx.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modoboa_installer/scripts/automx.py b/modoboa_installer/scripts/automx.py
index e278645..89e67db 100644
--- a/modoboa_installer/scripts/automx.py
+++ b/modoboa_installer/scripts/automx.py
@@ -58,6 +58,10 @@ class Automx(base.Installer):
             "future", "lxml", "ipaddress", "sqlalchemy", "python-memcached",
             "python-dateutil", "configparser"
         ]
+        if self.dbengine == "postgres":
+            packages.append("psycopg2")
+        else:
+            packages.append("MYSQL-Python")
         python.install_packages(packages, self.venv_path, sudo_user=self.user)
         target = "{}/master.zip".format(self.home_dir)
         if os.path.exists(target):