Merge branch 'master' into generate-photos

Author: tsolakoua

README.rst

@@ -270,7 +270,11 @@ List of supported endpoints
     amadeus.travel.predictions.trip_purpose.get(originLocationCode='ATH', destinationLocationCode='MAD', departureDate='2020-08-01', returnDate='2020-08-12', searchDate='2020-06-11')
 
     # Flight Delay Prediction
-    amadeus.travel.predictions.flight_delay.get(originLocationCode='BRU', destinationLocationCode='FRA', departureDate='2020-01-14', departureTime='11:05:00', arrivalDate='2020-01-14', arrivalTime='12:10:00', aircraftCode='32A', carrierCode='LH', flightNumber='1009', duration='PT1H05M')
+    amadeus.travel.predictions.flight_delay.get(originLocationCode='BRU', destinationLocationCode='FRA', departureDate='2020-01-14', \
+    departureTime='11:05:00', arrivalDate='2020-01-14', arrivalTime='12:10:00', aircraftCode='32A', carrierCode='LH', flightNumber='1009', duration='PT1H05M')
+
+    # Airport On-Time Performance
+    amadeus.airport.predictions.on_time.get(airportCode='JFK', date='2020-03-01')
 
     # AI Generated Photos
     amadeus.media.files.generated_photos.get(category='MOUNTAIN')

amadeus/airport/__init__.py

@@ -0,0 +1,3 @@
+from ._predictions import AirportOnTime
+
+__all__ = ['AirportOnTime']

amadeus/airport/_predictions.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from .predictions import AirportOnTime
+
+
+class Predictions(Decorator, object):
+    def __init__(self, client):
+        Decorator.__init__(self, client)
+        self.on_time = AirportOnTime(client)

amadeus/airport/predictions/__init__.py

@@ -0,0 +1,3 @@
+from ._on_time import AirportOnTime
+
+__all__ = ['AirportOnTime']

amadeus/airport/predictions/_on_time.py

@@ -0,0 +1,23 @@
+from amadeus.client.decorator import Decorator
+
+
+class AirportOnTime(Decorator, object):
+    def get(self, **params):
+        '''
+        Returns a percentage of on-time flight departures
+
+        .. code-block:: python
+
+            amadeus.airport.predictions.on_time.get(
+                            airportCode='JFK',
+                            date='2020-03-01')
+
+        :param airportCode: the City/Airport IATA code from which
+            the flight will depart. ``"NYC"``, for example for New York
+
+        :param date: 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/airport/predictions/on-time', **params)

amadeus/namespaces/_airport.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from amadeus.airport._predictions import Predictions
+
+
+class Airport(Decorator, object):
+    def __init__(self, client):
+        Decorator.__init__(self, client)
+        self.predictions = Predictions(client)

amadeus/namespaces/core.py

@@ -2,6 +2,7 @@
 from amadeus.namespaces._travel import Travel
 from amadeus.namespaces._shopping import Shopping
 from amadeus.namespaces._e_reputation import EReputation
+from amadeus.namespaces._airport import Airport
 from amadeus.namespaces._media import Media
 
 
@@ -11,4 +12,5 @@ def __init__(self):
         self.travel = Travel(self)
         self.shopping = Shopping(self)
         self.e_reputation = EReputation(self)
+        self.airport = Airport(self)
         self.media = Media(self)

docs/index.rst

@@ -103,6 +103,12 @@ Helper/Location
 
 .. autoclass:: amadeus.Location
 
+Airport/Predictions
+================
+
+.. autoclass:: amadeus.airport.predictions.AirportOnTime
+  :members: get
+
 Media/Files
 ================
 

specs/namespaces/namespaces_spec.py

@@ -50,6 +50,10 @@
 
         expect(client.e_reputation.hotel_sentiments).not_to(be_none)
 
+        expect(client.airport).not_to(be_none)
+        expect(client.airport.predictions).not_to(be_none)
+        expect(client.airport.predictions.on_time).not_to(be_none)
+
         expect(client.media).not_to(be_none)
         expect(client.media.files).not_to(be_none)
         expect(client.media.files.generated_photos).not_to(be_none)
@@ -91,6 +95,8 @@
 
         expect(client.e_reputation.hotel_sentiments.get).not_to(be_none)
 
+        expect(client.airport.predictions.on_time.get).not_to(be_none)
+
         expect(client.media.files.generated_photos.get).not_to(be_none)
 
     with context('testing all calls to the client'):
@@ -227,6 +233,12 @@
                 '/v2/e-reputation/hotel-sentiments', hotelIds='XKPARC12'
             ))
 
+        with it('.airport.predictions.on_time.get'):
+            self.client.airport.predictions.on_time.get(a='b')
+            expect(self.client.get).to(have_been_called_with(
+                '/v1/airport/predictions/on-time', a='b'
+            ))
+
         with it('.media.files.generated_photos.get'):
             self.client.media.files.generated_photos.get(a='b')
             expect(self.client.get).to(have_been_called_with(