gva/gnuviechadmin/websites/models.py

"""
This module defines the database models for website handling.

"""
from __future__ import absolute_import

from django.db import models, transaction
from django.utils.translation import gettext as _

from domains.models import HostingDomain
from fileservertasks.tasks import (
    create_file_website_hierarchy,
    delete_file_website_hierarchy,
)
from osusers.models import User as OsUser
from webtasks.tasks import (
    create_web_php_fpm_pool_config,
    create_web_vhost_config,
    delete_web_php_fpm_pool_config,
    delete_web_vhost_config,
    disable_web_vhost,
    enable_web_vhost,
)


class Website(models.Model):
    """
    This is the model for a website.

    """

    subdomain = models.CharField(_("sub domain"), max_length=64)
    osuser = models.ForeignKey(
        OsUser, verbose_name=_("operating system user"), on_delete=models.CASCADE
    )
    domain = models.ForeignKey(HostingDomain, models.CASCADE, verbose_name=_("domain"))
    wildcard = models.BooleanField(_("wildcard"), default=False)

    class Meta:
        unique_together = [("domain", "subdomain")]
        verbose_name = _("website")
        verbose_name_plural = _("websites")

    def __str__(self):
        return self.get_fqdn()

    def get_fqdn(self):
        return "{subdomain}.{domain}".format(
            subdomain=self.subdomain, domain=self.domain.domain
        )

    @transaction.atomic
    def save(self, *args, **kwargs):
        if not self.pk:
            fqdn = self.get_fqdn()
            if not Website.objects.filter(osuser=self.osuser).exists():
                create_web_php_fpm_pool_config.delay(self.osuser.username).get()
            create_file_website_hierarchy.delay(self.osuser.username, fqdn).get()
            create_web_vhost_config.delay(
                self.osuser.username, fqdn, self.wildcard
            ).get()
            enable_web_vhost.delay(fqdn).get()
        return super(Website, self).save(*args, **kwargs)

    @transaction.atomic
    def delete(self, *args, **kwargs):
        fqdn = self.get_fqdn()
        osuser = self.osuser
        disable_web_vhost.delay(fqdn).get()
        delete_web_vhost_config.delay(fqdn).get()
        delete_file_website_hierarchy.delay(osuser.username, fqdn).get()
        deleted = super(Website, self).delete(*args, **kwargs)
        if not Website.objects.filter(osuser=osuser):
            delete_web_php_fpm_pool_config.delay(osuser.username).get()
        return deleted