mirror of https://github.com/zulip/zulip.git
114 lines
4.1 KiB
Python
114 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright: (c) 2008, Jarek Zgoda <jarek.zgoda@gmail.com>
|
|
|
|
__revision__ = '$Id: models.py 28 2009-10-22 15:03:02Z jarek.zgoda $'
|
|
|
|
import os
|
|
import re
|
|
import base64
|
|
|
|
from django.db import models
|
|
from django.core.urlresolvers import reverse
|
|
from django.core.mail import send_mail
|
|
from django.conf import settings
|
|
from django.template import loader, Context
|
|
from django.contrib.sites.models import Site
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.contrib.contenttypes import generic
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.utils.timezone import now
|
|
|
|
from confirmation.util import get_status_field
|
|
|
|
try:
|
|
import mailer
|
|
send_mail = mailer.send_mail
|
|
except ImportError:
|
|
# no mailer app present, stick with default
|
|
pass
|
|
|
|
|
|
B16_RE = re.compile('^[a-f0-9]{40}$')
|
|
|
|
def generate_key():
|
|
return base64.b16encode(os.urandom(20)).lower()
|
|
|
|
def generate_activation_url(key):
|
|
current_site = Site.objects.get_current()
|
|
return u'https://%s%s' % (current_site.domain,
|
|
reverse('confirmation.views.confirm', kwargs={'confirmation_key': key}))
|
|
|
|
|
|
class ConfirmationManager(models.Manager):
|
|
|
|
def confirm(self, confirmation_key):
|
|
if B16_RE.search(confirmation_key):
|
|
try:
|
|
confirmation = self.get(confirmation_key=confirmation_key)
|
|
except self.model.DoesNotExist:
|
|
return False
|
|
obj = confirmation.content_object
|
|
status_field = get_status_field(obj._meta.app_label, obj._meta.module_name)
|
|
setattr(obj, status_field, getattr(settings, 'STATUS_ACTIVE', 1))
|
|
obj.save()
|
|
return obj
|
|
return False
|
|
|
|
def get_link_for_object(self, obj):
|
|
key = generate_key()
|
|
self.create(content_object=obj, date_sent=now(), confirmation_key=key)
|
|
return generate_activation_url(key)
|
|
|
|
def send_confirmation(self, obj, email_address, additional_context=None,
|
|
subject_template_path=None, body_template_path=None):
|
|
confirmation_key = generate_key()
|
|
current_site = Site.objects.get_current()
|
|
activate_url = generate_activation_url(confirmation_key)
|
|
context = Context({
|
|
'activate_url': activate_url,
|
|
'current_site': current_site,
|
|
'confirmation_key': confirmation_key,
|
|
'target': obj,
|
|
'days': getattr(settings, 'EMAIL_CONFIRMATION_DAYS', 10),
|
|
})
|
|
if additional_context is not None:
|
|
context.update(additional_context)
|
|
templates = [
|
|
'confirmation/%s_confirmation_email_subject.txt' % obj._meta.module_name,
|
|
'confirmation/confirmation_email_subject.txt',
|
|
]
|
|
if subject_template_path:
|
|
template = loader.get_template(subject_template_path)
|
|
else:
|
|
template = loader.select_template(templates)
|
|
subject = template.render(context).strip().replace(u'\n', u' ') # no newlines, please
|
|
templates = [
|
|
'confirmation/%s_confirmation_email_body.txt' % obj._meta.module_name,
|
|
'confirmation/confirmation_email_body.txt',
|
|
]
|
|
if body_template_path:
|
|
template = loader.get_template(body_template_path)
|
|
else:
|
|
template = loader.select_template(templates)
|
|
body = template.render(context)
|
|
send_mail(subject, body, settings.DEFAULT_FROM_EMAIL, [email_address])
|
|
return self.create(content_object=obj, date_sent=now(), confirmation_key=confirmation_key)
|
|
|
|
|
|
class Confirmation(models.Model):
|
|
content_type = models.ForeignKey(ContentType)
|
|
object_id = models.PositiveIntegerField()
|
|
content_object = generic.GenericForeignKey('content_type', 'object_id')
|
|
date_sent = models.DateTimeField(_('sent'))
|
|
confirmation_key = models.CharField(_('activation key'), max_length=40)
|
|
|
|
objects = ConfirmationManager()
|
|
|
|
class Meta:
|
|
verbose_name = _('confirmation email')
|
|
verbose_name_plural = _('confirmation emails')
|
|
|
|
def __unicode__(self):
|
|
return _('confirmation email for %s') % self.content_object
|