newsletter/models.py

   1 import datetime
   2 import hashlib
   3 import random
   4 import smtplib
   5 import subprocess
   6
   7 from email.mime.multipart import MIMEMultipart
   8 from email.mime.text import MIMEText
   9
  10 import html2text
  11
  12 from django.db import models, IntegrityError
  13 from django.core.urlresolvers import reverse
  14
  15 from django.template import loader
  16 from django.template.loader import render_to_string
  17 from django.conf import settings
  18
  19 from django.utils.encoding import force_bytes
  20 from django.utils.safestring import mark_safe
  21 from django.utils.translation import ugettext
  22 from django.utils.translation import ugettext_lazy as _
  23 from django.core.mail import send_mail
  24
  25 from ckeditor.fields import RichTextField
  26
  27
  28 class Subscriber(models.Model) :
  29     email = models.EmailField(unique=False)
  30     inscription_date = models.DateField(auto_now_add=True)
  31     is_validated = models.NullBooleanField()
  32     is_registered = models.NullBooleanField()
  33     password = models.CharField(max_length=100)
  34     bot_check1 = models.BooleanField(default=False)
  35     bot_check2 = models.BooleanField(default=False)
  36     user_agent = models.CharField(max_length=1000, blank=True)
  37     source_ip = models.CharField(max_length=100, blank=True)
  38
  39     def __unicode__(self) :
  40         return self.email
  41
  42     def is_from_bot(self):
  43         return bool(self.bot_check1 or self.bot_check2)
  44
  45     def send_confirmation_email(self):
  46         if self.is_from_bot():
  47             return
  48         self.password = hashlib.sha1(force_bytes(str(random.random()))).hexdigest()
  49         confirm_subject = loader.get_template('newsletter/confirmation_email_subject.txt')
  50         confirm_body = loader.get_template('newsletter/confirmation_email_body.txt')
  51         context = {'token': self.password}
  52         send_mail(confirm_subject.render(context).strip(), confirm_body.render(context),
  53                 settings.NEWSLETTER_SENDER, [self.email])
  54         self.is_validated = False
  55         self.save()
  56
  57     def subscribe_in_mailman(self):
  58         t = subprocess.call(['listadmin', '--add-member', self.email, settings.NEWSLETTER_NAME])
  59         if t != 0:
  60             return
  61         self.is_registered = True
  62         self.save()
  63
  64
  65 class Newsletter(models.Model):
  66
  67     class Meta:
  68         verbose_name = _('Newsletter')
  69         verbose_name_plural = _('Newsletters')
  70         ordering = ['date']
  71
  72     subject = models.CharField(_('Title'), max_length=50)
  73     date = models.DateField(_('Date'))
  74     text = RichTextField(_('Text'), null=True, blank=True)
  75
  76     expedition_datetime = models.DateTimeField(_('Expedition Date/time'), null=True, blank=True)
  77
  78     def send(self):
  79         msg = MIMEMultipart('alternative')
  80         msg['Subject'] = self.subject
  81         msg['From'] = settings.NEWSLETTER_SENDER
  82         msg['To'] = '%s@%s' % (settings.NEWSLETTER_NAME, settings.NEWSLETTER_DOMAIN)
  83
  84         h2t = html2text.HTML2Text()
  85         h2t.unicode_snob = True
  86         context = {
  87             'text_part': mark_safe(h2t.handle(self.text)),
  88             'html_part': mark_safe(self.text),
  89         }
  90
  91         part1 = MIMEText(render_to_string('newsletter/email_body.txt', context), 'plain', _charset='utf-8')
  92         part2 = MIMEText(render_to_string('newsletter/email_body.html', context), 'html', _charset='utf-8')
  93
  94         msg.attach(part1)
  95         msg.attach(part2)
  96
  97         s = smtplib.SMTP('localhost')
  98         s.sendmail(msg['From'], msg['To'], msg.as_string())
  99         s.quit()
 100
 101         self.expedition_datetime = datetime.datetime.now()
 102         self.save()
 103
 104     def get_absolute_url(self):
 105         return reverse('newsletter-view', kwargs={'pk': self.id})