added restricted_data_token

docs/client_usage.rst

@@ -16,5 +16,6 @@ All endpoint's clients have the following signature and default values:
         marketplace=Marketplaces.US, *,
         refresh_token=None,
         account='default',
-        credentials=None
+        credentials=None,
+        restricted_data_token=None
     )

docs/endpoints/orders.rst

@@ -6,3 +6,19 @@ Orders
 
 
 ..  autoclass:: sp_api.api.Orders
+
+
+
+Restricted Data Token
+=====================
+
+To use a restricted data token to access PII data, you can pass the token obtained from the Token endpoint to the client:
+
+..  code-block:: python
+
+    Orders(restricted_data_token='......').get_orders(...)
+
+
+
+
+

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='python-amazon-sp-api',
-    version='0.6.8',
+    version='0.7.0',
     install_requires=[
         "requests",
         "six>=1.15,<2",

sp_api/base/client.py

@@ -29,7 +29,8 @@ def __init__(
             *,
             refresh_token=None,
             account='default',
-            credentials=None
+            credentials=None,
+            restricted_data_token=None
     ):
         super().__init__(account, credentials)
         self.boto3_client = boto3.client(
@@ -40,6 +41,7 @@ def __init__(
         self.endpoint = marketplace.endpoint
         self.marketplace_id = marketplace.marketplace_id
         self.region = marketplace.region
+        self.restricted_data_token = restricted_data_token
         self._auth = AccessTokenClient(refresh_token=refresh_token, account=account, credentials=credentials)
 
     def _get_cache_key(self, token_flavor=''):
@@ -61,7 +63,7 @@ def headers(self):
         return {
             'host': self.endpoint[8:],
             'user-agent': self.user_agent,
-            'x-amz-access-token': self.auth.access_token,
+            'x-amz-access-token': self.restricted_data_token or self.auth.access_token,
             'x-amz-date': datetime.utcnow().strftime('%Y%m%dT%H%M%SZ'),
             'content-type': 'application/json'
         }

tests/api/tokens/test_tokens.py

@@ -0,0 +1,23 @@
+from datetime import timedelta, datetime
+
+import pytest
+
+from sp_api.api import Tokens, Orders
+from sp_api.base import SellingApiBadRequestException, Marketplaces
+
+#
+# def test_get_token_for_bulk_orders():
+#     with pytest.raises(SellingApiBadRequestException) as info:
+#         res = Tokens().create_restricted_data_token(restrictedResources=[
+#             {
+#                 "method": "GET",
+#                 "path": "/orders/v0/orders",
+#                 "dataElements": ["buyerInfo", "shippingAddress"]
+#             }
+#         ])
+#         print(res)
+#
+#
+# def test_get_order_with_token():
+#     res = Orders(restricted_data_token='foo').get_orders(CreatedAfter=(datetime.now() - timedelta(days=10)).isoformat())
+#     print(res)