1from __future__ import annotations
2
3from typing import TYPE_CHECKING
4
5from isoduration.constants import PERIOD_PREFIX, TIME_PREFIX
6from isoduration.formatter.checking import validate_date_duration
7
8if TYPE_CHECKING: # pragma: no cover
9 from isoduration.types import DateDuration, TimeDuration
10
11
12def format_date(date_duration: DateDuration, global_sign: int) -> str:
13 validate_date_duration(date_duration)
14
15 date_duration_str = PERIOD_PREFIX
16
17 if date_duration.weeks != 0:
18 date_duration_str += f"{(date_duration.weeks * global_sign):g}W"
19
20 if date_duration.years != 0:
21 date_duration_str += f"{(date_duration.years * global_sign):g}Y"
22 if date_duration.months != 0:
23 date_duration_str += f"{(date_duration.months * global_sign):g}M"
24 if date_duration.days != 0:
25 date_duration_str += f"{(date_duration.days * global_sign):g}D"
26
27 return date_duration_str
28
29
30def format_time(time_duration: TimeDuration, global_sign: int) -> str:
31 time_duration_str = TIME_PREFIX
32
33 if time_duration.hours != 0:
34 time_duration_str += f"{(time_duration.hours * global_sign):g}H"
35 if time_duration.minutes != 0:
36 time_duration_str += f"{(time_duration.minutes * global_sign):g}M"
37 if time_duration.seconds != 0:
38 time_duration_str += f"{(time_duration.seconds * global_sign):g}S"
39
40 if time_duration_str == TIME_PREFIX:
41 return ""
42 return time_duration_str