Added browse node method

Author: sergioteula

amazon/constant.py

@@ -3,6 +3,7 @@
 from paapi5_python_sdk.get_items_resource import GetItemsResource
 from paapi5_python_sdk.search_items_resource import SearchItemsResource
 from paapi5_python_sdk.get_variations_resource import GetVariationsResource
+from paapi5_python_sdk.get_browse_nodes_resource import GetBrowseNodesResource
 from paapi5_python_sdk.condition import Condition
 
 """Available regions for the Amazon API."""
@@ -228,3 +229,9 @@
     GetVariationsResource.VARIATIONSUMMARY_PRICE_LOWESTPRICE,
     GetVariationsResource.VARIATIONSUMMARY_VARIATIONDIMENSION
 ]
+
+"""Browse Node resources to get from Amazon API."""
+BROWSE_RESOURCES = [
+    GetBrowseNodesResource.ANCESTOR,
+    GetBrowseNodesResource.CHILDREN
+]

amazon/paapi.py

@@ -9,13 +9,15 @@
 from paapi5_python_sdk.get_items_request import GetItemsRequest
 from paapi5_python_sdk.search_items_request import SearchItemsRequest
 from paapi5_python_sdk.get_variations_request import GetVariationsRequest
+from paapi5_python_sdk.get_browse_nodes_request import GetBrowseNodesRequest
 from paapi5_python_sdk.partner_type import PartnerType
 from paapi5_python_sdk.rest import ApiException
 
 from amazon.constant import DOMAINS, REGIONS, CONDITION
 from amazon.constant import PRODUCT_RESOURCES, SEARCH_RESOURCES, VARIATION_RESOURCES
+from amazon.constant import BROWSE_RESOURCES
 from amazon.exception import AmazonException
-from amazon.parse import parse_product
+from amazon.parse import parse_product, AmazonBrowseNode, parse_browsenode
 from amazon.tools import get_asin, chunks
 
 import time
@@ -387,3 +389,56 @@ def get_variations(self, asin, item_count=10, item_page=1, items_per_page=10, co
             return results
         else:
             return None
+
+    def get_browsenodes(self, browse_nodes, async_req=False):
+        """Get browse nodes information from Amazon.
+
+        Args:
+            browse_nodes (list): List of strings containing the browse node ids.
+            async_req (bool, optional): Specify if a thread should be created to
+                run the request. Defaults to False.
+
+        Returns:
+            dict: A dictionary containing the browse node information.
+        """
+
+        if isinstance(browse_nodes, list) is False:
+            raise Exception('Browse nodes parameter should be a list')
+        elif not browse_nodes:
+            raise Exception('Browse nodes parameter can\'t be empty')
+
+        try:
+            request = GetBrowseNodesRequest(
+                partner_tag=self.tag,
+                partner_type=PartnerType.ASSOCIATES,
+                marketplace=self.marketplace,
+                browse_node_ids=browse_nodes,
+                languages_of_preference=None,
+                resources=BROWSE_RESOURCES)
+        except ValueError as e:
+            raise AmazonException("ValueError", e)
+
+        try:
+            self._throttle()
+            if async_req:
+                thread = self.api.get_browse_nodes(request, async_req=True)
+                response = thread.get()
+            else:
+                response = self.api.get_browse_nodes(request)
+        except ApiException as e:
+            raise AmazonException('ApiException', e)
+
+        try:
+            if response.browse_nodes_result is not None:
+                res = [AmazonBrowseNode(item) for item in response.browse_nodes_result.browse_nodes]
+                return parse_browsenode(res)
+            if response.errors is not None:
+                raise AmazonException(response.errors[0].code, response.errors[0].message)
+        except TypeError as e:
+            raise AmazonException("TypeError", e)
+        except ValueError as e:
+            raise AmazonException(ValueError, e)
+        except AmazonException as e:
+            raise AmazonException(e.status, e.reason)
+        except Exception as e:
+            raise AmazonException("General", e)

amazon/parse.py

@@ -6,6 +6,83 @@ class Class:
     pass
 
 
+class AmazonBrowseNode():
+    swagger_types = {
+        'ancestor': 'BrowseNodeAncestor',
+        'children': 'BrowseNodeChildren',
+        'context_free_name': 'str',
+        'display_name': 'str',
+        'id': 'str',
+        'is_root': 'bool',
+        'sales_rank': 'int'
+    }
+
+    def __init__(self, node):
+        self.ancestor = node.ancestor
+        self.children = node.children
+        self.context_free_name = node.context_free_name
+        self.display_name = node.display_name
+        self.id = node.id
+        self.is_root = node.is_root
+        self.sales_rank = node.sales_rank
+
+    def to_dict(self):
+        result = {}
+
+        for attr, _ in six.iteritems(self.swagger_types):
+            value = getattr(self, attr)
+            if isinstance(value, list):
+                result[attr] = list(map(
+                    lambda x: x.to_dict() if hasattr(x, "to_dict") else x,
+                    value
+                ))
+            elif hasattr(value, "to_dict"):
+                result[attr] = value.to_dict()
+            elif isinstance(value, dict):
+                result[attr] = dict(map(
+                    lambda item: (item[0], item[1].to_dict())
+                    if hasattr(item[1], "to_dict") else item,
+                    value.items()
+                ))
+            else:
+                result[attr] = value
+        if issubclass(BrowseNode, dict):
+            for key, value in self.items():
+                result[key] = value
+
+        return result
+
+    def to_str(self):
+        return pprint.pformat(self.to_dict())
+
+    def __repr__(self):
+        return self.to_str()
+
+    def __eq__(self, other):
+        if not isinstance(other, BrowseNode):
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other
+
+
+def parse_browsenode(browse_nodes_response):
+    """Parse browse node data and creates a dict.
+
+    Args:
+        browse_nodes_response (list): List of browse nodes responses.
+
+    Returns:
+        dict: Dict with browse node information.
+    """
+    mapped_response = {}
+    for browse_node in browse_nodes_response:
+        mapped_response[browse_node.id] = browse_node
+    return mapped_response
+
+
 def parse_product(item):
     """Parse item data and creates product instance.