Add option to customize the knocker behavior

djangocms_blog/admin.py

@@ -191,6 +191,12 @@ class BlogConfigAdmin(BaseAppHookConfig, TranslatableAdmin):
                 ),
                 'classes': ('collapse',)
             }),
+            ('Notifications', {
+                'fields': (
+                    'config.send_knock_create', 'config.send_knock_update'
+                ),
+                'classes': ('collapse',)
+            }),
             ('Sitemap', {
                 'fields': (
                     'config.sitemap_changefreq', 'config.sitemap_priority',

djangocms_blog/cms_appconfig.py

@@ -125,4 +125,13 @@ class BlogConfigForm(AppDataForm):
         max_length=200, label=_('Google+ author name'), required=False,
         initial=get_setting('GPLUS_AUTHOR')
     )
+
+    send_knock_create = forms.BooleanField(
+        label=_('Send notifications on post creation'), required=False, initial=False,
+        help_text=_('Emits a desktop notification -if enabled- when publishing a new post')
+    )
+    send_knock_update = forms.BooleanField(
+        label=_('Send notifications on post update'), required=False, initial=False,
+        help_text=_('Emits a desktop notification -if enabled- when editing a published post')
+    )
 setup_config(BlogConfigForm, BlogConfig)

djangocms_blog/models.py

@@ -8,6 +8,8 @@ from django.conf import settings as dj_settings
 from django.contrib.auth import get_user_model
 from django.core.urlresolvers import reverse
 from django.db import models
+from django.db.models.signals import pre_save
+from django.dispatch import receiver
 from django.utils import timezone
 from django.utils.encoding import force_text, python_2_unicode_compatible
 from django.utils.html import escape, strip_tags
@@ -259,6 +261,10 @@ class Post(KnockerModel, ModelMeta, TranslatableModel):
         return title.strip()
 
     def get_keywords(self):
+        """
+        Returns the list of keywords (as python list)
+        :return: list
+        """
         return self.safe_translation_getter('meta_keywords', default='').strip().split(',')
 
     def get_locale(self):
@@ -276,10 +282,16 @@ class Post(KnockerModel, ModelMeta, TranslatableModel):
         return ''
 
     def get_tags(self):
+        """
+        Returns the list of object tags as comma separated list
+        """
         taglist = [tag.name for tag in self.tags.all()]
         return ','.join(taglist)
 
     def get_author(self):
+        """
+        Return the author (user) objects
+        """
         return self.author
 
     def _set_default_author(self, current_user):
@@ -303,6 +315,9 @@ class Post(KnockerModel, ModelMeta, TranslatableModel):
             return get_setting('IMAGE_FULL_SIZE')
 
     def get_full_url(self):
+        """
+        Return the url with protocol and domain url
+        """
         return self.build_absolute_uri(self.get_absolute_url())
 
     @property
@@ -315,6 +330,15 @@ class Post(KnockerModel, ModelMeta, TranslatableModel):
                 (self.date_published_end is None or self.date_published_end > timezone.now())
                 )
 
+    def should_knock(self, created=False):
+        """
+        Returns whether to emit knocks according to the post state
+        """
+        new = (self.app_config.send_knock_create and self.is_published and
+               self.date_published == self.date_modified)
+        updated = self.app_config.send_knock_update and self.is_published
+        return new or updated
+
 
 class BasePostPlugin(CMSPlugin):
     app_config = AppHookConfigField(

tests/base.py

@@ -117,8 +117,12 @@ class BaseTest(BaseTestCase):
             namespace='sample_app2', app_title='app2', object_name='Article'
         )
         cls.app_config_1.app_data.config.paginate_by = 1
+        cls.app_config_1.app_data.config.send_knock_create = True
+        cls.app_config_1.app_data.config.send_knock_update = True
         cls.app_config_1.save()
         cls.app_config_2.app_data.config.paginate_by = 2
+        cls.app_config_2.app_data.config.send_knock_create = True
+        cls.app_config_2.app_data.config.send_knock_update = True
         cls.app_config_2.save()
         cls.app_configs = {
             'sample_app': cls.app_config_1,

tests/test_models.py

@@ -666,3 +666,51 @@ class KnockerTest(BaseTest):
                                  'new {0}'.format(posts[0]._meta.verbose_name))
                 self.assertEqual(knock_create['message'], posts[0].title)
                 self.assertEqual(knock_create['language'], language)
+
+    def test_should_knock(self):
+        self.get_pages()
+
+        post_data = {
+            'author': self.user,
+            'title': 'post 1',
+            'abstract': 'post 1',
+            'meta_description': 'post 1',
+            'meta_keywords': 'post 1',
+            'app_config': self.app_config_1
+        }
+        post = Post.objects.create(**post_data)
+        # Object is not published, no knock
+        self.assertFalse(post.should_knock())
+        post.publish = True
+        post.save()
+        # Object is published, send knock
+        self.assertTrue(post.should_knock())
+
+        # Knock disabled for updates
+        self.app_config_1.app_data.config.send_knock_update = False
+        self.app_config_1.save()
+        post.abstract = 'what'
+        post.save()
+        self.assertFalse(post.should_knock())
+
+        # Knock disabled for publishing
+        self.app_config_1.app_data.config.send_knock_create = False
+        self.app_config_1.save()
+        post_data = {
+            'author': self.user,
+            'title': 'post 2',
+            'abstract': 'post 2',
+            'meta_description': 'post 2',
+            'meta_keywords': 'post 2',
+            'app_config': self.app_config_1
+        }
+        post = Post.objects.create(**post_data)
+        self.assertFalse(post.should_knock())
+        post.publish = True
+        post.save()
+        self.assertFalse(post.should_knock())
+
+        # Restore default values
+        self.app_config_1.app_data.config.send_knock_create = True
+        self.app_config_1.app_data.config.send_knock_update = True
+        self.app_config_1.save()