django-orchestra-test/orchestra/contrib/saas/services/helpers.py

from collections import defaultdict
from urllib.parse import urlparse

from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _

from orchestra.contrib.websites.models import Website, WebsiteDirective, Content
from orchestra.contrib.websites.validators import validate_domain_protocol
from orchestra.contrib.orchestration.models import Server
from orchestra.utils.python import AttrDict


def full_clean(obj, exclude=None):
    try:
        obj.full_clean(exclude=exclude)
    except ValidationError as e:
        raise ValidationError({
            'custom_url': _("Error validating related %s: %s") % (type(obj).__name__, e),
        })


def clean_custom_url(saas):
    instance = saas.instance
    instance.custom_url = instance.custom_url.strip()
    url = urlparse(instance.custom_url)
    if not url.path:
        instance.custom_url += '/'
        url = urlparse(instance.custom_url)
    try:
        protocol, valid_protocols = saas.PROTOCOL_MAP[url.scheme]
    except KeyError:
        raise ValidationError({
            'custom_url': _("%s scheme not supported (http/https)") % url.scheme,
        })
    account = instance.account
    # get or create website
    try:
        website = Website.objects.get(
            protocol__in=valid_protocols,
            domains__name=url.netloc,
            account=account,
        )
    except Website.DoesNotExist:
        # get or create domain
        Domain = Website.domains.field.related_model
        try:
            domain = Domain.objects.get(name=url.netloc)
        except Domain.DoesNotExist:
            raise ValidationError({
                'custom_url': _("Domain %s does not exist.") % url.netloc,
            })
        if domain.account != account:
            raise ValidationError({
                'custom_url': _("Domain %s does not belong to account %s, it's from %s.") %
                    (url.netloc, account, domain.account),
            })
        # Create new website for custom_url
        # Changed by daniel: hardcode target_server to web.pangea.lan, consider putting it into settings.py
        tgt_server = Server.objects.get(name='web.pangea.lan')
        website = Website(name=url.netloc, protocol=protocol, account=account, target_server=tgt_server)
        full_clean(website)
        try:
            validate_domain_protocol(website, domain, protocol)
        except ValidationError as e:
            raise ValidationError({
                'custom_url': _("Error validating related %s: %s") % (type(website).__name__, e),
            })
    # get or create directive
    try:
        directive = website.directives.get(name=saas.get_directive_name())
    except WebsiteDirective.DoesNotExist:
        directive = WebsiteDirective(name=saas.get_directive_name(), value=url.path)
    if not directive.pk or directive.value != url.path:
        directive.value = url.path
        if website.pk:
            directive.website = website
            full_clean(directive)
            # Adaptation of orchestra.websites.forms.WebsiteDirectiveInlineFormSet.clean()
            locations = set(
                Content.objects.filter(website=website).values_list('path', flat=True)
            )
            values = defaultdict(list)
            directives = WebsiteDirective.objects.filter(website=website)
            for wdirective in directives.exclude(pk=directive.pk):
                fdirective = AttrDict({
                    'name': wdirective.name,
                    'value': wdirective.value
                })
                try:
                    wdirective.directive_instance.validate_uniqueness(fdirective, values, locations)
                except ValidationError as err:
                    raise ValidationError({
                        'custom_url': _("Another directive with this URL path exists (%s)." % err)
                    })
        else:
            full_clean(directive, exclude=('website',))
    return directive


def create_or_update_directive(saas):
    instance = saas.instance
    url = urlparse(instance.custom_url)
    protocol, valid_protocols = saas.PROTOCOL_MAP[url.scheme]
    account = instance.account
    # get or create website
    try:
        website = Website.objects.get(
            protocol__in=valid_protocols,
            domains__name=url.netloc,
            account=account,
        )
    except Website.DoesNotExist:
        Domain = Website.domains.field.related_model
        domain = Domain.objects.get(name=url.netloc)
        # Create new website for custom_url
        tgt_server = Server.objects.get(name='web.pangea.lan')
        website = Website(name=url.netloc, protocol=protocol, account=account, target_server=tgt_server)
        website.save()
        website.domains.add(domain)
    # get or create directive
    try:
        directive = website.directives.get(name=saas.get_directive_name())
    except WebsiteDirective.DoesNotExist:
        directive = WebsiteDirective(name=saas.get_directive_name(), value=url.path)
    if not directive.pk or directive.value != url.path:
        directive.value = url.path
        directive.website = website
        directive.save()
    return directive


def update_directive(saas):
    saas.instance.custom_url = saas.instance.custom_url.strip()
    url = urlparse(saas.instance.custom_url)