""" 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