From 551c91358cab1537f7c4243128840f9c550b7bc3 Mon Sep 17 00:00:00 2001
From: Dominik George <nik@naturalnet.de>
Date: Wed, 4 Sep 2019 21:47:33 +0200
Subject: [PATCH] Add template fitler to filter query sets by week number.
---
biscuit/apps/chronos/templatetags/week_helpers.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/biscuit/apps/chronos/templatetags/week_helpers.py b/biscuit/apps/chronos/templatetags/week_helpers.py
index 31df6f58..5c4e0956 100644
--- a/biscuit/apps/chronos/templatetags/week_helpers.py
+++ b/biscuit/apps/chronos/templatetags/week_helpers.py
@@ -1,8 +1,10 @@
from datetime import date
+from typing import Optional
from django import template
+from django.db.query import QuerySet
-from ..util import week_days, week_weekday_to_date
+from ..util import current_week, week_days, week_weekday_to_date
register = template.Library()
@@ -18,6 +20,12 @@ def week_end(week: int) -> date:
return week_days(week)[-1]
+@register.filter
+def only_week(qs: QuerySet, week: Optional[int]) -> QuerySet:
+ wanted_week = week or current_week()
+ return qs.filter(week=wanted_week)
+
+
@register.simple_tag
def weekday_to_date(week: int, weekday: int) -> date:
return week_weekday_to_date(week, weekday)
--
GitLab