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
|
from __future__ import annotations
from datetime import date, datetime, timedelta, timezone, tzinfo
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from faker import Faker
def handle_constrained_date(
faker: Faker,
ge: date | None = None,
gt: date | None = None,
le: date | None = None,
lt: date | None = None,
tz: tzinfo = timezone.utc,
) -> date:
"""Generates a date value fulfilling the expected constraints.
:param faker: An instance of faker.
:param lt: Less than value.
:param le: Less than or equal value.
:param gt: Greater than value.
:param ge: Greater than or equal value.
:param tz: A timezone.
:returns: A date instance.
"""
start_date = datetime.now(tz=tz).date() - timedelta(days=100)
if ge:
start_date = ge
elif gt:
start_date = gt + timedelta(days=1)
end_date = datetime.now(tz=timezone.utc).date() + timedelta(days=100)
if le:
end_date = le
elif lt:
end_date = lt - timedelta(days=1)
return faker.date_between(start_date=start_date, end_date=end_date)
|
