""" This module defines the database models for website handling. """ from __future__ import absolute_import, unicode_literals from django.db import models, transaction from django.utils.encoding import python_2_unicode_compatible from django.utils.translation import ugettext as _ from domains.models import HostingDomain from osusers.models import User as OsUser from fileservertasks.tasks import ( create_file_website_hierarchy, delete_file_website_hierarchy, ) 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, ) @python_2_unicode_compatible 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