NDOJ/judge/jinja2/timedelta.py

import datetime

from judge.utils.timedelta import nice_repr
from . import registry


@registry.filter
def timedelta(value, display='long'):
    if value is None:
        return value
    return nice_repr(value, display)


@registry.filter
def timestampdelta(value, display='long'):
    value = datetime.timedelta(seconds=value)
    return timedelta(value, display)


@registry.filter
def seconds(timedelta):
    return timedelta.total_seconds()


@registry.filter
@registry.render_with('time-remaining-fragment.html')
def as_countdown(time):
    time_now = datetime.datetime.now(datetime.timezone.utc)
    initial = abs(time - time_now)
    return {'countdown': time, 'initial': initial}