Source code for restapi.utilities.time

from datetime import datetime, timedelta
from typing import Literal

import pytz

from restapi.exceptions import BadRequest

EPOCH: datetime = datetime.fromtimestamp(0, pytz.utc)

AllowedTimedeltaPeriods = Literal[
    "days", "seconds", "microseconds", "milliseconds", "minutes", "hours", "weeks"
]




def date_lower_than(a: datetime, b: datetime) -> bool:
    return a.replace(tzinfo=pytz.utc) < b.replace(tzinfo=pytz.utc)





def get_timedelta(every: int, period: AllowedTimedeltaPeriods) -> timedelta:
    if period == "seconds":
        return timedelta(seconds=every)

    if period == "days":
        return timedelta(days=every)

    if period == "microseconds":
        return timedelta(microseconds=every)

    if period == "milliseconds":
        return timedelta(milliseconds=every)

    if period == "minutes":
        return timedelta(minutes=every)

    if period == "hours":
        return timedelta(hours=every)

    if period == "weeks":
        return timedelta(weeks=every)

    raise BadRequest(f"Invalid timedelta period: {period}")





def pluralize(value: int, unit: str) -> str:
    if value == 1:
        return f"{value} {unit}"
    return f"{value} {unit}s"





def seconds_to_human(seconds: int) -> str:
    elements: list[str] = []
    if seconds < 60:
        elements.append(pluralize(seconds, "second"))

    elif seconds < 3600:
        m, s = divmod(seconds, 60)
        elements.append(pluralize(m, "minute"))
        if s > 0:
            elements.append(pluralize(s, "second"))

    elif seconds < 86400:
        m, s = divmod(seconds, 60)
        h, m = divmod(m, 60)
        elements.append(pluralize(h, "hour"))
        if m > 0 or s > 0:
            elements.append(pluralize(m, "minute"))
        if s > 0:
            elements.append(pluralize(s, "second"))
    else:
        m, s = divmod(seconds, 60)
        h, m = divmod(m, 60)
        d, h = divmod(h, 24)
        elements.append(pluralize(d, "day"))
        if h > 0 or m > 0 or s > 0:
            elements.append(pluralize(h, "hour"))
        if m > 0 or s > 0:
            elements.append(pluralize(m, "minute"))
        if s > 0:
            elements.append(pluralize(s, "second"))

    return ", ".join(elements)