Adds support for Flight Offers Price (#58)

* initial commit

* update specs of new path

* update docs

* implement post with get parameters in the url and docs

* add tests for post and update post method

Author: tsolakoua

README.rst

@@ -296,6 +296,12 @@ List of supported endpoints
     # Flight Offers Search POST
     amadeus.shopping.flight_offers_search.post(body)
 
+    # Flight Offers Price
+    flights = amadeus.shopping.flight_offers_search.get(originLocationCode='SYD', destinationLocationCode='BKK', departureDate='2020-05-01', adults=1).data
+    amadeus.shopping.flight_offers.pricing.post(flights[0])
+    amadeus.shopping.flight_offers.pricing.post(flights[0:2], include='credit-card-fees,other-services')
+
+
 Development & Contributing
 --------------------------
 

amadeus/shopping/_flight_offers.py

@@ -1,11 +1,13 @@
 from amadeus.client.decorator import Decorator
 from amadeus.shopping.flight_offers._prediction import FlightChoicePrediction
+from amadeus.shopping.flight_offers._pricing import FlightOffersPrice
 
 
 class FlightOffers(Decorator, object):
     def __init__(self, client):
         Decorator.__init__(self, client)
         self.prediction = FlightChoicePrediction(client)
+        self.pricing = FlightOffersPrice(client)
 
     def get(self, **params):
         '''

amadeus/shopping/flight_offers/__init__.py

@@ -1,3 +1,4 @@
 from ._prediction import FlightChoicePrediction
+from ._pricing import FlightOffersPrice
 
-__all__ = ['FlightChoicePrediction']
+__all__ = ['FlightChoicePrediction', 'FlightOffersPrice']

amadeus/shopping/flight_offers/_pricing.py

@@ -0,0 +1,33 @@
+from amadeus.client.decorator import Decorator
+try:
+    from urllib.parse import urlencode
+except ImportError:     # pragma: no cover
+    from urllib import urlencode    # pragma: no cover
+
+
+class FlightOffersPrice(Decorator, object):
+    def post(self, body, **params):
+        '''
+        Gets a confirmed price and availability of a flight
+
+        .. code-block:: python
+
+            amadeus.shopping.flight_offers.pricing.post(body, params)
+
+        :rtype: amadeus.Response
+        :raises amadeus.ResponseError: if the request could not be completed
+        '''
+        url = '/v1/shopping/flight-offers/pricing'
+        flight_offers = []
+
+        if params is not None:
+            url = '/v1/shopping/flight-offers/pricing?'
+        if type(body) is not list:
+            flight_offers.append(body)
+        else:
+            flight_offers.extend(body)
+
+        return self.client.post(url + urlencode(params),
+                                {'data':
+                                {'type': 'flight-offers-pricing',
+                                    'flightOffers': flight_offers}})

docs/index.rst

@@ -65,6 +65,12 @@ Shopping/Hotels
 .. autoclass:: amadeus.shopping.hotel.Offer
   :members: get
 
+Shopping/Flight Offers
+===============
+
+.. autoclass:: amadeus.shopping.FlightOffersPrice
+  :members: post
+
 Travel/Analytics
 ================
 

specs/namespaces/namespaces_spec.py

@@ -44,6 +44,7 @@
         expect(client.shopping.flight_destinations).not_to(be_none)
         expect(client.shopping.flight_offers).not_to(be_none)
         expect(client.shopping.flight_offers_search).not_to(be_none)
+        expect(client.shopping.flight_offers.pricing).not_to(be_none)
 
         expect(client.shopping.hotel_offers).not_to(be_none)
         expect(client.shopping.hotel_offer).not_to(be_none)
@@ -297,3 +298,11 @@
             expect(self.client.post).to(have_been_called_with(
                 '/v2/shopping/flight-offers', {'foo': 'bar'}
             ))
+
+        with it('.shopping.flight_offers.pricing.post'):
+            self.client.shopping.flight_offers.pricing.post({'foo': 'bar'})
+            expect(self.client.post).to(have_been_called_with(
+                '/v1/shopping/flight-offers/pricing?',
+                {'data': {'type': 'flight-offers-pricing',
+                          'flightOffers': [{'foo': 'bar'}]}}
+            ))