Kumi
89dc526a69
Reinforced user data access rules to bolster security and reorganized distribution files into separate directories for cleaner structure. Added a new heatmap visualization for mood statistics on the dashboard, making user engagements more interactive and insightful. Implemented a JSON view to support the heatmap feature, fetching mood entries within a specified time range. This change responds to the need for improved data security and a more engaging user interface, directly addressing user feedback for clearer insights into their mood patterns over time.
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
from django import template
|
|
|
|
from datetime import datetime, timedelta
|
|
from collections import Counter
|
|
|
|
register = template.Library()
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def total_dreams(context):
|
|
return len(context["user"].dream_set.all())
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def weekly_dreams(context):
|
|
now = datetime.now()
|
|
start = now - timedelta(days=7)
|
|
|
|
return len(
|
|
context["user"].dream_set.filter(timestamp__gte=start, timestamp__lte=now)
|
|
)
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def most_common_theme(context, start, end=None):
|
|
dream_list = context["user"].dream_set.filter(
|
|
timestamp__gte=start.date(),
|
|
timestamp__lte=(end.date() if end else start.date()),
|
|
)
|
|
themes = list()
|
|
|
|
for dream in dream_list:
|
|
for theme in dream.dreamtheme_set.all():
|
|
themes.append(theme.theme)
|
|
|
|
try:
|
|
most_common = Counter(themes).most_common(1)[0]
|
|
return most_common[0], most_common[1]
|
|
except Exception:
|
|
return None, None
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def most_common_theme_weekly(context):
|
|
now = datetime.now()
|
|
start = now - timedelta(days=7)
|
|
|
|
return most_common_theme(context, start, now)
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def special_dreams(context, start, end=None):
|
|
dream_list = context["user"].dream_set.filter(
|
|
timestamp__gte=start.date(),
|
|
timestamp__lte=(end.date() if end else start.date()),
|
|
)
|
|
lucid = sum([1 for dream in dream_list if dream.lucid])
|
|
wet = sum([1 for dream in dream_list if dream.wet])
|
|
return lucid, wet
|
|
|
|
|
|
@register.simple_tag(takes_context=True)
|
|
def special_dreams_weekly(context):
|
|
now = datetime.now()
|
|
start = now - timedelta(days=7)
|
|
|
|
return special_dreams(context, start, now)
|