django-orchestra-test/orchestra/contrib/webapps/backends/__init__.py

import pkgutil
import textwrap
from django.template import Template, Context
from .. import settings
from orchestra.settings import NEW_SERVERS

class WebAppServiceMixin(object):
    model = 'webapps.WebApp'
    related_models = (
        ('webapps.WebAppOption', 'webapp'),
    )
    directive = None
    doc_settings = (settings,
        ('WEBAPPS_UNDER_CONSTRUCTION_PATH', 'WEBAPPS_MOVE_ON_DELETE_PATH',)
    )
    def check_webapp_dir(self, context):
        self.append(textwrap.dedent("""
            # Create webapp dir
            CREATED=0
            if [[ ! -e %(app_path)s ]]; then
                mkdir -p %(app_path)s
                #chown %(sftpuser)s:%(sftpuser)s %(app_path)s
                CREATED=1
            elif [[ -z $( ls -A %(app_path)s ) ]]; then
                CREATED=1
            fi""") % context
        )

    def create_webapp_dir(self, context):
        self.append(textwrap.dedent("""
            # Create webapp dir
            CREATED=0
            if [[ ! -e %(app_path)s ]]; then
                CREATED=1
                mkdir -p %(app_path)s
                chown %(user)s:%(group)s %(app_path)s
            fi""") % context
        )
    
    def set_under_construction(self, context):
        if context['under_construction_path']:
            # cambios de permisos en servidores nuevos
            perms = Template(textwrap.dedent("""\
            {% if sftpuser %}
                        chown -R {{ sftpuser }}:{{ sftpuser }} {{ app_path }}/* {% else %}
                        chown -R {{ user }}:{{ group }} {{ app_path }}/*
            {% endif %}
            """
            ))
            context.update({'perms' :  perms.render(Context(context))})
            self.append(textwrap.dedent("""
                # Set under construction if needed
                if [[ $CREATED == 1 && ! $(ls -A %(app_path)s | head -n1) ]]; then
                    # Async wait some seconds for other backends to lock app_path or cp under construction
                    nohup bash -c '
                        sleep 2
                        if [[ ! $(ls -A %(app_path)s | head -n1) ]]; then
                            cp -r %(under_construction_path)s %(app_path)s
                            %(perms)s
                        fi' &> /dev/null &
                fi""") % context
            )

    def delete_webapp_dir(self, context):
        if context['deleted_app_path']:
            self.append(textwrap.dedent("""\
                # Move app into WEBAPPS_MOVE_ON_DELETE_PATH, nesting if exists.
                deleted_app_path="%(deleted_app_path)s"
                while [[ -e $deleted_app_path ]]; do
                    deleted_app_path="${deleted_app_path}/$(basename ${deleted_app_path})"
                done
                mv %(app_path)s $deleted_app_path || exit_code=$?
                """) % context
            )
        else:
            self.append("rm -fr %(app_path)s" % context)
    
    def get_context(self, webapp):
        context = webapp.type_instance.get_directive_context()
        context.update({
            'user': webapp.get_username(),
            'group': webapp.get_groupname(),
            'app_name': webapp.name,
            'app_type': webapp.type,
            'app_path': webapp.get_path(),
            'banner': self.get_banner(),
            'under_construction_path': settings.WEBAPPS_UNDER_CONSTRUCTION_PATH,
            'is_mounted': webapp.content_set.exists(),
            'target_server': webapp.target_server,
            'sftpuser' : webapp.sftpuser.username if webapp.target_server.name in NEW_SERVERS else None
        })
        context['deleted_app_path'] = settings.WEBAPPS_MOVE_ON_DELETE_PATH % context
        return context


for __, module_name, __ in pkgutil.walk_packages(__path__):
    # sorry for the exec(), but Import module function fails :(
    exec('from . import %s' % module_name)