Add Support for Transfer Management API

Author: siddydutta

README.rst

@@ -356,6 +356,9 @@ List of supported endpoints
     # Transfer Create Order
     amadeus.ordering.transfer_orders.post(body, offerId='1000000000')
 
+    # Transfer Order Management
+    amadeus.ordering.transfer_order('ABC').transfers.cancellation.post(body, confirmNbr=123)
+
 Development & Contributing
 --------------------------
 

amadeus/namespaces/_ordering.py

@@ -1,8 +1,15 @@
 from amadeus.client.decorator import Decorator
 from amadeus.ordering._transfer_orders import TransferOrders
+from amadeus.ordering._transfer_order import TransferOrder
 
 
 class Ordering(Decorator, object):
     def __init__(self, client):
         Decorator.__init__(self, client)
         self.transfer_orders = TransferOrders(client)
+
+    def transfer_order(self, order_id):
+        return TransferOrder(self.client, order_id)
+
+
+__all__ = ['TransferOrders', 'TransferOrder']

amadeus/ordering/_transfer_order.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from amadeus.ordering.transfer_orders import Transfers
+
+
+class TransferOrder(Decorator, object):
+    def __init__(self, client, order_id):
+        Decorator.__init__(self, client)
+        self.transfers = Transfers(client, order_id)

amadeus/ordering/_transfer_orders.py

@@ -10,7 +10,7 @@ def post(self, body, **params):
 
         .. code-block:: python
 
-            amadeus.ordering.transfer_orders.post(body, offerId='1000000000')
+            amadeus.ordering.transfer_orders.post(body, offerId=offer_id)
 
         :rtype: amadeus.Response
         :raises amadeus.ResponseError: if the request could not be completed

amadeus/ordering/transfer_orders/__init__.py

@@ -0,0 +1,3 @@
+from ._transfers import Transfers
+
+__all__ = ['Transfers']

amadeus/ordering/transfer_orders/_transfers.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from amadeus.ordering.transfer_orders.transfers import Cancellation
+
+
+class Transfers(Decorator, object):
+    def __init__(self, client, order_id):
+        Decorator.__init__(self, client)
+        self.cancellation = Cancellation(client, order_id)

amadeus/ordering/transfer_orders/transfers/__init__.py

@@ -0,0 +1,3 @@
+from ._cancellation import Cancellation
+
+__all__ = ['Cancellation']

amadeus/ordering/transfer_orders/transfers/_cancellation.py

@@ -0,0 +1,23 @@
+from amadeus.client.decorator import Decorator
+
+from urllib.parse import urlencode
+
+
+class Cancellation(Decorator, object):
+    def __init__(self, client, order_id):
+        Decorator.__init__(self, client)
+        self.order_id = order_id
+
+    def post(self, body, **params):
+        '''
+        Cancels a transfer reservation
+
+        .. code-block:: python
+
+            amadeus.ordering.transfer_order(order_id).transfers.cancellation.post(body, confirmNbr=confirm_nbr)
+
+        :rtype: amadeus.Response
+        :raises amadeus.ResponseError: if the request could not be completed
+        '''
+        url = '/v1/ordering/transfer-orders/{0}/transfers/cancellation?'.format(self.order_id)
+        return self.client.post(url + urlencode(params), body)

docs/index.rst

@@ -263,5 +263,8 @@ Airline/Destinations
 Ordering/Transfers
 ================
 
-.. autoclass:: amadeus.Ordering.TransferOrders
+.. autoclass:: amadeus.ordering.TransferOrders
+  :members: post
+
+.. autoclass:: amadeus.ordering.transfer_orders.transfers.Cancellation
   :members: post

specs/namespaces/test_namespaces.py

@@ -69,6 +69,7 @@ def test_expected_paths(client):
     assert client.airline.destinations is not None
     assert client.shopping.transfer_offers_search is not None
     assert client.ordering.transfer_orders is not None
+    assert client.ordering.transfer_order.transfers.cancellation is not None
 
 
 def test_expected_get_methods(client):
@@ -560,3 +561,12 @@ def test_ordering_transfer_orders_post(client_setup):
     client_setup.post.assert_called_with(
         '/v1/ordering/transfer-orders?'+'offerId=1', {'foo': 'bar'}
     )
+
+
+def test_ordering_transfer_order_transfers_cancellation_post(client_setup):
+    client_setup.ordering.transfer_order('XXX').transfers.cancellation.post(
+        {'foo': 'bar'}, confirmNbr=123)
+    client_setup.post.assert_called_with(
+        '/v1/ordering/transfer-orders/XXX/transfers/cancellation?'+'confirmNbr=123',
+        {'foo': 'bar'}
+    )