from django.core.exceptions import ValidationError from django.db import models from django.utils.functional import cached_property from django.utils.translation import gettext_lazy as _ from orchestra.core.validators import validate_name from . import settings class ListQuerySet(models.QuerySet): def create(self, **kwargs): """ Sets password if provided, all within a single DB operation """ password = kwargs.pop('password') instance = self.model(**kwargs) if password: instance.set_password(password) instance.save() return instance # TODO address and domain, perhaps allow only domain? class List(models.Model): name = models.CharField(_("name"), max_length=64, unique=True, validators=[validate_name], help_text=_("Default list address <name>@%s") % settings.LISTS_DEFAULT_DOMAIN) address_name = models.CharField(_("address name"), max_length=64, validators=[validate_name], blank=True) address_domain = models.ForeignKey(settings.LISTS_DOMAIN_MODEL, on_delete=models.SET_NULL, verbose_name=_("address domain"), blank=True, null=True) admin_email = models.EmailField(_("admin email"), help_text=_("Administration email address")) account = models.ForeignKey('accounts.Account', verbose_name=_("Account"), related_name='lists', on_delete=models.CASCADE) # TODO also admin is_active = models.BooleanField(_("active"), default=True, help_text=_("Designates whether this account should be treated as active. " "Unselect this instead of deleting accounts.")) password = None objects = ListQuerySet.as_manager() class Meta: unique_together = ('address_name', 'address_domain') def __str__(self): return self.name @property def address(self): if self.address_name and self.address_domain: return "%s@%s" % (self.address_name, self.address_domain) return '' @cached_property def active(self): return self.is_active and self.account.is_active def clean(self): if self.address_name and not self.address_domain_id: raise ValidationError({ 'address_domain': _("Domain should be selected for provided address name."), }) def disable(self): self.is_active = False self.save(update_fields=('is_active',)) def enable(self): self.is_active = False self.save(update_fields=('is_active',)) def get_address_name(self): return self.address_name or self.name def get_username(self): return self.name def set_password(self, password): self.password = password def get_absolute_url(self): context = { 'name': self.name } return settings.LISTS_LIST_URL % context