Low-fare Search deco and update Flight Choice Prediction v2 (#87)

* update classes & tests & examples

* delete low-fare search example

* update readme

* update departure date in example

Author: tsolakoua

README.rst

@@ -242,7 +242,11 @@ List of supported endpoints
     amadeus.shopping.seatmaps.post(body)
 
     # Flight Choice Prediction
-    body = amadeus.shopping.flight_offers.get(origin='MAD', destination='NYC', departureDate='2020-10-01').result
+    body = amadeus.shopping.flight_offers_search.get(
+            originLocationCode='MAD',
+            destinationLocationCode='NYC',
+            departureDate='2020-11-01',
+            adults=1).result
     amadeus.shopping.flight_offers.prediction.post(body)
 
     # Flight Checkin Links

amadeus/shopping/_flight_offers.py

@@ -8,29 +8,3 @@ def __init__(self, client):
         Decorator.__init__(self, client)
         self.prediction = FlightChoicePrediction(client)
         self.pricing = FlightOffersPrice(client)
-
-    def get(self, **params):
-        '''
-        Find the cheapest bookable flights.
-
-        .. code-block:: python
-
-            amadeus.shopping.flight_offers.get(
-                origin='MAD',
-                destination='NYC',
-                departureDate='2020-08-01'
-            )
-
-        :param origin: the City/Airport IATA code from which the flight will
-            depart. ``"MAD"``, for example for Madrid.
-
-        :param destination: the City/Airport IATA code to which the flight is
-            going. ``"BOS"``, for example for Boston.
-
-        :param departureDate: the date on which to fly out, in `YYYY-MM-DD`
-            format
-
-        :rtype: amadeus.Response
-        :raises amadeus.ResponseError: if the request could not be completed
-        '''
-        return self.client.get('/v1/shopping/flight-offers', **params)

amadeus/shopping/flight_offers/_prediction.py

@@ -9,13 +9,14 @@ def post(self, body):
         .. code-block:: python
 
             amadeus.shopping.flight_offers.prediction.post(
-                amadeus.shopping.flight_offers.get(origin='MAD',
-                destination='NYC',
-                departureDate='2020-08-01'
-            ).result
+                amadeus.shopping.flight_offers_search.get(
+                originLocationCode='SYD',
+                destinationLocationCode='BKK',
+                departureDate='2020-11-01',
+                adults=1).result
         )
 
         :rtype: amadeus.Response
         :raises amadeus.ResponseError: if the request could not be completed
         '''
-        return self.client.post('/v1/shopping/flight-offers/prediction', body)
+        return self.client.post('/v2/shopping/flight-offers/prediction', body)

docs/index.rst

@@ -44,9 +44,6 @@ Shopping/Flights
 .. autoclass:: amadeus.shopping.FlightDates
   :members: get
 
-.. autoclass:: amadeus.shopping.FlightOffers
-  :members: get
-
 .. autoclass:: amadeus.shopping.FlightOffersSearch
   :members: get
 

examples/flight_choice_prediction/flight_choice_prediction.py

@@ -6,8 +6,11 @@
     '''
     Find the probability of the flight MAD to NYC to be chosen
     '''
-    result = amadeus.shopping.flight_offers.get(origin='MAD', destination='NYC',
-                                                departureDate='2020-10-01').result
+    result = amadeus.shopping.flight_offers_search.get(
+            originLocationCode='MAD',
+            destinationLocationCode='NYC',
+            departureDate='2020-11-01',
+            adults=1).result
     response = amadeus.shopping.flight_offers.prediction.post(result)
     # print(response.data)
 except ResponseError as error:

examples/flight_low_fare_search/flight_low_fare_search.py

@@ -94,7 +94,6 @@
 
         expect(client.shopping.flight_dates.get).not_to(be_none)
         expect(client.shopping.flight_destinations.get).not_to(be_none)
-        expect(client.shopping.flight_offers.get).not_to(be_none)
         expect(client.shopping.flight_offers_search.get).not_to(be_none)
 
         expect(client.shopping.seatmaps.get).not_to(be_none)
@@ -221,12 +220,6 @@
                 '/v1/shopping/flight-destinations', a='b'
             ))
 
-        with it('.shopping.flight_offers.get'):
-            self.client.shopping.flight_offers.get(a='b')
-            expect(self.client.get).to(have_been_called_with(
-                '/v1/shopping/flight-offers', a='b'
-            ))
-
         with it('.shopping.flight_offers_search.get'):
             self.client.shopping.flight_offers_search.get(a='b')
             expect(self.client.get).to(have_been_called_with(
@@ -290,7 +283,7 @@
         with it('.shopping.flight_offers.prediction.post'):
             self.client.shopping.flight_offers.prediction.post({'foo': 'bar'})
             expect(self.client.post).to(have_been_called_with(
-                '/v1/shopping/flight-offers/prediction', {'foo': 'bar'}
+                '/v2/shopping/flight-offers/prediction', {'foo': 'bar'}
             ))
 
         with it('.travel.trip_parser_jobs.post'):