from django.db import models from django.db.models.signals import pre_save, pre_delete from django.dispatch import receiver from django.utils.functional import cached_property from django.utils.translation import ugettext_lazy as _ from jsonfield import JSONField from orchestra.core import services, validators from .fields import VirtualDatabaseRelation from .services import SoftwareService class SaaS(models.Model): service = models.CharField(_("service"), max_length=32, choices=SoftwareService.get_choices()) name = models.CharField(_("Name"), max_length=64, help_text=_("Required. 64 characters or fewer. Letters, digits and ./-/_ only."), validators=[validators.validate_username]) account = models.ForeignKey('accounts.Account', verbose_name=_("account"), related_name='saas') is_active = models.BooleanField(_("active"), default=True, help_text=_("Designates whether this service should be treated as active. ")) data = JSONField(_("data"), default={}, help_text=_("Extra information dependent of each service.")) database = models.ForeignKey('databases.Database', null=True, blank=True) # Some SaaS sites may need a database, with this virtual field we tell the ORM to delete them databases = VirtualDatabaseRelation('databases.Database') class Meta: verbose_name = "SaaS" verbose_name_plural = "SaaS" unique_together = ( ('name', 'service'), ) def __str__(self): return "%s@%s" % (self.name, self.service) @cached_property def service_class(self): return SoftwareService.get(self.service) @cached_property def service_instance(self): """ Per request lived service_instance """ return self.service_class(self) @cached_property def active(self): return self.is_active and self.account.is_active def clean(self): self.data = self.service_instance.clean_data() def get_site_domain(self): return self.service_instance.get_site_domain() def set_password(self, password): self.password = password services.register(SaaS) # Admin bulk deletion doesn't call model.delete() # So, signals are used instead of model method overriding @receiver(pre_save, sender=SaaS, dispatch_uid='saas.service.save') def type_save(sender, *args, **kwargs): instance = kwargs['instance'] instance.service_instance.save() @receiver(pre_delete, sender=SaaS, dispatch_uid='saas.service.delete') def type_delete(sender, *args, **kwargs): instance = kwargs['instance'] try: instance.service_instance.delete() except KeyError: pass