Add direction (arriving, departing) as parameter for Busiest Travel Period

Author: samheyman

amadeus/__init__.py

@@ -2,14 +2,15 @@
 from .version import version
 
 from .client.location import Location
+from .client.direction import Direction
 from .client.request import Request
 from .client.response import Response
 from .client.errors import ResponseError
 from .client.errors import ParserError, ServerError, AuthenticationError
 from .client.errors import NotFoundError, ClientError, NetworkError
 
 __all__ = [
-    'Client', 'Location', 'version', 'ResponseError',
+    'Client', 'Location', 'Direction', 'version', 'ResponseError',
     'ParserError', 'ServerError', 'AuthenticationError',
     'NotFoundError', 'ClientError', 'Request', 'Response',
     'NetworkError'

amadeus/client/direction.py

@@ -0,0 +1,23 @@
+#
+class Direction (object):
+    '''
+    A list of direction types, as used in Busiest Travel Period
+
+    .. code-block:: python
+
+
+        from amadeus import Direction
+
+        amadeus.travel.analytics.air_traffic.busiest_period.get(
+            cityCode = 'MAD',
+            period = '2017',
+            direction = Direction.ARRIVING
+        )
+
+    :cvar ARRIVING: ``"ARRIVING"``
+    :cvar DEPARTING: ``"DEPARTING"``
+    '''
+    # Arriving
+    ARRIVING = 'ARRIVING'
+    # Departing
+    DEPARTING = 'DEPARTING'

amadeus/travel/analytics/air_traffic/_busiest_period.py

@@ -8,15 +8,16 @@ def get(self, **params):
 
         .. code-block:: python
 
-            amadeus.travel.analytics.air_traffic.traveled.get(
-                origin='LHR',
-                period='2017-01'
+            amadeus.travel.analytics.air_traffic.busiest_period.get(
+                origin='MAD',
+                period='2017',
+                direction=Direction.ARRIVING
             )
 
         :param cityCode: IATA code of the origin city, for
             example ``"BOS"`` for Boston.
         :param query: period when consumers are traveling
-            in ``YYYY-MM`` format
+            in ``YYYY`` format
 
         :rtype: amadeus.Response
         :raises amadeus.ResponseError: if the request could not be completed