Skip to content

Commit 8ae67b2

Browse files
committed
chore: fix now refs
1 parent 081a36a commit 8ae67b2

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

airbyte_cdk/sources/declarative/interpolation/macros.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def today_utc() -> datetime.date:
4141
Usage:
4242
`"{{ today_utc() }}"`
4343
"""
44-
return datetime.datetime.now(datetime.timezone.utc).date()
44+
return ab_datetime_now().date()
4545

4646

4747
def today_with_timezone(timezone: str) -> datetime.date:
@@ -51,7 +51,7 @@ def today_with_timezone(timezone: str) -> datetime.date:
5151
:param timezone: timezone expressed as IANA keys format. Example: "Pacific/Tarawa"
5252
:return:
5353
"""
54-
return datetime.datetime.now(tz=pytz.timezone(timezone)).date()
54+
return ab_datetime_now(tz=timezone).date()
5555

5656

5757
def timestamp(dt: Union[float, str]) -> Union[int, float]:

airbyte_cdk/utils/datetime_helpers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,9 @@ def from_str(cls, dt_str: str) -> "AirbyteDateTime":
344344
return ab_datetime_parse(dt_str)
345345

346346

347-
def ab_datetime_now() -> AirbyteDateTime:
347+
def ab_datetime_now(
348+
tz: str | timezone | None = None,
349+
) -> AirbyteDateTime:
348350
"""Returns the current time as an AirbyteDateTime in UTC timezone.
349351
350352
Returns:

0 commit comments

Comments
 (0)