blob: dad219dd115a805f216bd61629803ecf418426bf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
from __future__ import annotations
import typing as t
from pendulum.locales.locale import Locale
if t.TYPE_CHECKING:
from pendulum import Duration
class DifferenceFormatter:
"""
Handles formatting differences in text.
"""
def __init__(self, locale: str = "en") -> None:
self._locale = Locale.load(locale)
def format(
self,
diff: Duration,
is_now: bool = True,
absolute: bool = False,
locale: str | Locale | None = None,
) -> str:
"""
Formats a difference.
:param diff: The difference to format
:param is_now: Whether the difference includes now
:param absolute: Whether it's an absolute difference or not
:param locale: The locale to use
"""
if locale is None:
locale = self._locale
else:
locale = Locale.load(locale)
if diff.years > 0:
unit = "year"
count = diff.years
if diff.months > 6:
count += 1
elif diff.months == 11 and (diff.weeks * 7 + diff.remaining_days) > 15:
unit = "year"
count = 1
elif diff.months > 0:
unit = "month"
count = diff.months
if (diff.weeks * 7 + diff.remaining_days) >= 27:
count += 1
elif diff.weeks > 0:
unit = "week"
count = diff.weeks
if diff.remaining_days > 3:
count += 1
elif diff.remaining_days > 0:
unit = "day"
count = diff.remaining_days
if diff.hours >= 22:
count += 1
elif diff.hours > 0:
unit = "hour"
count = diff.hours
elif diff.minutes > 0:
unit = "minute"
count = diff.minutes
elif 10 < diff.remaining_seconds <= 59:
unit = "second"
count = diff.remaining_seconds
else:
# We check if the "a few seconds" unit exists
time = locale.get("custom.units.few_second")
if time is not None:
if absolute:
return t.cast(str, time)
key = "custom"
is_future = diff.invert
if is_now:
if is_future:
key += ".from_now"
else:
key += ".ago"
else:
if is_future:
key += ".after"
else:
key += ".before"
return t.cast(str, locale.get(key).format(time))
else:
unit = "second"
count = diff.remaining_seconds
if count == 0:
count = 1
if absolute:
key = f"translations.units.{unit}"
else:
is_future = diff.invert
if is_now:
# Relative to now, so we can use
# the CLDR data
key = f"translations.relative.{unit}"
if is_future:
key += ".future"
else:
key += ".past"
else:
# Absolute comparison
# So we have to use the custom locale data
# Checking for special pluralization rules
key = "custom.units_relative"
if is_future:
key += f".{unit}.future"
else:
key += f".{unit}.past"
trans = locale.get(key)
if not trans:
# No special rule
key = f"translations.units.{unit}.{locale.plural(count)}"
time = locale.get(key).format(count)
else:
time = trans[locale.plural(count)].format(count)
key = "custom"
if is_future:
key += ".after"
else:
key += ".before"
return t.cast(str, locale.get(key).format(time))
key += f".{locale.plural(count)}"
return t.cast(str, locale.get(key).format(count))
|