Add Amazon shipping endpoint

Author: jchen293

CHANGELOG.md

@@ -16,6 +16,7 @@
 
 - Adds `tracking_codes` as a parameter of the `all` method on the TrackerService
 - Removes the deprecated `create_list` tracker endpoint function as it is no longer available via API
+- Routes `AmazonShippingAccount` create/update requests to the new `/register_oauth` endpoint
 
 ## v9.4.1 (2024-08-09)
 

easypost/constant.py

@@ -36,6 +36,9 @@
     "FedexAccount",
     "FedexSmartpostAccount",
 ]
+_CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH = [
+    "AmazonShippingAccount",
+]
 _UPS_OAUTH_CARRIER_ACCOUNT_TYPES = [
     "UpsAccount",
     "UpsMailInnovationsAccount",

easypost/services/base_service.py

@@ -85,7 +85,6 @@ def _update_resource(
         """Update an EasyPost object via the EasyPost API."""
         url = self._instance_url(class_name, id)
         wrapped_params = {self._snakecase_name(class_name): params}
-
         response = Requestor(self._client).request(method=method, url=url, params=wrapped_params, beta=beta)
 
         return convert_to_easypost_object(response=response)

easypost/services/carrier_account_service.py

@@ -4,6 +4,7 @@
 )
 
 from easypost.constant import (
+    _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH,
     _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_WORKFLOWS,
     _UPS_OAUTH_CARRIER_ACCOUNT_TYPES,
     MISSING_PARAMETER_ERROR,
@@ -33,6 +34,8 @@ def create(self, **params) -> CarrierAccount:
         url = self._select_carrier_account_creation_endpoint(carrier_account_type=carrier_account_type)
         if carrier_account_type in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES:
             wrapped_params = {"ups_oauth_registrations": params}
+        elif carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH:
+            wrapped_params = {"carrier_account_oauth_registrations": params}
         else:
             wrapped_params = {self._snakecase_name(self._model_class): params}
 
@@ -54,6 +57,13 @@ def update(self, id: str, **params) -> CarrierAccount:
 
         if carrier_account.get("type") in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES:
             class_name = "UpsOauthRegistrations"
+        elif carrier_account.get("type") in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH:
+            response = Requestor(self._client).request(
+                method=RequestMethod.PATCH,
+                url=f"/carrier_accounts/register_oauth/{id}",
+                params={},
+            )
+            return convert_to_easypost_object(response=response)
         else:
             class_name = self._model_class
 
@@ -75,5 +85,7 @@ def _select_carrier_account_creation_endpoint(self, carrier_account_type: Option
             return "/carrier_accounts/register"
         elif carrier_account_type in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES:
             return "/ups_oauth_registrations"
+        elif carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH:
+            return "/carrier_accounts/register_oauth"
 
         return "/carrier_accounts"