kumify/dreams/templatetags/dream_stats.py

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)