Amazon SP: ref

Signed-off-by: Artem Inzhyyants <[email protected]>
airbytehq · Jan 8, 2025 · fecf6fa · fecf6fa
1 parent 1e646a2
commit fecf6fa
Show file tree

Hide file tree

Showing 9 changed files with 97 additions and 89 deletions.
diff --git a/...nnectors/source-amazon-seller-partner/source_amazon_seller_partner/components/__init__.py b/...nnectors/source-amazon-seller-partner/source_amazon_seller_partner/components/__init__.py
@@ -3,8 +3,16 @@
 
 from .decoder import GzipCsvDecoder
 
+from .type_transformers import LedgerDetailedViewReportsTypeTransformer, MerchantListingsFypReportTypeTransformer, FlatFileSettlementV2ReportsTypeTransformer, \
+    MerchantReportsTypeTransformer, SellerFeedbackReportsTypeTransformer
+
 __all__ = [
     "AmazonSPOauthAuthenticator",
     "AmazonSPWaitTimeFromHeaderBackoffStrategy",
-    "GzipCsvDecoder"
-]
+    "GzipCsvDecoder",
+    "LedgerDetailedViewReportsTypeTransformer",
+    "MerchantListingsFypReportTypeTransformer",
+    "FlatFileSettlementV2ReportsTypeTransformer",
+    "MerchantReportsTypeTransformer",
+    "SellerFeedbackReportsTypeTransformer"
+]
diff --git a/...s/connectors/source-amazon-seller-partner/source_amazon_seller_partner/components/auth.py b/...s/connectors/source-amazon-seller-partner/source_amazon_seller_partner/components/auth.py
@@ -8,13 +8,15 @@
 
 import backoff
 import requests
+
 from airbyte_cdk import InterpolatedString
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.sources.declarative.auth import DeclarativeOauth2Authenticator
 from airbyte_cdk.sources.streams.http.exceptions import DefaultBackoffException
 from airbyte_cdk.utils import AirbyteTracedException
 from airbyte_cdk.utils.airbyte_secrets_utils import add_to_secrets
 
+
 logger = logging.getLogger("airbyte")
 import pendulum
 

diff --git a/.../source-amazon-seller-partner/source_amazon_seller_partner/components/backoff_strategy.py b/.../source-amazon-seller-partner/source_amazon_seller_partner/components/backoff_strategy.py
@@ -4,6 +4,7 @@
 from typing import Optional, Union
 
 import requests
+
 from airbyte_cdk.sources.declarative.requesters.error_handlers.backoff_strategies.wait_time_from_header_backoff_strategy import (
     WaitTimeFromHeaderBackoffStrategy,
 )

diff --git a/...onnectors/source-amazon-seller-partner/source_amazon_seller_partner/components/decoder.py b/...onnectors/source-amazon-seller-partner/source_amazon_seller_partner/components/decoder.py
@@ -13,6 +13,7 @@
 
 from airbyte_cdk.sources.declarative.decoders.decoder import Decoder
 
+
 logger = logging.getLogger("airbyte")
 
 
@@ -94,7 +95,6 @@ class GetXmlBrowseTreeDataDecoder(Decoder):
     parameters: InitVar[Mapping[str, Any]]
     NORMALIZED_FIELD_NAMES = ["date", "rating", "comments", "response", "order_id", "rater_email"]
 
-
     def is_stream_response(self) -> bool:
         return False
 
@@ -119,4 +119,4 @@ def decode(self, response: requests.Response) -> Generator[MutableMapping[str, A
             logger.warning(f"Unable to parse the report for the stream, error: {str(e)}")
             parsed = {}
 
-        yield from parsed.get("Result", {}).get("Node", [])
+        yield from parsed.get("Result", {}).get("Node", [])
diff --git a/...source-amazon-seller-partner/source_amazon_seller_partner/components/type_transformers.py b/...source-amazon-seller-partner/source_amazon_seller_partner/components/type_transformers.py
@@ -1,14 +1,14 @@
-from typing import Dict, Any
+# Copyright (c) 2024 Airbyte, Inc., all rights reserved.
+
+from typing import Any, Dict
 
 import dateparser
 import pendulum
 
-from airbyte_cdk.sources.utils.transform import TypeTransformer, TransformConfig
-
+from airbyte_cdk.sources.utils.transform import TransformConfig, TypeTransformer
 
 
 class LedgerDetailedViewReportsTypeTransformer(TypeTransformer):
-
     def __init__(self, *args, **kwargs):
         config = TransformConfig.DefaultSchemaNormalization | TransformConfig.CustomSchemaNormalization
         super().__init__(config)
@@ -28,8 +28,8 @@ def transform_function(original_value: str, field_schema: Dict[str, Any]) -> str
 
         return transform_function
 
-class MerchantListingsFypReportTypeTransformer(TypeTransformer):
 
+class MerchantListingsFypReportTypeTransformer(TypeTransformer):
     def __init__(self, *args, **kwargs):
         config = TransformConfig.DefaultSchemaNormalization | TransformConfig.CustomSchemaNormalization
         super().__init__(config)
@@ -48,8 +48,8 @@ def transform_function(original_value: Any, field_schema: Dict[str, Any]) -> Any
 
         return transform_function
 
-class MerchantReportsTypeTransformer(TypeTransformer):
 
+class MerchantReportsTypeTransformer(TypeTransformer):
     def __init__(self, *args, **kwargs):
         config = TransformConfig.DefaultSchemaNormalization | TransformConfig.CustomSchemaNormalization
         super().__init__(config)
@@ -67,7 +67,6 @@ def transform_function(original_value: Any, field_schema: Dict[str, Any]) -> Any
         return transform_function
 
 
-
 class SellerFeedbackReportsTypeTransformer(TypeTransformer):
     MARKETPLACE_DATE_FORMAT_MAP = dict(
         # eu
@@ -120,7 +119,6 @@ def transform_function(original_value: Any, field_schema: Dict[str, Any]) -> Any
 
 
 class FlatFileSettlementV2ReportsTypeTransformer(TypeTransformer):
-
     def __init__(self, *args, **kwargs):
         config = TransformConfig.DefaultSchemaNormalization | TransformConfig.CustomSchemaNormalization
         super().__init__(config)
@@ -134,4 +132,3 @@ def transform_function(original_value: Any, field_schema: Dict[str, Any]) -> Any
             return original_value
 
         return transform_function
-