add a subscribe form cell

[django-panik-newsletter.git] / newsletter / models.py

diff --git a/newsletter/models.py b/newsletter/models.py
index a2aee89637ffa51cd9eef5a57f7d2c971d0f1e5e..75739c14cfb51a191ebce99713b2a70fce126814 100644 (file)
--- a/newsletter/models.py
+++ b/newsletter/models.py
@@ -11,6 +11,8 @@ import html2text
 import mechanize
 import requests
 from ckeditor.fields import RichTextField
+from combo.data.library import register_cell_class
+from combo.data.models import CellBase
 from django.conf import settings
 from django.core.mail import send_mail
 from django.db import IntegrityError, models
@@ -162,3 +164,18 @@ class Newsletter(models.Model):
 
     def get_absolute_url(self):
         return reverse('newsletter-view', kwargs={'pk': self.id})
+
+
+@register_cell_class
+class NewsletterSubscribeFormCell(CellBase):
+    default_template_name = 'newsletter/subscribe_form_cell.html'
+
+    class Meta:
+        verbose_name = _('Newsletter Subscribe Form')
+
+    def get_cell_extra_context(self, context):
+        from .forms import SubscribeForm
+
+        context = super().get_cell_extra_context(context)
+        context['newsletter_form'] = SubscribeForm()
+        return context