Hopiu/django-notifications
1
0
Fork 0
mirror of https://github.com/Hopiu/django-notifications.git synced 2026-05-02 10:44:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[feat] add caching and a setting CACHE_TIMEOUT to configure the timeout

Browse source
This commit is contained in:
areski 2022-11-16 17:32:45 +01:00 committed by Alvaro Mariano
parent e63bd3f45f
commit 2b24894432
2 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
notifications/settings.py
View file

@ -8,6 +8,7 @@ CONFIG_DEFAULTS = {
'USE_JSONFIELD': False,
'SOFT_DELETE': False,
'NUM_TO_FETCH': 10,
'CACHE_TIMEOUT': 2,
}

15
notifications/templatetags/notifications_tags.py
View file

@ -5,6 +5,9 @@ from distutils.version import StrictVersion # pylint: disable=no-name-in-module
from django import get_version
from django.template import Library
from django.utils.html import format_html
from django.core.cache import cache
from notifications import settings
from notifications.settings import get_config
try:
from django.urls import reverse
@ -14,11 +17,19 @@ except ImportError:
register = Library()
def get_cached_notification_unread_count(user):
return cache.get_or_set(
'cache_notification_unread_count',
user.notifications.unread().count,
settings.get_config()['CACHE_TIMEOUT']
)
def notifications_unread(context):
user = user_context(context)
if not user:
return ''
return user.notifications.unread().count()
return get_cached_notification_unread_count(user)
if StrictVersion(get_version()) >= StrictVersion('2.0'):
@ -90,7 +101,7 @@ def live_notify_badge(context, badge_class='live_notify_badge'):
return ''
html = "<span class='{badge_class}'>{unread}</span>".format(
badge_class=badge_class, unread=user.notifications.unread().count()
badge_class=badge_class, unread=get_cached_notification_unread_count(user)
)
return format_html(html)