Merge pull request #214 from chescales/feature/configure-db-extensions

Added the possibility to configure `db_extensions` in the `.aldryn` file

Author: chescales

divio_cli/localdev/main.py

@@ -162,7 +162,6 @@ def del_rw(action, name, exc):
 
 
 class DatabaseImportBase(object):
-    database_extensions = ['hstore', 'postgis']
     restore_commands = {
         'sql': 'psql -U postgres db < {}',
         'binary': (
@@ -183,6 +182,7 @@ def __init__(self, *args, **kwargs):
         self.website_id = utils.get_aldryn_project_settings(self.path)['id']
         self.website_slug = utils.get_aldryn_project_settings(self.path)['slug']
         self.docker_compose = utils.get_docker_compose_cmd(self.path)
+        self.database_extensions = self.get_active_db_extensions()
         docker_compose_config = utils.DockerComposeConfig(self.docker_compose)
         if not docker_compose_config.has_service('db'):
             click.secho('No service "db" found in local project', fg='red')
@@ -202,6 +202,21 @@ def run(self):
         self.restore_db()
         self.finish()
 
+    def get_active_db_extensions(self):
+        project_settings = utils.get_aldryn_project_settings(self.path)
+        default_db_extensions = ['hstore', 'postgis']
+
+        if 'db_extensions' in project_settings:
+            if not isinstance(project_settings['db_extensions'], list):
+                raise click.ClickException(
+                    '{} file contains invalid "db_extensions" value. '
+                    'It should contain a list of extensions, for instance: {}'
+                    .format(settings.ALDRYN_DOT_FILE, default_db_extensions)
+                )
+            return project_settings['db_extensions']
+        else:
+            return default_db_extensions
+
     def prepare_db_server(self):
         utils.start_database_server(self.docker_compose)