takahe/activities/templatetags/activity_tags.py
2023-02-19 10:49:06 -07:00

64 lines
1.6 KiB
Python

import datetime
from urllib.parse import urlencode
from django import template
from django.utils import timezone
register = template.Library()
@register.filter
def timedeltashort(value: datetime.datetime):
"""
A more compact version of timesince
"""
if not value:
return ""
delta = timezone.now() - value
seconds = int(delta.total_seconds())
sign = "-" if seconds < 0 else ""
seconds = abs(seconds)
days = abs(delta.days)
if seconds < 60:
text = f"{seconds:0n}s"
elif seconds < 60 * 60:
minutes = seconds // 60
text = f"{minutes:0n}m"
elif seconds < 60 * 60 * 24:
hours = seconds // (60 * 60)
text = f"{hours:0n}h"
elif days < 365:
text = f"{days:0n}d"
else:
years = max(days // 365.25, 1)
text = f"{years:0n}y"
return sign + text
@register.filter
def timedeltashortenddate(value: datetime.datetime):
"""
Formatter for end dates - timedeltashort but it adds "ended ... ago" or
"left" depending on the direction.
"""
output = timedeltashort(value)
if output.startswith("-"):
return f"{output[1:]} left"
else:
return f"Ended {output} ago"
@register.simple_tag(takes_context=True)
def urlparams(context, **kwargs):
"""
Generates a URL parameter string the same as the current page but with
the given items changed.
"""
params = dict(context["request"].GET.items())
for name, value in kwargs.items():
if value:
params[name] = value
elif name in params:
del params[name]
return urlencode(params)