airbyte-cdk 0.72.1__py3-none-any.whl → 6.17.1.dev0__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- airbyte_cdk/__init__.py +355 -6
- airbyte_cdk/cli/__init__.py +1 -0
- airbyte_cdk/cli/source_declarative_manifest/__init__.py +5 -0
- airbyte_cdk/cli/source_declarative_manifest/_run.py +230 -0
- airbyte_cdk/cli/source_declarative_manifest/spec.json +17 -0
- airbyte_cdk/config_observation.py +29 -10
- airbyte_cdk/connector.py +24 -24
- airbyte_cdk/connector_builder/README.md +53 -0
- airbyte_cdk/connector_builder/connector_builder_handler.py +37 -11
- airbyte_cdk/connector_builder/main.py +45 -13
- airbyte_cdk/connector_builder/message_grouper.py +189 -50
- airbyte_cdk/connector_builder/models.py +3 -2
- airbyte_cdk/destinations/__init__.py +4 -3
- airbyte_cdk/destinations/destination.py +54 -20
- airbyte_cdk/destinations/vector_db_based/README.md +37 -0
- airbyte_cdk/destinations/vector_db_based/config.py +40 -17
- airbyte_cdk/destinations/vector_db_based/document_processor.py +56 -17
- airbyte_cdk/destinations/vector_db_based/embedder.py +57 -15
- airbyte_cdk/destinations/vector_db_based/test_utils.py +14 -4
- airbyte_cdk/destinations/vector_db_based/utils.py +8 -2
- airbyte_cdk/destinations/vector_db_based/writer.py +24 -5
- airbyte_cdk/entrypoint.py +153 -44
- airbyte_cdk/exception_handler.py +21 -3
- airbyte_cdk/logger.py +30 -44
- airbyte_cdk/models/__init__.py +13 -2
- airbyte_cdk/models/airbyte_protocol.py +86 -1
- airbyte_cdk/models/airbyte_protocol_serializers.py +44 -0
- airbyte_cdk/models/file_transfer_record_message.py +13 -0
- airbyte_cdk/models/well_known_types.py +1 -1
- airbyte_cdk/sources/__init__.py +5 -1
- airbyte_cdk/sources/abstract_source.py +125 -79
- airbyte_cdk/sources/concurrent_source/__init__.py +7 -2
- airbyte_cdk/sources/concurrent_source/concurrent_read_processor.py +102 -36
- airbyte_cdk/sources/concurrent_source/concurrent_source.py +29 -36
- airbyte_cdk/sources/concurrent_source/concurrent_source_adapter.py +94 -10
- airbyte_cdk/sources/concurrent_source/stream_thread_exception.py +25 -0
- airbyte_cdk/sources/concurrent_source/thread_pool_manager.py +20 -14
- airbyte_cdk/sources/config.py +3 -2
- airbyte_cdk/sources/connector_state_manager.py +49 -83
- airbyte_cdk/sources/declarative/async_job/job.py +52 -0
- airbyte_cdk/sources/declarative/async_job/job_orchestrator.py +497 -0
- airbyte_cdk/sources/declarative/async_job/job_tracker.py +75 -0
- airbyte_cdk/sources/declarative/async_job/repository.py +35 -0
- airbyte_cdk/sources/declarative/async_job/status.py +24 -0
- airbyte_cdk/sources/declarative/async_job/timer.py +39 -0
- airbyte_cdk/sources/declarative/auth/__init__.py +2 -3
- airbyte_cdk/sources/declarative/auth/declarative_authenticator.py +3 -1
- airbyte_cdk/sources/declarative/auth/jwt.py +191 -0
- airbyte_cdk/sources/declarative/auth/oauth.py +60 -20
- airbyte_cdk/sources/declarative/auth/selective_authenticator.py +10 -2
- airbyte_cdk/sources/declarative/auth/token.py +28 -10
- airbyte_cdk/sources/declarative/auth/token_provider.py +9 -8
- airbyte_cdk/sources/declarative/checks/check_stream.py +16 -8
- airbyte_cdk/sources/declarative/checks/connection_checker.py +4 -2
- airbyte_cdk/sources/declarative/concurrency_level/__init__.py +7 -0
- airbyte_cdk/sources/declarative/concurrency_level/concurrency_level.py +50 -0
- airbyte_cdk/sources/declarative/concurrent_declarative_source.py +490 -0
- airbyte_cdk/sources/declarative/datetime/datetime_parser.py +4 -0
- airbyte_cdk/sources/declarative/datetime/min_max_datetime.py +26 -6
- airbyte_cdk/sources/declarative/declarative_component_schema.yaml +1185 -85
- airbyte_cdk/sources/declarative/declarative_source.py +5 -2
- airbyte_cdk/sources/declarative/declarative_stream.py +95 -9
- airbyte_cdk/sources/declarative/decoders/__init__.py +23 -2
- airbyte_cdk/sources/declarative/decoders/composite_raw_decoder.py +97 -0
- airbyte_cdk/sources/declarative/decoders/decoder.py +11 -4
- airbyte_cdk/sources/declarative/decoders/json_decoder.py +92 -5
- airbyte_cdk/sources/declarative/decoders/noop_decoder.py +21 -0
- airbyte_cdk/sources/declarative/decoders/pagination_decoder_decorator.py +39 -0
- airbyte_cdk/sources/declarative/decoders/xml_decoder.py +98 -0
- airbyte_cdk/sources/declarative/extractors/__init__.py +12 -1
- airbyte_cdk/sources/declarative/extractors/dpath_extractor.py +29 -24
- airbyte_cdk/sources/declarative/extractors/http_selector.py +4 -5
- airbyte_cdk/sources/declarative/extractors/record_extractor.py +2 -3
- airbyte_cdk/sources/declarative/extractors/record_filter.py +63 -8
- airbyte_cdk/sources/declarative/extractors/record_selector.py +85 -26
- airbyte_cdk/sources/declarative/extractors/response_to_file_extractor.py +177 -0
- airbyte_cdk/sources/declarative/extractors/type_transformer.py +55 -0
- airbyte_cdk/sources/declarative/incremental/__init__.py +31 -3
- airbyte_cdk/sources/declarative/incremental/concurrent_partition_cursor.py +346 -0
- airbyte_cdk/sources/declarative/incremental/datetime_based_cursor.py +156 -48
- airbyte_cdk/sources/declarative/incremental/declarative_cursor.py +13 -0
- airbyte_cdk/sources/declarative/incremental/global_substream_cursor.py +350 -0
- airbyte_cdk/sources/declarative/incremental/per_partition_cursor.py +173 -74
- airbyte_cdk/sources/declarative/incremental/per_partition_with_global.py +200 -0
- airbyte_cdk/sources/declarative/incremental/resumable_full_refresh_cursor.py +122 -0
- airbyte_cdk/sources/declarative/interpolation/filters.py +27 -1
- airbyte_cdk/sources/declarative/interpolation/interpolated_boolean.py +23 -5
- airbyte_cdk/sources/declarative/interpolation/interpolated_mapping.py +12 -8
- airbyte_cdk/sources/declarative/interpolation/interpolated_nested_mapping.py +13 -6
- airbyte_cdk/sources/declarative/interpolation/interpolated_string.py +21 -6
- airbyte_cdk/sources/declarative/interpolation/interpolation.py +9 -3
- airbyte_cdk/sources/declarative/interpolation/jinja.py +72 -37
- airbyte_cdk/sources/declarative/interpolation/macros.py +72 -17
- airbyte_cdk/sources/declarative/manifest_declarative_source.py +193 -52
- airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py +98 -0
- airbyte_cdk/sources/declarative/migrations/state_migration.py +24 -0
- airbyte_cdk/sources/declarative/models/__init__.py +1 -1
- airbyte_cdk/sources/declarative/models/declarative_component_schema.py +1319 -603
- airbyte_cdk/sources/declarative/parsers/custom_exceptions.py +2 -2
- airbyte_cdk/sources/declarative/parsers/manifest_component_transformer.py +26 -4
- airbyte_cdk/sources/declarative/parsers/manifest_reference_resolver.py +26 -15
- airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py +1759 -225
- airbyte_cdk/sources/declarative/partition_routers/__init__.py +24 -4
- airbyte_cdk/sources/declarative/partition_routers/async_job_partition_router.py +65 -0
- airbyte_cdk/sources/declarative/partition_routers/cartesian_product_stream_slicer.py +176 -0
- airbyte_cdk/sources/declarative/partition_routers/list_partition_router.py +39 -9
- airbyte_cdk/sources/declarative/partition_routers/partition_router.py +62 -0
- airbyte_cdk/sources/declarative/partition_routers/single_partition_router.py +15 -3
- airbyte_cdk/sources/declarative/partition_routers/substream_partition_router.py +222 -39
- airbyte_cdk/sources/declarative/requesters/error_handlers/__init__.py +19 -5
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/__init__.py +3 -1
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/constant_backoff_strategy.py +19 -7
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/exponential_backoff_strategy.py +19 -7
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/header_helper.py +4 -2
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/wait_time_from_header_backoff_strategy.py +41 -9
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/wait_until_time_from_header_backoff_strategy.py +29 -14
- airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategy.py +5 -13
- airbyte_cdk/sources/declarative/requesters/error_handlers/composite_error_handler.py +32 -16
- airbyte_cdk/sources/declarative/requesters/error_handlers/default_error_handler.py +46 -56
- airbyte_cdk/sources/declarative/requesters/error_handlers/default_http_response_filter.py +40 -0
- airbyte_cdk/sources/declarative/requesters/error_handlers/error_handler.py +6 -32
- airbyte_cdk/sources/declarative/requesters/error_handlers/http_response_filter.py +119 -41
- airbyte_cdk/sources/declarative/requesters/http_job_repository.py +228 -0
- airbyte_cdk/sources/declarative/requesters/http_requester.py +98 -344
- airbyte_cdk/sources/declarative/requesters/paginators/__init__.py +14 -3
- airbyte_cdk/sources/declarative/requesters/paginators/default_paginator.py +105 -46
- airbyte_cdk/sources/declarative/requesters/paginators/no_pagination.py +14 -8
- airbyte_cdk/sources/declarative/requesters/paginators/paginator.py +19 -8
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/__init__.py +9 -3
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/cursor_pagination_strategy.py +53 -21
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/offset_increment.py +42 -19
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/page_increment.py +25 -12
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/pagination_strategy.py +13 -10
- airbyte_cdk/sources/declarative/requesters/paginators/strategies/stop_condition.py +26 -13
- airbyte_cdk/sources/declarative/requesters/request_options/__init__.py +15 -2
- airbyte_cdk/sources/declarative/requesters/request_options/datetime_based_request_options_provider.py +91 -0
- airbyte_cdk/sources/declarative/requesters/request_options/default_request_options_provider.py +60 -0
- airbyte_cdk/sources/declarative/requesters/request_options/interpolated_nested_request_input_provider.py +31 -14
- airbyte_cdk/sources/declarative/requesters/request_options/interpolated_request_input_provider.py +27 -15
- airbyte_cdk/sources/declarative/requesters/request_options/interpolated_request_options_provider.py +63 -10
- airbyte_cdk/sources/declarative/requesters/request_options/request_options_provider.py +1 -1
- airbyte_cdk/sources/declarative/requesters/requester.py +9 -17
- airbyte_cdk/sources/declarative/resolvers/__init__.py +41 -0
- airbyte_cdk/sources/declarative/resolvers/components_resolver.py +55 -0
- airbyte_cdk/sources/declarative/resolvers/config_components_resolver.py +136 -0
- airbyte_cdk/sources/declarative/resolvers/http_components_resolver.py +112 -0
- airbyte_cdk/sources/declarative/retrievers/__init__.py +6 -2
- airbyte_cdk/sources/declarative/retrievers/async_retriever.py +100 -0
- airbyte_cdk/sources/declarative/retrievers/retriever.py +1 -3
- airbyte_cdk/sources/declarative/retrievers/simple_retriever.py +229 -73
- airbyte_cdk/sources/declarative/schema/__init__.py +14 -1
- airbyte_cdk/sources/declarative/schema/default_schema_loader.py +5 -3
- airbyte_cdk/sources/declarative/schema/dynamic_schema_loader.py +236 -0
- airbyte_cdk/sources/declarative/schema/json_file_schema_loader.py +8 -8
- airbyte_cdk/sources/declarative/spec/spec.py +12 -5
- airbyte_cdk/sources/declarative/stream_slicers/__init__.py +1 -2
- airbyte_cdk/sources/declarative/stream_slicers/declarative_partition_generator.py +88 -0
- airbyte_cdk/sources/declarative/stream_slicers/stream_slicer.py +9 -14
- airbyte_cdk/sources/declarative/transformations/add_fields.py +19 -11
- airbyte_cdk/sources/declarative/transformations/flatten_fields.py +52 -0
- airbyte_cdk/sources/declarative/transformations/keys_replace_transformation.py +61 -0
- airbyte_cdk/sources/declarative/transformations/keys_to_lower_transformation.py +22 -0
- airbyte_cdk/sources/declarative/transformations/keys_to_snake_transformation.py +68 -0
- airbyte_cdk/sources/declarative/transformations/remove_fields.py +13 -10
- airbyte_cdk/sources/declarative/transformations/transformation.py +5 -5
- airbyte_cdk/sources/declarative/types.py +19 -110
- airbyte_cdk/sources/declarative/yaml_declarative_source.py +31 -10
- airbyte_cdk/sources/embedded/base_integration.py +16 -5
- airbyte_cdk/sources/embedded/catalog.py +16 -4
- airbyte_cdk/sources/embedded/runner.py +19 -3
- airbyte_cdk/sources/embedded/tools.py +5 -2
- airbyte_cdk/sources/file_based/README.md +152 -0
- airbyte_cdk/sources/file_based/__init__.py +24 -0
- airbyte_cdk/sources/file_based/availability_strategy/__init__.py +9 -2
- airbyte_cdk/sources/file_based/availability_strategy/abstract_file_based_availability_strategy.py +22 -6
- airbyte_cdk/sources/file_based/availability_strategy/default_file_based_availability_strategy.py +46 -10
- airbyte_cdk/sources/file_based/config/abstract_file_based_spec.py +47 -10
- airbyte_cdk/sources/file_based/config/avro_format.py +2 -1
- airbyte_cdk/sources/file_based/config/csv_format.py +29 -10
- airbyte_cdk/sources/file_based/config/excel_format.py +18 -0
- airbyte_cdk/sources/file_based/config/file_based_stream_config.py +16 -4
- airbyte_cdk/sources/file_based/config/jsonl_format.py +2 -1
- airbyte_cdk/sources/file_based/config/parquet_format.py +2 -1
- airbyte_cdk/sources/file_based/config/unstructured_format.py +13 -5
- airbyte_cdk/sources/file_based/discovery_policy/__init__.py +6 -2
- airbyte_cdk/sources/file_based/discovery_policy/abstract_discovery_policy.py +2 -4
- airbyte_cdk/sources/file_based/discovery_policy/default_discovery_policy.py +7 -2
- airbyte_cdk/sources/file_based/exceptions.py +18 -15
- airbyte_cdk/sources/file_based/file_based_source.py +140 -33
- airbyte_cdk/sources/file_based/file_based_stream_reader.py +69 -5
- airbyte_cdk/sources/file_based/file_types/__init__.py +14 -1
- airbyte_cdk/sources/file_based/file_types/avro_parser.py +75 -24
- airbyte_cdk/sources/file_based/file_types/csv_parser.py +116 -34
- airbyte_cdk/sources/file_based/file_types/excel_parser.py +196 -0
- airbyte_cdk/sources/file_based/file_types/file_transfer.py +37 -0
- airbyte_cdk/sources/file_based/file_types/file_type_parser.py +4 -1
- airbyte_cdk/sources/file_based/file_types/jsonl_parser.py +24 -8
- airbyte_cdk/sources/file_based/file_types/parquet_parser.py +60 -18
- airbyte_cdk/sources/file_based/file_types/unstructured_parser.py +141 -41
- airbyte_cdk/sources/file_based/remote_file.py +1 -1
- airbyte_cdk/sources/file_based/schema_helpers.py +38 -10
- airbyte_cdk/sources/file_based/schema_validation_policies/__init__.py +3 -1
- airbyte_cdk/sources/file_based/schema_validation_policies/abstract_schema_validation_policy.py +3 -1
- airbyte_cdk/sources/file_based/schema_validation_policies/default_schema_validation_policies.py +16 -5
- airbyte_cdk/sources/file_based/stream/abstract_file_based_stream.py +50 -13
- airbyte_cdk/sources/file_based/stream/concurrent/adapters.py +67 -27
- airbyte_cdk/sources/file_based/stream/concurrent/cursor/__init__.py +5 -1
- airbyte_cdk/sources/file_based/stream/concurrent/cursor/abstract_concurrent_file_based_cursor.py +14 -23
- airbyte_cdk/sources/file_based/stream/concurrent/cursor/file_based_concurrent_cursor.py +54 -18
- airbyte_cdk/sources/file_based/stream/concurrent/cursor/file_based_final_state_cursor.py +21 -9
- airbyte_cdk/sources/file_based/stream/cursor/abstract_file_based_cursor.py +3 -1
- airbyte_cdk/sources/file_based/stream/cursor/default_file_based_cursor.py +27 -10
- airbyte_cdk/sources/file_based/stream/default_file_based_stream.py +147 -45
- airbyte_cdk/sources/http_logger.py +8 -3
- airbyte_cdk/sources/message/__init__.py +7 -1
- airbyte_cdk/sources/message/repository.py +18 -4
- airbyte_cdk/sources/source.py +42 -38
- airbyte_cdk/sources/streams/__init__.py +2 -2
- airbyte_cdk/sources/streams/availability_strategy.py +54 -3
- airbyte_cdk/sources/streams/call_rate.py +64 -21
- airbyte_cdk/sources/streams/checkpoint/__init__.py +26 -0
- airbyte_cdk/sources/streams/checkpoint/checkpoint_reader.py +335 -0
- airbyte_cdk/sources/{declarative/incremental → streams/checkpoint}/cursor.py +17 -14
- airbyte_cdk/sources/streams/checkpoint/per_partition_key_serializer.py +22 -0
- airbyte_cdk/sources/streams/checkpoint/resumable_full_refresh_cursor.py +51 -0
- airbyte_cdk/sources/streams/checkpoint/substream_resumable_full_refresh_cursor.py +110 -0
- airbyte_cdk/sources/streams/concurrent/README.md +7 -0
- airbyte_cdk/sources/streams/concurrent/abstract_stream.py +7 -2
- airbyte_cdk/sources/streams/concurrent/adapters.py +84 -75
- airbyte_cdk/sources/streams/concurrent/availability_strategy.py +30 -2
- airbyte_cdk/sources/streams/concurrent/cursor.py +298 -42
- airbyte_cdk/sources/streams/concurrent/default_stream.py +12 -3
- airbyte_cdk/sources/streams/concurrent/exceptions.py +3 -0
- airbyte_cdk/sources/streams/concurrent/helpers.py +14 -3
- airbyte_cdk/sources/streams/concurrent/partition_enqueuer.py +12 -3
- airbyte_cdk/sources/streams/concurrent/partition_reader.py +10 -3
- airbyte_cdk/sources/streams/concurrent/partitions/partition.py +1 -16
- airbyte_cdk/sources/streams/concurrent/partitions/stream_slicer.py +21 -0
- airbyte_cdk/sources/streams/concurrent/partitions/types.py +15 -5
- airbyte_cdk/sources/streams/concurrent/state_converters/abstract_stream_state_converter.py +109 -17
- airbyte_cdk/sources/streams/concurrent/state_converters/datetime_stream_state_converter.py +90 -72
- airbyte_cdk/sources/streams/core.py +412 -87
- airbyte_cdk/sources/streams/http/__init__.py +2 -1
- airbyte_cdk/sources/streams/http/availability_strategy.py +12 -101
- airbyte_cdk/sources/streams/http/error_handlers/__init__.py +22 -0
- airbyte_cdk/sources/streams/http/error_handlers/backoff_strategy.py +28 -0
- airbyte_cdk/sources/streams/http/error_handlers/default_backoff_strategy.py +17 -0
- airbyte_cdk/sources/streams/http/error_handlers/default_error_mapping.py +86 -0
- airbyte_cdk/sources/streams/http/error_handlers/error_handler.py +42 -0
- airbyte_cdk/sources/streams/http/error_handlers/error_message_parser.py +19 -0
- airbyte_cdk/sources/streams/http/error_handlers/http_status_error_handler.py +110 -0
- airbyte_cdk/sources/streams/http/error_handlers/json_error_message_parser.py +52 -0
- airbyte_cdk/sources/streams/http/error_handlers/response_models.py +65 -0
- airbyte_cdk/sources/streams/http/exceptions.py +27 -7
- airbyte_cdk/sources/streams/http/http.py +369 -246
- airbyte_cdk/sources/streams/http/http_client.py +531 -0
- airbyte_cdk/sources/streams/http/rate_limiting.py +76 -12
- airbyte_cdk/sources/streams/http/requests_native_auth/abstract_oauth.py +28 -9
- airbyte_cdk/sources/streams/http/requests_native_auth/abstract_token.py +2 -1
- airbyte_cdk/sources/streams/http/requests_native_auth/oauth.py +90 -35
- airbyte_cdk/sources/streams/http/requests_native_auth/token.py +13 -3
- airbyte_cdk/sources/types.py +154 -0
- airbyte_cdk/sources/utils/record_helper.py +36 -21
- airbyte_cdk/sources/utils/schema_helpers.py +13 -6
- airbyte_cdk/sources/utils/slice_logger.py +4 -1
- airbyte_cdk/sources/utils/transform.py +54 -20
- airbyte_cdk/sql/_util/hashing.py +34 -0
- airbyte_cdk/sql/_util/name_normalizers.py +92 -0
- airbyte_cdk/sql/constants.py +32 -0
- airbyte_cdk/sql/exceptions.py +235 -0
- airbyte_cdk/sql/secrets.py +123 -0
- airbyte_cdk/sql/shared/__init__.py +15 -0
- airbyte_cdk/sql/shared/catalog_providers.py +145 -0
- airbyte_cdk/sql/shared/sql_processor.py +786 -0
- airbyte_cdk/sql/types.py +160 -0
- airbyte_cdk/test/catalog_builder.py +70 -18
- airbyte_cdk/test/entrypoint_wrapper.py +117 -42
- airbyte_cdk/test/mock_http/__init__.py +1 -1
- airbyte_cdk/test/mock_http/matcher.py +6 -0
- airbyte_cdk/test/mock_http/mocker.py +57 -10
- airbyte_cdk/test/mock_http/request.py +19 -3
- airbyte_cdk/test/mock_http/response.py +3 -1
- airbyte_cdk/test/mock_http/response_builder.py +32 -16
- airbyte_cdk/test/state_builder.py +18 -10
- airbyte_cdk/test/utils/__init__.py +1 -0
- airbyte_cdk/test/utils/data.py +24 -0
- airbyte_cdk/test/utils/http_mocking.py +16 -0
- airbyte_cdk/test/utils/manifest_only_fixtures.py +60 -0
- airbyte_cdk/test/utils/reading.py +26 -0
- airbyte_cdk/utils/__init__.py +2 -1
- airbyte_cdk/utils/airbyte_secrets_utils.py +5 -3
- airbyte_cdk/utils/analytics_message.py +10 -2
- airbyte_cdk/utils/datetime_format_inferrer.py +4 -1
- airbyte_cdk/utils/event_timing.py +10 -10
- airbyte_cdk/utils/mapping_helpers.py +3 -1
- airbyte_cdk/utils/message_utils.py +20 -11
- airbyte_cdk/utils/print_buffer.py +75 -0
- airbyte_cdk/utils/schema_inferrer.py +198 -28
- airbyte_cdk/utils/slice_hasher.py +30 -0
- airbyte_cdk/utils/spec_schema_transformations.py +6 -3
- airbyte_cdk/utils/stream_status_utils.py +8 -1
- airbyte_cdk/utils/traced_exception.py +61 -21
- airbyte_cdk-6.17.1.dev0.dist-info/METADATA +109 -0
- airbyte_cdk-6.17.1.dev0.dist-info/RECORD +350 -0
- {airbyte_cdk-0.72.1.dist-info → airbyte_cdk-6.17.1.dev0.dist-info}/WHEEL +1 -2
- airbyte_cdk-6.17.1.dev0.dist-info/entry_points.txt +3 -0
- airbyte_cdk/sources/declarative/create_partial.py +0 -92
- airbyte_cdk/sources/declarative/parsers/class_types_registry.py +0 -102
- airbyte_cdk/sources/declarative/parsers/default_implementation_registry.py +0 -64
- airbyte_cdk/sources/declarative/requesters/error_handlers/response_action.py +0 -16
- airbyte_cdk/sources/declarative/requesters/error_handlers/response_status.py +0 -68
- airbyte_cdk/sources/declarative/stream_slicers/cartesian_product_stream_slicer.py +0 -114
- airbyte_cdk/sources/deprecated/base_source.py +0 -94
- airbyte_cdk/sources/deprecated/client.py +0 -99
- airbyte_cdk/sources/singer/__init__.py +0 -8
- airbyte_cdk/sources/singer/singer_helpers.py +0 -304
- airbyte_cdk/sources/singer/source.py +0 -186
- airbyte_cdk/sources/streams/concurrent/partitions/record.py +0 -23
- airbyte_cdk/sources/streams/http/auth/__init__.py +0 -17
- airbyte_cdk/sources/streams/http/auth/core.py +0 -29
- airbyte_cdk/sources/streams/http/auth/oauth.py +0 -113
- airbyte_cdk/sources/streams/http/auth/token.py +0 -47
- airbyte_cdk/sources/streams/utils/stream_helper.py +0 -40
- airbyte_cdk/sources/utils/catalog_helpers.py +0 -22
- airbyte_cdk/sources/utils/schema_models.py +0 -84
- airbyte_cdk-0.72.1.dist-info/METADATA +0 -243
- airbyte_cdk-0.72.1.dist-info/RECORD +0 -466
- airbyte_cdk-0.72.1.dist-info/top_level.txt +0 -3
- source_declarative_manifest/main.py +0 -29
- unit_tests/connector_builder/__init__.py +0 -3
- unit_tests/connector_builder/test_connector_builder_handler.py +0 -871
- unit_tests/connector_builder/test_message_grouper.py +0 -713
- unit_tests/connector_builder/utils.py +0 -27
- unit_tests/destinations/test_destination.py +0 -243
- unit_tests/singer/test_singer_helpers.py +0 -56
- unit_tests/singer/test_singer_source.py +0 -112
- unit_tests/sources/__init__.py +0 -0
- unit_tests/sources/concurrent_source/__init__.py +0 -3
- unit_tests/sources/concurrent_source/test_concurrent_source_adapter.py +0 -106
- unit_tests/sources/declarative/__init__.py +0 -3
- unit_tests/sources/declarative/auth/__init__.py +0 -3
- unit_tests/sources/declarative/auth/test_oauth.py +0 -331
- unit_tests/sources/declarative/auth/test_selective_authenticator.py +0 -39
- unit_tests/sources/declarative/auth/test_session_token_auth.py +0 -182
- unit_tests/sources/declarative/auth/test_token_auth.py +0 -200
- unit_tests/sources/declarative/auth/test_token_provider.py +0 -73
- unit_tests/sources/declarative/checks/__init__.py +0 -3
- unit_tests/sources/declarative/checks/test_check_stream.py +0 -146
- unit_tests/sources/declarative/decoders/__init__.py +0 -0
- unit_tests/sources/declarative/decoders/test_json_decoder.py +0 -16
- unit_tests/sources/declarative/external_component.py +0 -13
- unit_tests/sources/declarative/extractors/__init__.py +0 -3
- unit_tests/sources/declarative/extractors/test_dpath_extractor.py +0 -55
- unit_tests/sources/declarative/extractors/test_record_filter.py +0 -55
- unit_tests/sources/declarative/extractors/test_record_selector.py +0 -179
- unit_tests/sources/declarative/incremental/__init__.py +0 -0
- unit_tests/sources/declarative/incremental/test_datetime_based_cursor.py +0 -860
- unit_tests/sources/declarative/incremental/test_per_partition_cursor.py +0 -406
- unit_tests/sources/declarative/incremental/test_per_partition_cursor_integration.py +0 -332
- unit_tests/sources/declarative/interpolation/__init__.py +0 -3
- unit_tests/sources/declarative/interpolation/test_filters.py +0 -80
- unit_tests/sources/declarative/interpolation/test_interpolated_boolean.py +0 -40
- unit_tests/sources/declarative/interpolation/test_interpolated_mapping.py +0 -35
- unit_tests/sources/declarative/interpolation/test_interpolated_nested_mapping.py +0 -45
- unit_tests/sources/declarative/interpolation/test_interpolated_string.py +0 -25
- unit_tests/sources/declarative/interpolation/test_jinja.py +0 -240
- unit_tests/sources/declarative/interpolation/test_macros.py +0 -73
- unit_tests/sources/declarative/parsers/__init__.py +0 -3
- unit_tests/sources/declarative/parsers/test_manifest_component_transformer.py +0 -406
- unit_tests/sources/declarative/parsers/test_manifest_reference_resolver.py +0 -139
- unit_tests/sources/declarative/parsers/test_model_to_component_factory.py +0 -1847
- unit_tests/sources/declarative/parsers/testing_components.py +0 -36
- unit_tests/sources/declarative/partition_routers/__init__.py +0 -3
- unit_tests/sources/declarative/partition_routers/test_list_partition_router.py +0 -155
- unit_tests/sources/declarative/partition_routers/test_single_partition_router.py +0 -14
- unit_tests/sources/declarative/partition_routers/test_substream_partition_router.py +0 -404
- unit_tests/sources/declarative/requesters/__init__.py +0 -3
- unit_tests/sources/declarative/requesters/error_handlers/__init__.py +0 -3
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/__init__.py +0 -3
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_constant_backoff.py +0 -34
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_exponential_backoff.py +0 -36
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_header_helper.py +0 -38
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_wait_time_from_header.py +0 -35
- unit_tests/sources/declarative/requesters/error_handlers/backoff_strategies/test_wait_until_time_from_header.py +0 -64
- unit_tests/sources/declarative/requesters/error_handlers/test_composite_error_handler.py +0 -213
- unit_tests/sources/declarative/requesters/error_handlers/test_default_error_handler.py +0 -178
- unit_tests/sources/declarative/requesters/error_handlers/test_http_response_filter.py +0 -121
- unit_tests/sources/declarative/requesters/error_handlers/test_response_status.py +0 -44
- unit_tests/sources/declarative/requesters/paginators/__init__.py +0 -3
- unit_tests/sources/declarative/requesters/paginators/test_cursor_pagination_strategy.py +0 -64
- unit_tests/sources/declarative/requesters/paginators/test_default_paginator.py +0 -313
- unit_tests/sources/declarative/requesters/paginators/test_no_paginator.py +0 -12
- unit_tests/sources/declarative/requesters/paginators/test_offset_increment.py +0 -58
- unit_tests/sources/declarative/requesters/paginators/test_page_increment.py +0 -70
- unit_tests/sources/declarative/requesters/paginators/test_request_option.py +0 -43
- unit_tests/sources/declarative/requesters/paginators/test_stop_condition.py +0 -105
- unit_tests/sources/declarative/requesters/request_options/__init__.py +0 -3
- unit_tests/sources/declarative/requesters/request_options/test_interpolated_request_options_provider.py +0 -101
- unit_tests/sources/declarative/requesters/test_http_requester.py +0 -974
- unit_tests/sources/declarative/requesters/test_interpolated_request_input_provider.py +0 -32
- unit_tests/sources/declarative/retrievers/__init__.py +0 -3
- unit_tests/sources/declarative/retrievers/test_simple_retriever.py +0 -542
- unit_tests/sources/declarative/schema/__init__.py +0 -6
- unit_tests/sources/declarative/schema/source_test/SourceTest.py +0 -8
- unit_tests/sources/declarative/schema/source_test/__init__.py +0 -3
- unit_tests/sources/declarative/schema/test_default_schema_loader.py +0 -32
- unit_tests/sources/declarative/schema/test_inline_schema_loader.py +0 -19
- unit_tests/sources/declarative/schema/test_json_file_schema_loader.py +0 -26
- unit_tests/sources/declarative/states/__init__.py +0 -3
- unit_tests/sources/declarative/stream_slicers/__init__.py +0 -3
- unit_tests/sources/declarative/stream_slicers/test_cartesian_product_stream_slicer.py +0 -225
- unit_tests/sources/declarative/test_create_partial.py +0 -83
- unit_tests/sources/declarative/test_declarative_stream.py +0 -103
- unit_tests/sources/declarative/test_manifest_declarative_source.py +0 -1260
- unit_tests/sources/declarative/test_types.py +0 -39
- unit_tests/sources/declarative/test_yaml_declarative_source.py +0 -148
- unit_tests/sources/file_based/__init__.py +0 -0
- unit_tests/sources/file_based/availability_strategy/__init__.py +0 -0
- unit_tests/sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py +0 -100
- unit_tests/sources/file_based/config/__init__.py +0 -0
- unit_tests/sources/file_based/config/test_abstract_file_based_spec.py +0 -28
- unit_tests/sources/file_based/config/test_csv_format.py +0 -34
- unit_tests/sources/file_based/config/test_file_based_stream_config.py +0 -84
- unit_tests/sources/file_based/discovery_policy/__init__.py +0 -0
- unit_tests/sources/file_based/discovery_policy/test_default_discovery_policy.py +0 -31
- unit_tests/sources/file_based/file_types/__init__.py +0 -0
- unit_tests/sources/file_based/file_types/test_avro_parser.py +0 -243
- unit_tests/sources/file_based/file_types/test_csv_parser.py +0 -546
- unit_tests/sources/file_based/file_types/test_jsonl_parser.py +0 -158
- unit_tests/sources/file_based/file_types/test_parquet_parser.py +0 -274
- unit_tests/sources/file_based/file_types/test_unstructured_parser.py +0 -593
- unit_tests/sources/file_based/helpers.py +0 -70
- unit_tests/sources/file_based/in_memory_files_source.py +0 -211
- unit_tests/sources/file_based/scenarios/__init__.py +0 -0
- unit_tests/sources/file_based/scenarios/avro_scenarios.py +0 -744
- unit_tests/sources/file_based/scenarios/check_scenarios.py +0 -220
- unit_tests/sources/file_based/scenarios/concurrent_incremental_scenarios.py +0 -2844
- unit_tests/sources/file_based/scenarios/csv_scenarios.py +0 -3105
- unit_tests/sources/file_based/scenarios/file_based_source_builder.py +0 -91
- unit_tests/sources/file_based/scenarios/incremental_scenarios.py +0 -1926
- unit_tests/sources/file_based/scenarios/jsonl_scenarios.py +0 -930
- unit_tests/sources/file_based/scenarios/parquet_scenarios.py +0 -754
- unit_tests/sources/file_based/scenarios/scenario_builder.py +0 -234
- unit_tests/sources/file_based/scenarios/unstructured_scenarios.py +0 -608
- unit_tests/sources/file_based/scenarios/user_input_schema_scenarios.py +0 -746
- unit_tests/sources/file_based/scenarios/validation_policy_scenarios.py +0 -726
- unit_tests/sources/file_based/stream/__init__.py +0 -0
- unit_tests/sources/file_based/stream/concurrent/__init__.py +0 -0
- unit_tests/sources/file_based/stream/concurrent/test_adapters.py +0 -362
- unit_tests/sources/file_based/stream/concurrent/test_file_based_concurrent_cursor.py +0 -458
- unit_tests/sources/file_based/stream/test_default_file_based_cursor.py +0 -310
- unit_tests/sources/file_based/stream/test_default_file_based_stream.py +0 -244
- unit_tests/sources/file_based/test_file_based_scenarios.py +0 -320
- unit_tests/sources/file_based/test_file_based_stream_reader.py +0 -272
- unit_tests/sources/file_based/test_scenarios.py +0 -253
- unit_tests/sources/file_based/test_schema_helpers.py +0 -346
- unit_tests/sources/fixtures/__init__.py +0 -3
- unit_tests/sources/fixtures/source_test_fixture.py +0 -153
- unit_tests/sources/message/__init__.py +0 -0
- unit_tests/sources/message/test_repository.py +0 -153
- unit_tests/sources/streams/__init__.py +0 -0
- unit_tests/sources/streams/concurrent/__init__.py +0 -3
- unit_tests/sources/streams/concurrent/scenarios/__init__.py +0 -3
- unit_tests/sources/streams/concurrent/scenarios/incremental_scenarios.py +0 -250
- unit_tests/sources/streams/concurrent/scenarios/stream_facade_builder.py +0 -140
- unit_tests/sources/streams/concurrent/scenarios/stream_facade_scenarios.py +0 -452
- unit_tests/sources/streams/concurrent/scenarios/test_concurrent_scenarios.py +0 -76
- unit_tests/sources/streams/concurrent/scenarios/thread_based_concurrent_stream_scenarios.py +0 -418
- unit_tests/sources/streams/concurrent/scenarios/thread_based_concurrent_stream_source_builder.py +0 -142
- unit_tests/sources/streams/concurrent/scenarios/utils.py +0 -55
- unit_tests/sources/streams/concurrent/test_adapters.py +0 -380
- unit_tests/sources/streams/concurrent/test_concurrent_read_processor.py +0 -684
- unit_tests/sources/streams/concurrent/test_cursor.py +0 -139
- unit_tests/sources/streams/concurrent/test_datetime_state_converter.py +0 -369
- unit_tests/sources/streams/concurrent/test_default_stream.py +0 -197
- unit_tests/sources/streams/concurrent/test_partition_enqueuer.py +0 -90
- unit_tests/sources/streams/concurrent/test_partition_reader.py +0 -67
- unit_tests/sources/streams/concurrent/test_thread_pool_manager.py +0 -106
- unit_tests/sources/streams/http/__init__.py +0 -0
- unit_tests/sources/streams/http/auth/__init__.py +0 -0
- unit_tests/sources/streams/http/auth/test_auth.py +0 -173
- unit_tests/sources/streams/http/requests_native_auth/__init__.py +0 -0
- unit_tests/sources/streams/http/requests_native_auth/test_requests_native_auth.py +0 -423
- unit_tests/sources/streams/http/test_availability_strategy.py +0 -180
- unit_tests/sources/streams/http/test_http.py +0 -635
- unit_tests/sources/streams/test_availability_strategy.py +0 -70
- unit_tests/sources/streams/test_call_rate.py +0 -300
- unit_tests/sources/streams/test_stream_read.py +0 -405
- unit_tests/sources/streams/test_streams_core.py +0 -184
- unit_tests/sources/test_abstract_source.py +0 -1442
- unit_tests/sources/test_concurrent_source.py +0 -112
- unit_tests/sources/test_config.py +0 -92
- unit_tests/sources/test_connector_state_manager.py +0 -482
- unit_tests/sources/test_http_logger.py +0 -252
- unit_tests/sources/test_integration_source.py +0 -86
- unit_tests/sources/test_source.py +0 -684
- unit_tests/sources/test_source_read.py +0 -460
- unit_tests/test/__init__.py +0 -0
- unit_tests/test/mock_http/__init__.py +0 -0
- unit_tests/test/mock_http/test_matcher.py +0 -53
- unit_tests/test/mock_http/test_mocker.py +0 -214
- unit_tests/test/mock_http/test_request.py +0 -117
- unit_tests/test/mock_http/test_response_builder.py +0 -177
- unit_tests/test/test_entrypoint_wrapper.py +0 -240
- unit_tests/utils/__init__.py +0 -0
- unit_tests/utils/test_datetime_format_inferrer.py +0 -60
- unit_tests/utils/test_mapping_helpers.py +0 -54
- unit_tests/utils/test_message_utils.py +0 -91
- unit_tests/utils/test_rate_limiting.py +0 -26
- unit_tests/utils/test_schema_inferrer.py +0 -202
- unit_tests/utils/test_secret_utils.py +0 -135
- unit_tests/utils/test_stream_status_utils.py +0 -61
- unit_tests/utils/test_traced_exception.py +0 -107
- /airbyte_cdk/sources/{deprecated → declarative/async_job}/__init__.py +0 -0
- {source_declarative_manifest → airbyte_cdk/sources/declarative/migrations}/__init__.py +0 -0
- {unit_tests/destinations → airbyte_cdk/sql}/__init__.py +0 -0
- {unit_tests/singer → airbyte_cdk/sql/_util}/__init__.py +0 -0
- {airbyte_cdk-0.72.1.dist-info → airbyte_cdk-6.17.1.dev0.dist-info}/LICENSE.txt +0 -0
@@ -0,0 +1,490 @@
|
|
1
|
+
#
|
2
|
+
# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
|
3
|
+
#
|
4
|
+
|
5
|
+
import logging
|
6
|
+
from typing import Any, Generic, Iterator, List, Mapping, Optional, Tuple
|
7
|
+
|
8
|
+
from airbyte_cdk.models import (
|
9
|
+
AirbyteCatalog,
|
10
|
+
AirbyteMessage,
|
11
|
+
AirbyteStateMessage,
|
12
|
+
ConfiguredAirbyteCatalog,
|
13
|
+
)
|
14
|
+
from airbyte_cdk.sources.concurrent_source.concurrent_source import ConcurrentSource
|
15
|
+
from airbyte_cdk.sources.connector_state_manager import ConnectorStateManager
|
16
|
+
from airbyte_cdk.sources.declarative.concurrency_level import ConcurrencyLevel
|
17
|
+
from airbyte_cdk.sources.declarative.declarative_stream import DeclarativeStream
|
18
|
+
from airbyte_cdk.sources.declarative.extractors import RecordSelector
|
19
|
+
from airbyte_cdk.sources.declarative.extractors.record_filter import (
|
20
|
+
ClientSideIncrementalRecordFilterDecorator,
|
21
|
+
)
|
22
|
+
from airbyte_cdk.sources.declarative.incremental.datetime_based_cursor import DatetimeBasedCursor
|
23
|
+
from airbyte_cdk.sources.declarative.incremental.per_partition_with_global import (
|
24
|
+
PerPartitionWithGlobalCursor,
|
25
|
+
)
|
26
|
+
from airbyte_cdk.sources.declarative.interpolation import InterpolatedString
|
27
|
+
from airbyte_cdk.sources.declarative.manifest_declarative_source import ManifestDeclarativeSource
|
28
|
+
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
|
29
|
+
ConcurrencyLevel as ConcurrencyLevelModel,
|
30
|
+
)
|
31
|
+
from airbyte_cdk.sources.declarative.models.declarative_component_schema import (
|
32
|
+
DatetimeBasedCursor as DatetimeBasedCursorModel,
|
33
|
+
)
|
34
|
+
from airbyte_cdk.sources.declarative.parsers.model_to_component_factory import (
|
35
|
+
ModelToComponentFactory,
|
36
|
+
)
|
37
|
+
from airbyte_cdk.sources.declarative.requesters import HttpRequester
|
38
|
+
from airbyte_cdk.sources.declarative.retrievers import SimpleRetriever
|
39
|
+
from airbyte_cdk.sources.declarative.stream_slicers.declarative_partition_generator import (
|
40
|
+
DeclarativePartitionFactory,
|
41
|
+
StreamSlicerPartitionGenerator,
|
42
|
+
)
|
43
|
+
from airbyte_cdk.sources.declarative.transformations.add_fields import AddFields
|
44
|
+
from airbyte_cdk.sources.declarative.types import ConnectionDefinition
|
45
|
+
from airbyte_cdk.sources.source import TState
|
46
|
+
from airbyte_cdk.sources.streams import Stream
|
47
|
+
from airbyte_cdk.sources.streams.concurrent.abstract_stream import AbstractStream
|
48
|
+
from airbyte_cdk.sources.streams.concurrent.availability_strategy import (
|
49
|
+
AlwaysAvailableAvailabilityStrategy,
|
50
|
+
)
|
51
|
+
from airbyte_cdk.sources.streams.concurrent.cursor import FinalStateCursor
|
52
|
+
from airbyte_cdk.sources.streams.concurrent.default_stream import DefaultStream
|
53
|
+
from airbyte_cdk.sources.streams.concurrent.helpers import get_primary_key_from_stream
|
54
|
+
|
55
|
+
|
56
|
+
class ConcurrentDeclarativeSource(ManifestDeclarativeSource, Generic[TState]):
|
57
|
+
# By default, we defer to a value of 2. A value lower than than could cause a PartitionEnqueuer to be stuck in a state of deadlock
|
58
|
+
# because it has hit the limit of futures but not partition reader is consuming them.
|
59
|
+
_LOWEST_SAFE_CONCURRENCY_LEVEL = 2
|
60
|
+
|
61
|
+
def __init__(
|
62
|
+
self,
|
63
|
+
catalog: Optional[ConfiguredAirbyteCatalog],
|
64
|
+
config: Optional[Mapping[str, Any]],
|
65
|
+
state: TState,
|
66
|
+
source_config: ConnectionDefinition,
|
67
|
+
debug: bool = False,
|
68
|
+
emit_connector_builder_messages: bool = False,
|
69
|
+
component_factory: Optional[ModelToComponentFactory] = None,
|
70
|
+
**kwargs: Any,
|
71
|
+
) -> None:
|
72
|
+
# To reduce the complexity of the concurrent framework, we are not enabling RFR with synthetic
|
73
|
+
# cursors. We do this by no longer automatically instantiating RFR cursors when converting
|
74
|
+
# the declarative models into runtime components. Concurrent sources will continue to checkpoint
|
75
|
+
# incremental streams running in full refresh.
|
76
|
+
component_factory = component_factory or ModelToComponentFactory(
|
77
|
+
emit_connector_builder_messages=emit_connector_builder_messages,
|
78
|
+
disable_resumable_full_refresh=True,
|
79
|
+
)
|
80
|
+
|
81
|
+
super().__init__(
|
82
|
+
source_config=source_config,
|
83
|
+
debug=debug,
|
84
|
+
emit_connector_builder_messages=emit_connector_builder_messages,
|
85
|
+
component_factory=component_factory,
|
86
|
+
)
|
87
|
+
|
88
|
+
# todo: We could remove state from initialization. Now that streams are grouped during the read(), a source
|
89
|
+
# no longer needs to store the original incoming state. But maybe there's an edge case?
|
90
|
+
self._state = state
|
91
|
+
|
92
|
+
concurrency_level_from_manifest = self._source_config.get("concurrency_level")
|
93
|
+
if concurrency_level_from_manifest:
|
94
|
+
concurrency_level_component = self._constructor.create_component(
|
95
|
+
model_type=ConcurrencyLevelModel,
|
96
|
+
component_definition=concurrency_level_from_manifest,
|
97
|
+
config=config or {},
|
98
|
+
)
|
99
|
+
if not isinstance(concurrency_level_component, ConcurrencyLevel):
|
100
|
+
raise ValueError(
|
101
|
+
f"Expected to generate a ConcurrencyLevel component, but received {concurrency_level_component.__class__}"
|
102
|
+
)
|
103
|
+
|
104
|
+
concurrency_level = concurrency_level_component.get_concurrency_level()
|
105
|
+
initial_number_of_partitions_to_generate = max(
|
106
|
+
concurrency_level // 2, 1
|
107
|
+
) # Partition_generation iterates using range based on this value. If this is floored to zero we end up in a dead lock during start up
|
108
|
+
else:
|
109
|
+
concurrency_level = self._LOWEST_SAFE_CONCURRENCY_LEVEL
|
110
|
+
initial_number_of_partitions_to_generate = self._LOWEST_SAFE_CONCURRENCY_LEVEL // 2
|
111
|
+
|
112
|
+
self._concurrent_source = ConcurrentSource.create(
|
113
|
+
num_workers=concurrency_level,
|
114
|
+
initial_number_of_partitions_to_generate=initial_number_of_partitions_to_generate,
|
115
|
+
logger=self.logger,
|
116
|
+
slice_logger=self._slice_logger,
|
117
|
+
message_repository=self.message_repository,
|
118
|
+
)
|
119
|
+
|
120
|
+
def read(
|
121
|
+
self,
|
122
|
+
logger: logging.Logger,
|
123
|
+
config: Mapping[str, Any],
|
124
|
+
catalog: ConfiguredAirbyteCatalog,
|
125
|
+
state: Optional[List[AirbyteStateMessage]] = None,
|
126
|
+
) -> Iterator[AirbyteMessage]:
|
127
|
+
concurrent_streams, _ = self._group_streams(config=config)
|
128
|
+
|
129
|
+
# ConcurrentReadProcessor pops streams that are finished being read so before syncing, the names of
|
130
|
+
# the concurrent streams must be saved so that they can be removed from the catalog before starting
|
131
|
+
# synchronous streams
|
132
|
+
if len(concurrent_streams) > 0:
|
133
|
+
concurrent_stream_names = set(
|
134
|
+
[concurrent_stream.name for concurrent_stream in concurrent_streams]
|
135
|
+
)
|
136
|
+
|
137
|
+
selected_concurrent_streams = self._select_streams(
|
138
|
+
streams=concurrent_streams, configured_catalog=catalog
|
139
|
+
)
|
140
|
+
# It would appear that passing in an empty set of streams causes an infinite loop in ConcurrentReadProcessor.
|
141
|
+
# This is also evident in concurrent_source_adapter.py so I'll leave this out of scope to fix for now
|
142
|
+
if selected_concurrent_streams:
|
143
|
+
yield from self._concurrent_source.read(selected_concurrent_streams)
|
144
|
+
|
145
|
+
# Sync all streams that are not concurrent compatible. We filter out concurrent streams because the
|
146
|
+
# existing AbstractSource.read() implementation iterates over the catalog when syncing streams. Many
|
147
|
+
# of which were already synced using the Concurrent CDK
|
148
|
+
filtered_catalog = self._remove_concurrent_streams_from_catalog(
|
149
|
+
catalog=catalog, concurrent_stream_names=concurrent_stream_names
|
150
|
+
)
|
151
|
+
else:
|
152
|
+
filtered_catalog = catalog
|
153
|
+
|
154
|
+
yield from super().read(logger, config, filtered_catalog, state)
|
155
|
+
|
156
|
+
def discover(self, logger: logging.Logger, config: Mapping[str, Any]) -> AirbyteCatalog:
|
157
|
+
concurrent_streams, synchronous_streams = self._group_streams(config=config)
|
158
|
+
return AirbyteCatalog(
|
159
|
+
streams=[
|
160
|
+
stream.as_airbyte_stream() for stream in concurrent_streams + synchronous_streams
|
161
|
+
]
|
162
|
+
)
|
163
|
+
|
164
|
+
def streams(self, config: Mapping[str, Any]) -> List[Stream]:
|
165
|
+
"""
|
166
|
+
The `streams` method is used as part of the AbstractSource in the following cases:
|
167
|
+
* ConcurrentDeclarativeSource.check -> ManifestDeclarativeSource.check -> AbstractSource.check -> DeclarativeSource.check_connection -> CheckStream.check_connection -> streams
|
168
|
+
* ConcurrentDeclarativeSource.read -> AbstractSource.read -> streams (note that we filter for a specific catalog which excludes concurrent streams so not all streams actually read from all the streams returned by `streams`)
|
169
|
+
Note that `super.streams(config)` is also called when splitting the streams between concurrent or not in `_group_streams`.
|
170
|
+
|
171
|
+
In both case, we will assume that calling the DeclarativeStream is perfectly fine as the result for these is the same regardless of if it is a DeclarativeStream or a DefaultStream (concurrent). This should simply be removed once we have moved away from the mentioned code paths above.
|
172
|
+
"""
|
173
|
+
return super().streams(config)
|
174
|
+
|
175
|
+
def _group_streams(
|
176
|
+
self, config: Mapping[str, Any]
|
177
|
+
) -> Tuple[List[AbstractStream], List[Stream]]:
|
178
|
+
concurrent_streams: List[AbstractStream] = []
|
179
|
+
synchronous_streams: List[Stream] = []
|
180
|
+
|
181
|
+
state_manager = ConnectorStateManager(state=self._state) # type: ignore # state is always in the form of List[AirbyteStateMessage]. The ConnectorStateManager should use generics, but this can be done later
|
182
|
+
|
183
|
+
# Combine streams and dynamic_streams. Note: both cannot be empty at the same time,
|
184
|
+
# and this is validated during the initialization of the source.
|
185
|
+
streams = self._stream_configs(self._source_config) + self._dynamic_stream_configs(
|
186
|
+
self._source_config, config
|
187
|
+
)
|
188
|
+
|
189
|
+
name_to_stream_mapping = {stream["name"]: stream for stream in streams}
|
190
|
+
|
191
|
+
for declarative_stream in self.streams(config=config):
|
192
|
+
# Some low-code sources use a combination of DeclarativeStream and regular Python streams. We can't inspect
|
193
|
+
# these legacy Python streams the way we do low-code streams to determine if they are concurrent compatible,
|
194
|
+
# so we need to treat them as synchronous
|
195
|
+
if isinstance(declarative_stream, DeclarativeStream) and (
|
196
|
+
name_to_stream_mapping[declarative_stream.name]["retriever"]["type"]
|
197
|
+
== "SimpleRetriever"
|
198
|
+
or name_to_stream_mapping[declarative_stream.name]["retriever"]["type"]
|
199
|
+
== "AsyncRetriever"
|
200
|
+
):
|
201
|
+
incremental_sync_component_definition = name_to_stream_mapping[
|
202
|
+
declarative_stream.name
|
203
|
+
].get("incremental_sync")
|
204
|
+
|
205
|
+
partition_router_component_definition = (
|
206
|
+
name_to_stream_mapping[declarative_stream.name]
|
207
|
+
.get("retriever", {})
|
208
|
+
.get("partition_router")
|
209
|
+
)
|
210
|
+
is_without_partition_router_or_cursor = not bool(
|
211
|
+
incremental_sync_component_definition
|
212
|
+
) and not bool(partition_router_component_definition)
|
213
|
+
|
214
|
+
is_substream_without_incremental = (
|
215
|
+
partition_router_component_definition
|
216
|
+
and not incremental_sync_component_definition
|
217
|
+
)
|
218
|
+
|
219
|
+
if self._is_datetime_incremental_without_partition_routing(
|
220
|
+
declarative_stream, incremental_sync_component_definition
|
221
|
+
):
|
222
|
+
stream_state = state_manager.get_stream_state(
|
223
|
+
stream_name=declarative_stream.name, namespace=declarative_stream.namespace
|
224
|
+
)
|
225
|
+
|
226
|
+
cursor = self._constructor.create_concurrent_cursor_from_datetime_based_cursor(
|
227
|
+
state_manager=state_manager,
|
228
|
+
model_type=DatetimeBasedCursorModel,
|
229
|
+
component_definition=incremental_sync_component_definition, # type: ignore # Not None because of the if condition above
|
230
|
+
stream_name=declarative_stream.name,
|
231
|
+
stream_namespace=declarative_stream.namespace,
|
232
|
+
config=config or {},
|
233
|
+
stream_state=stream_state,
|
234
|
+
)
|
235
|
+
|
236
|
+
retriever = declarative_stream.retriever
|
237
|
+
|
238
|
+
# This is an optimization so that we don't invoke any cursor or state management flows within the
|
239
|
+
# low-code framework because state management is handled through the ConcurrentCursor.
|
240
|
+
if declarative_stream and isinstance(retriever, SimpleRetriever):
|
241
|
+
# Also a temporary hack. In the legacy Stream implementation, as part of the read,
|
242
|
+
# set_initial_state() is called to instantiate incoming state on the cursor. Although we no
|
243
|
+
# longer rely on the legacy low-code cursor for concurrent checkpointing, low-code components
|
244
|
+
# like StopConditionPaginationStrategyDecorator and ClientSideIncrementalRecordFilterDecorator
|
245
|
+
# still rely on a DatetimeBasedCursor that is properly initialized with state.
|
246
|
+
if retriever.cursor:
|
247
|
+
retriever.cursor.set_initial_state(stream_state=stream_state)
|
248
|
+
# We zero it out here, but since this is a cursor reference, the state is still properly
|
249
|
+
# instantiated for the other components that reference it
|
250
|
+
retriever.cursor = None
|
251
|
+
|
252
|
+
partition_generator = StreamSlicerPartitionGenerator(
|
253
|
+
DeclarativePartitionFactory(
|
254
|
+
declarative_stream.name,
|
255
|
+
declarative_stream.get_json_schema(),
|
256
|
+
retriever,
|
257
|
+
self.message_repository,
|
258
|
+
),
|
259
|
+
cursor,
|
260
|
+
)
|
261
|
+
|
262
|
+
concurrent_streams.append(
|
263
|
+
DefaultStream(
|
264
|
+
partition_generator=partition_generator,
|
265
|
+
name=declarative_stream.name,
|
266
|
+
json_schema=declarative_stream.get_json_schema(),
|
267
|
+
availability_strategy=AlwaysAvailableAvailabilityStrategy(),
|
268
|
+
primary_key=get_primary_key_from_stream(declarative_stream.primary_key),
|
269
|
+
cursor_field=cursor.cursor_field.cursor_field_key
|
270
|
+
if hasattr(cursor, "cursor_field")
|
271
|
+
and hasattr(
|
272
|
+
cursor.cursor_field, "cursor_field_key"
|
273
|
+
) # FIXME this will need to be updated once we do the per partition
|
274
|
+
else None,
|
275
|
+
logger=self.logger,
|
276
|
+
cursor=cursor,
|
277
|
+
)
|
278
|
+
)
|
279
|
+
elif (
|
280
|
+
is_substream_without_incremental or is_without_partition_router_or_cursor
|
281
|
+
) and hasattr(declarative_stream.retriever, "stream_slicer"):
|
282
|
+
partition_generator = StreamSlicerPartitionGenerator(
|
283
|
+
DeclarativePartitionFactory(
|
284
|
+
declarative_stream.name,
|
285
|
+
declarative_stream.get_json_schema(),
|
286
|
+
declarative_stream.retriever,
|
287
|
+
self.message_repository,
|
288
|
+
),
|
289
|
+
declarative_stream.retriever.stream_slicer,
|
290
|
+
)
|
291
|
+
|
292
|
+
final_state_cursor = FinalStateCursor(
|
293
|
+
stream_name=declarative_stream.name,
|
294
|
+
stream_namespace=declarative_stream.namespace,
|
295
|
+
message_repository=self.message_repository,
|
296
|
+
)
|
297
|
+
|
298
|
+
concurrent_streams.append(
|
299
|
+
DefaultStream(
|
300
|
+
partition_generator=partition_generator,
|
301
|
+
name=declarative_stream.name,
|
302
|
+
json_schema=declarative_stream.get_json_schema(),
|
303
|
+
availability_strategy=AlwaysAvailableAvailabilityStrategy(),
|
304
|
+
primary_key=get_primary_key_from_stream(declarative_stream.primary_key),
|
305
|
+
cursor_field=None,
|
306
|
+
logger=self.logger,
|
307
|
+
cursor=final_state_cursor,
|
308
|
+
)
|
309
|
+
)
|
310
|
+
elif (
|
311
|
+
incremental_sync_component_definition
|
312
|
+
and incremental_sync_component_definition.get("type", "")
|
313
|
+
== DatetimeBasedCursorModel.__name__
|
314
|
+
and self._stream_supports_concurrent_partition_processing(
|
315
|
+
declarative_stream=declarative_stream
|
316
|
+
)
|
317
|
+
and hasattr(declarative_stream.retriever, "stream_slicer")
|
318
|
+
and isinstance(
|
319
|
+
declarative_stream.retriever.stream_slicer, PerPartitionWithGlobalCursor
|
320
|
+
)
|
321
|
+
):
|
322
|
+
stream_state = state_manager.get_stream_state(
|
323
|
+
stream_name=declarative_stream.name, namespace=declarative_stream.namespace
|
324
|
+
)
|
325
|
+
partition_router = declarative_stream.retriever.stream_slicer._partition_router
|
326
|
+
|
327
|
+
cursor = self._constructor.create_concurrent_cursor_from_perpartition_cursor(
|
328
|
+
state_manager=state_manager,
|
329
|
+
model_type=DatetimeBasedCursorModel,
|
330
|
+
component_definition=incremental_sync_component_definition,
|
331
|
+
stream_name=declarative_stream.name,
|
332
|
+
stream_namespace=declarative_stream.namespace,
|
333
|
+
config=config or {},
|
334
|
+
stream_state=stream_state,
|
335
|
+
partition_router=partition_router,
|
336
|
+
)
|
337
|
+
|
338
|
+
retriever = declarative_stream.retriever
|
339
|
+
|
340
|
+
# This is an optimization so that we don't invoke any cursor or state management flows within the
|
341
|
+
# low-code framework because state management is handled through the ConcurrentCursor.
|
342
|
+
if declarative_stream and isinstance(retriever, SimpleRetriever):
|
343
|
+
# Also a temporary hack. In the legacy Stream implementation, as part of the read,
|
344
|
+
# set_initial_state() is called to instantiate incoming state on the cursor. Although we no
|
345
|
+
# longer rely on the legacy low-code cursor for concurrent checkpointing, low-code components
|
346
|
+
# like StopConditionPaginationStrategyDecorator and ClientSideIncrementalRecordFilterDecorator
|
347
|
+
# still rely on a DatetimeBasedCursor that is properly initialized with state.
|
348
|
+
if retriever.cursor:
|
349
|
+
retriever.cursor.set_initial_state(stream_state=stream_state)
|
350
|
+
# We zero it out here, but since this is a cursor reference, the state is still properly
|
351
|
+
# instantiated for the other components that reference it
|
352
|
+
retriever.cursor = None
|
353
|
+
|
354
|
+
partition_generator = StreamSlicerPartitionGenerator(
|
355
|
+
DeclarativePartitionFactory(
|
356
|
+
declarative_stream.name,
|
357
|
+
declarative_stream.get_json_schema(),
|
358
|
+
retriever,
|
359
|
+
self.message_repository,
|
360
|
+
),
|
361
|
+
cursor,
|
362
|
+
)
|
363
|
+
|
364
|
+
concurrent_streams.append(
|
365
|
+
DefaultStream(
|
366
|
+
partition_generator=partition_generator,
|
367
|
+
name=declarative_stream.name,
|
368
|
+
json_schema=declarative_stream.get_json_schema(),
|
369
|
+
availability_strategy=AlwaysAvailableAvailabilityStrategy(),
|
370
|
+
primary_key=get_primary_key_from_stream(declarative_stream.primary_key),
|
371
|
+
cursor_field=cursor.cursor_field.cursor_field_key,
|
372
|
+
logger=self.logger,
|
373
|
+
cursor=cursor,
|
374
|
+
)
|
375
|
+
)
|
376
|
+
else:
|
377
|
+
synchronous_streams.append(declarative_stream)
|
378
|
+
else:
|
379
|
+
synchronous_streams.append(declarative_stream)
|
380
|
+
|
381
|
+
return concurrent_streams, synchronous_streams
|
382
|
+
|
383
|
+
def _is_datetime_incremental_without_partition_routing(
|
384
|
+
self,
|
385
|
+
declarative_stream: DeclarativeStream,
|
386
|
+
incremental_sync_component_definition: Mapping[str, Any] | None,
|
387
|
+
) -> bool:
|
388
|
+
return (
|
389
|
+
incremental_sync_component_definition is not None
|
390
|
+
and bool(incremental_sync_component_definition)
|
391
|
+
and incremental_sync_component_definition.get("type", "")
|
392
|
+
== DatetimeBasedCursorModel.__name__
|
393
|
+
and self._stream_supports_concurrent_partition_processing(
|
394
|
+
declarative_stream=declarative_stream
|
395
|
+
)
|
396
|
+
and hasattr(declarative_stream.retriever, "stream_slicer")
|
397
|
+
and isinstance(declarative_stream.retriever.stream_slicer, DatetimeBasedCursor)
|
398
|
+
)
|
399
|
+
|
400
|
+
def _stream_supports_concurrent_partition_processing(
|
401
|
+
self, declarative_stream: DeclarativeStream
|
402
|
+
) -> bool:
|
403
|
+
"""
|
404
|
+
Many connectors make use of stream_state during interpolation on a per-partition basis under the assumption that
|
405
|
+
state is updated sequentially. Because the concurrent CDK engine processes different partitions in parallel,
|
406
|
+
stream_state is no longer a thread-safe interpolation context. It would be a race condition because a cursor's
|
407
|
+
stream_state can be updated in any order depending on which stream partition's finish first.
|
408
|
+
|
409
|
+
We should start to move away from depending on the value of stream_state for low-code components that operate
|
410
|
+
per-partition, but we need to gate this otherwise some connectors will be blocked from publishing. See the
|
411
|
+
cdk-migrations.md for the full list of connectors.
|
412
|
+
"""
|
413
|
+
|
414
|
+
if isinstance(declarative_stream.retriever, SimpleRetriever) and isinstance(
|
415
|
+
declarative_stream.retriever.requester, HttpRequester
|
416
|
+
):
|
417
|
+
http_requester = declarative_stream.retriever.requester
|
418
|
+
if "stream_state" in http_requester._path.string:
|
419
|
+
self.logger.warning(
|
420
|
+
f"Low-code stream '{declarative_stream.name}' uses interpolation of stream_state in the HttpRequester which is not thread-safe. Defaulting to synchronous processing"
|
421
|
+
)
|
422
|
+
return False
|
423
|
+
|
424
|
+
request_options_provider = http_requester._request_options_provider
|
425
|
+
if request_options_provider.request_options_contain_stream_state():
|
426
|
+
self.logger.warning(
|
427
|
+
f"Low-code stream '{declarative_stream.name}' uses interpolation of stream_state in the HttpRequester which is not thread-safe. Defaulting to synchronous processing"
|
428
|
+
)
|
429
|
+
return False
|
430
|
+
|
431
|
+
record_selector = declarative_stream.retriever.record_selector
|
432
|
+
if isinstance(record_selector, RecordSelector):
|
433
|
+
if (
|
434
|
+
record_selector.record_filter
|
435
|
+
and not isinstance(
|
436
|
+
record_selector.record_filter, ClientSideIncrementalRecordFilterDecorator
|
437
|
+
)
|
438
|
+
and "stream_state" in record_selector.record_filter.condition
|
439
|
+
):
|
440
|
+
self.logger.warning(
|
441
|
+
f"Low-code stream '{declarative_stream.name}' uses interpolation of stream_state in the RecordFilter which is not thread-safe. Defaulting to synchronous processing"
|
442
|
+
)
|
443
|
+
return False
|
444
|
+
|
445
|
+
for add_fields in [
|
446
|
+
transformation
|
447
|
+
for transformation in record_selector.transformations
|
448
|
+
if isinstance(transformation, AddFields)
|
449
|
+
]:
|
450
|
+
for field in add_fields.fields:
|
451
|
+
if isinstance(field.value, str) and "stream_state" in field.value:
|
452
|
+
self.logger.warning(
|
453
|
+
f"Low-code stream '{declarative_stream.name}' uses interpolation of stream_state in the AddFields which is not thread-safe. Defaulting to synchronous processing"
|
454
|
+
)
|
455
|
+
return False
|
456
|
+
if (
|
457
|
+
isinstance(field.value, InterpolatedString)
|
458
|
+
and "stream_state" in field.value.string
|
459
|
+
):
|
460
|
+
self.logger.warning(
|
461
|
+
f"Low-code stream '{declarative_stream.name}' uses interpolation of stream_state in the AddFields which is not thread-safe. Defaulting to synchronous processing"
|
462
|
+
)
|
463
|
+
return False
|
464
|
+
return True
|
465
|
+
|
466
|
+
@staticmethod
|
467
|
+
def _select_streams(
|
468
|
+
streams: List[AbstractStream], configured_catalog: ConfiguredAirbyteCatalog
|
469
|
+
) -> List[AbstractStream]:
|
470
|
+
stream_name_to_instance: Mapping[str, AbstractStream] = {s.name: s for s in streams}
|
471
|
+
abstract_streams: List[AbstractStream] = []
|
472
|
+
for configured_stream in configured_catalog.streams:
|
473
|
+
stream_instance = stream_name_to_instance.get(configured_stream.stream.name)
|
474
|
+
if stream_instance:
|
475
|
+
abstract_streams.append(stream_instance)
|
476
|
+
|
477
|
+
return abstract_streams
|
478
|
+
|
479
|
+
@staticmethod
|
480
|
+
def _remove_concurrent_streams_from_catalog(
|
481
|
+
catalog: ConfiguredAirbyteCatalog,
|
482
|
+
concurrent_stream_names: set[str],
|
483
|
+
) -> ConfiguredAirbyteCatalog:
|
484
|
+
return ConfiguredAirbyteCatalog(
|
485
|
+
streams=[
|
486
|
+
stream
|
487
|
+
for stream in catalog.streams
|
488
|
+
if stream.stream.name not in concurrent_stream_names
|
489
|
+
]
|
490
|
+
)
|
@@ -27,6 +27,8 @@ class DatetimeParser:
|
|
27
27
|
# See https://stackoverflow.com/a/4974930
|
28
28
|
if format == "%s":
|
29
29
|
return datetime.datetime.fromtimestamp(int(date), tz=datetime.timezone.utc)
|
30
|
+
elif format == "%s_as_float":
|
31
|
+
return datetime.datetime.fromtimestamp(float(date), tz=datetime.timezone.utc)
|
30
32
|
elif format == "%ms":
|
31
33
|
return self._UNIX_EPOCH + datetime.timedelta(milliseconds=int(date))
|
32
34
|
|
@@ -41,6 +43,8 @@ class DatetimeParser:
|
|
41
43
|
# See https://stackoverflow.com/a/4974930
|
42
44
|
if format == "%s":
|
43
45
|
return str(int(dt.timestamp()))
|
46
|
+
if format == "%s_as_float":
|
47
|
+
return str(float(dt.timestamp()))
|
44
48
|
if format == "%ms":
|
45
49
|
# timstamp() returns a float representing the number of seconds since the unix epoch
|
46
50
|
return str(int(dt.timestamp() * 1000))
|
@@ -32,7 +32,7 @@ class MinMaxDatetime:
|
|
32
32
|
parameters: InitVar[Mapping[str, Any]]
|
33
33
|
# datetime_format is a unique case where we inherit it from the parent if it is not specified before using the default value
|
34
34
|
# which is why we need dedicated getter/setter methods and private dataclass field
|
35
|
-
datetime_format: str
|
35
|
+
datetime_format: str
|
36
36
|
_datetime_format: str = field(init=False, repr=False, default="")
|
37
37
|
min_datetime: Union[InterpolatedString, str] = ""
|
38
38
|
max_datetime: Union[InterpolatedString, str] = ""
|
@@ -40,10 +40,20 @@ class MinMaxDatetime:
|
|
40
40
|
def __post_init__(self, parameters: Mapping[str, Any]) -> None:
|
41
41
|
self.datetime = InterpolatedString.create(self.datetime, parameters=parameters or {})
|
42
42
|
self._parser = DatetimeParser()
|
43
|
-
self.min_datetime =
|
44
|
-
|
43
|
+
self.min_datetime = (
|
44
|
+
InterpolatedString.create(self.min_datetime, parameters=parameters) # type: ignore [assignment] # expression has type "InterpolatedString | None", variable has type "InterpolatedString | str"
|
45
|
+
if self.min_datetime
|
46
|
+
else None
|
47
|
+
) # type: ignore
|
48
|
+
self.max_datetime = (
|
49
|
+
InterpolatedString.create(self.max_datetime, parameters=parameters) # type: ignore [assignment] # expression has type "InterpolatedString | None", variable has type "InterpolatedString | str"
|
50
|
+
if self.max_datetime
|
51
|
+
else None
|
52
|
+
) # type: ignore
|
45
53
|
|
46
|
-
def get_datetime(
|
54
|
+
def get_datetime(
|
55
|
+
self, config: Mapping[str, Any], **additional_parameters: Mapping[str, Any]
|
56
|
+
) -> dt.datetime:
|
47
57
|
"""
|
48
58
|
Evaluates and returns the datetime
|
49
59
|
:param config: The user-provided configuration as specified by the source's spec
|
@@ -55,7 +65,15 @@ class MinMaxDatetime:
|
|
55
65
|
if not datetime_format:
|
56
66
|
datetime_format = "%Y-%m-%dT%H:%M:%S.%f%z"
|
57
67
|
|
58
|
-
time = self._parser.parse(
|
68
|
+
time = self._parser.parse(
|
69
|
+
str(
|
70
|
+
self.datetime.eval( # type: ignore[union-attr] # str has no attribute "eval"
|
71
|
+
config,
|
72
|
+
**additional_parameters,
|
73
|
+
)
|
74
|
+
),
|
75
|
+
datetime_format,
|
76
|
+
) # type: ignore # datetime is always cast to an interpolated string
|
59
77
|
|
60
78
|
if self.min_datetime:
|
61
79
|
min_time = str(self.min_datetime.eval(config, **additional_parameters)) # type: ignore # min_datetime is always cast to an interpolated string
|
@@ -93,6 +111,8 @@ class MinMaxDatetime:
|
|
93
111
|
if isinstance(interpolated_string_or_min_max_datetime, InterpolatedString) or isinstance(
|
94
112
|
interpolated_string_or_min_max_datetime, str
|
95
113
|
):
|
96
|
-
return MinMaxDatetime(
|
114
|
+
return MinMaxDatetime( # type: ignore [call-arg]
|
115
|
+
datetime=interpolated_string_or_min_max_datetime, parameters=parameters
|
116
|
+
)
|
97
117
|
else:
|
98
118
|
return interpolated_string_or_min_max_datetime
|