add support for flight availability

Author: tsolakoua

README.rst

@@ -219,6 +219,9 @@ List of supported endpoints
     # Flight SeatMap Display POST
     amadeus.shopping.seatmaps.post(body)
 
+    # Flight Availabilities POST
+    amadeus.shopping.availability​.flight_availabilities.post(body)
+
     # Flight Choice Prediction
     body = amadeus.shopping.flight_offers_search.get(
             originLocationCode='MAD',

amadeus/namespaces/_shopping.py

@@ -9,6 +9,7 @@
 from amadeus.shopping._seatmaps import Seatmaps
 from amadeus.shopping._activities import Activities
 from amadeus.shopping._activity import Activity
+from amadeus.shopping._availability import Availability
 
 
 class Shopping(Decorator, object):
@@ -22,6 +23,7 @@ def __init__(self, client):
         self.flight_offers_search = FlightOffersSearch(client)
         self.seatmaps = Seatmaps(client)
         self.activities = Activities(client)
+        self.availability = Availability(client)
 
     def hotel_offer(self, offer_id):
         return HotelOffer(self.client, offer_id)
@@ -31,4 +33,4 @@ def activity(self, activity_id):
 
 
 __all__ = ['FlightDates', 'FlightDestinations', 'FlightOffers',
-           'FlightOffersSearch']
+           'FlightOffersSearch', 'Availability']

amadeus/shopping/_availability.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from .availability import FlightAvailabilities
+
+
+class Availability(Decorator, object):
+    def __init__(self, client):
+        Decorator.__init__(self, client)
+        self.flight_availabilities = FlightAvailabilities(client)

amadeus/shopping/availability/__init__.py

@@ -0,0 +1,3 @@
+from ._flight_availabilities import FlightAvailabilities
+
+__all__ = ['FlightAvailabilities']

amadeus/shopping/availability/_flight_availabilities.py

@@ -0,0 +1,19 @@
+from amadeus.client.decorator import Decorator
+
+
+class FlightAvailabilities(Decorator, object):
+    def post(self, body):
+        '''
+        Get available seats in different fare classes
+
+        .. code-block:: python
+
+            amadeus.shopping.availability​.flight_availabilities.post(body)
+
+        :param body: the parameters to send to the API
+
+        :rtype: amadeus.Response
+        :raises amadeus.ResponseError: if the request could not be completed
+        '''
+        return self.client.post(
+            '/v1/shopping/availability/flight-availabilities', body)

docs/index.rst

@@ -80,6 +80,12 @@ Shopping/Activities
 .. autoclass:: amadeus.shopping.Activity
   :members: get
 
+Shopping/Availability
+===============
+
+.. autoclass:: amadeus.shopping.availability.FlightAvailabilities
+  :members: post
+
 Travel/Analytics
 ================
 

specs/namespaces/namespaces_spec.py

@@ -50,6 +50,9 @@
 
         expect(client.shopping.activities).not_to(be_none)
 
+        expect(client.shopping.availability).not_to(be_none)
+        expect(client.shopping.availability.flight_availabilities).not_to(be_none)
+
         expect(client.e_reputation.hotel_sentiments).not_to(be_none)
 
         expect(client.airport).not_to(be_none)
@@ -386,6 +389,13 @@
                           }}
             ))
 
+        with it('.shopping.availability.flight_availabilities.post'):
+            self.client.shopping.availability.flight_availabilities.post(
+                {'foo': 'bar'})
+            expect(self.client.post).to(have_been_called_with(
+                '/v1/shopping/availability/flight-availabilities', {'foo': 'bar'}
+            ))
+
         with it('.booking.flight_order().get'):
             self.client.booking.flight_order('123').get(a='b')
             expect(self.client.get).to(have_been_called_with(