class CachedEmailAddress(EmailAddress, cachemodel.CacheModel):
objects = CachedEmailAddressManager()
class Meta:
proxy = True
verbose_name = _("email address")
verbose_name_plural = _("email addresses")
def publish(self):
super(CachedEmailAddress, self).publish()
self.publish_by('email')
self.user.publish()
def delete(self, *args, **kwargs):
user = self.user
self.publish_delete('email')
self.publish_delete('pk')
super(CachedEmailAddress, self).delete(*args, **kwargs)
user.publish()
def set_as_primary(self, conditional=False):
# shadow parent function, but use CachedEmailAddress manager to ensure cache gets updated
old_primary = CachedEmailAddress.objects.get_primary(self.user)
if old_primary:
if conditional:
return False
old_primary.primary = False
old_primary.save()
return super(CachedEmailAddress,
self).set_as_primary(conditional=conditional)
def save(self, *args, **kwargs):
super(CachedEmailAddress, self).save(*args, **kwargs)
if not self.emailaddressvariant_set.exists(
) and self.email != self.email.lower():
self.add_variant(self.email.lower())
@cachemodel.cached_method(auto_publish=True)
def cached_variants(self):
return self.emailaddressvariant_set.all()
def add_variant(self, email_variation):
existing_variants = EmailAddressVariant.objects.filter(
canonical_email=self, email=email_variation)
if email_variation not in [e.email for e in existing_variants.all()]:
return EmailAddressVariant.objects.create(canonical_email=self,
email=email_variation)
else:
raise ValidationError(
"Email variant {} already exists".format(email_variation))
class CachedEmailAddress(EmailAddress, cachemodel.CacheModel):
objects = CachedEmailAddressManager()
class Meta:
proxy = True
verbose_name = _("email address")
verbose_name_plural = _("email addresses")
def generate_forgot_password_time_cache_key(self):
return "{}_forgot_request_date".format(self.email)
def get_last_forgot_password_sent_time(self):
cached_time = cache.get(self.generate_forgot_password_time_cache_key())
return cached_time
def set_last_forgot_password_sent_time(self, new_datetime):
cache.set(self.generate_forgot_password_time_cache_key(), new_datetime)
def generate_verification_time_cache_key(self):
return "{}_verification_request_date".format(self.email)
def get_last_verification_sent_time(self):
cached_time = cache.get(self.generate_verification_time_cache_key())
return cached_time
def set_last_verification_sent_time(self, new_datetime):
cache.set(self.generate_verification_time_cache_key(), new_datetime)
def publish(self):
super(CachedEmailAddress, self).publish()
self.publish_by('email')
self.user.publish()
def delete(self, *args, **kwargs):
user = self.user
self.publish_delete('email')
self.publish_delete('pk')
process_post_recipient_id_deletion.delay(self.email)
super(CachedEmailAddress, self).delete(*args, **kwargs)
user.publish()
def set_as_primary(self, conditional=False):
# shadow parent function, but use CachedEmailAddress manager to ensure cache gets updated
old_primary = CachedEmailAddress.objects.get_primary(self.user)
if old_primary:
if conditional:
return False
old_primary.primary = False
old_primary.save()
return super(CachedEmailAddress, self).set_as_primary(conditional=conditional)
def save(self, *args, **kwargs):
super(CachedEmailAddress, self).save(*args, **kwargs)
process_post_recipient_id_verification_change.delay(self.email, 'email', self.verified)
if not self.emailaddressvariant_set.exists() and self.email != self.email.lower():
self.add_variant(self.email.lower())
@cachemodel.cached_method(auto_publish=True)
def cached_variants(self):
return self.emailaddressvariant_set.all()
def cached_variant_emails(self):
return [e.email for e in self.cached_variants()]
def add_variant(self, email_variation):
existing_variants = EmailAddressVariant.objects.filter(
canonical_email=self, email=email_variation
)
if email_variation not in [e.email for e in existing_variants.all()]:
return EmailAddressVariant.objects.create(
canonical_email=self, email=email_variation
)
else:
raise ValidationError("Email variant {} already exists".format(email_variation))