66from easypost .constant import (
77 _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH ,
88 _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_WORKFLOWS ,
9- _UPS_OAUTH_CARRIER_ACCOUNT_TYPES ,
109 MISSING_PARAMETER_ERROR ,
1110)
1211from easypost .easypost_object import convert_to_easypost_object
@@ -32,9 +31,7 @@ def create(self, **params) -> CarrierAccount:
3231 raise MissingParameterError (MISSING_PARAMETER_ERROR .format ("type" ))
3332
3433 url = self ._select_carrier_account_creation_endpoint (carrier_account_type = carrier_account_type )
35- if carrier_account_type in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES :
36- wrapped_params = {"ups_oauth_registrations" : params }
37- elif carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH :
34+ if carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH :
3835 wrapped_params = {"carrier_account_oauth_registrations" : params }
3936 else :
4037 wrapped_params = {self ._snakecase_name (self ._model_class ): params }
@@ -53,14 +50,7 @@ def retrieve(self, id: str) -> CarrierAccount:
5350
5451 def update (self , id : str , ** params ) -> CarrierAccount :
5552 """Update a CarrierAccount."""
56- carrier_account = self .retrieve (id )
57-
58- if carrier_account .get ("type" ) in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES :
59- class_name = "UpsOauthRegistrations"
60- else :
61- class_name = self ._model_class
62-
63- return self ._update_resource (class_name , id , ** params )
53+ return self ._update_resource (self ._model_class , id , ** params )
6454
6555 def delete (self , id : str ) -> None :
6656 """Delete a CarrierAccount."""
@@ -76,8 +66,6 @@ def _select_carrier_account_creation_endpoint(self, carrier_account_type: Option
7666 """Determines which API endpoint to use for the creation call."""
7767 if carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_WORKFLOWS :
7868 return "/carrier_accounts/register"
79- elif carrier_account_type in _UPS_OAUTH_CARRIER_ACCOUNT_TYPES :
80- return "/ups_oauth_registrations"
8169 elif carrier_account_type in _CARRIER_ACCOUNT_TYPES_WITH_CUSTOM_OAUTH :
8270 return "/carrier_accounts/register_oauth"
8371
0 commit comments