more now() updates

Author: aaronsteers

airbyte_cdk/sources/streams/call_rate.py

@@ -334,7 +334,7 @@ def try_acquire(self, request: Any, weight: int) -> None:
             self._update_current_window()
 
             if self._calls_num + weight > self._call_limit:
-                reset_in = self._next_reset_ts - datetime.datetime.now()
+                reset_in = self._next_reset_ts - ab_datetime_now()
                 error_message = (
                     f"reached maximum number of allowed calls {self._call_limit} "
                     f"per {self._offset} interval, next reset in {reset_in}."

airbyte_cdk/utils/datetime_helpers.py

@@ -347,8 +347,6 @@ def from_str(cls, dt_str: str) -> "AirbyteDateTime":
 def ab_datetime_now() -> AirbyteDateTime:
     """Returns the current time as an AirbyteDateTime in UTC timezone.
 
-    Previously named: now()
-
     Returns:
         AirbyteDateTime: Current UTC time.
 

unit_tests/sources/declarative/interpolation/test_jinja.py

@@ -11,6 +11,7 @@
 from airbyte_cdk import StreamSlice
 from airbyte_cdk.sources.declarative.interpolation.jinja import JinjaInterpolation
 from airbyte_cdk.utils import AirbyteTracedException
+from airbyte_cdk.utils.datetime_helpers import ab_datetime_now
 
 interpolation = JinjaInterpolation()
 
@@ -148,9 +149,9 @@ def test_negative_day_delta():
     delta_template = "{{ day_delta(-25) }}"
     val = interpolation.eval(delta_template, {})
 
-    assert val <= (
-        datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(days=25)
-    ).strftime("%Y-%m-%dT%H:%M:%S.%f%z")
+    assert val <= (ab_datetime_now() - datetime.timedelta(days=25)).strftime(
+        "%Y-%m-%dT%H:%M:%S.%f%z"
+    )
 
 
 @pytest.mark.parametrize(

unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py

@@ -16,9 +16,10 @@
 from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
 from airbyte_cdk.sources.file_based.stream import AbstractFileBasedStream
+from airbyte_cdk.utils.datetime_helpers import ab_datetime_now
 
 _FILE_WITH_UNKNOWN_EXTENSION = RemoteFile(
-    uri="a.unknown_extension", last_modified=datetime.now(), file_type="csv"
+    uri="a.unknown_extension", last_modified=ab_datetime_now(), file_type="csv"
 )
 _ANY_CONFIG = FileBasedStreamConfig(
     name="config.name",