Merge pull request #54 from amadeus4dev/generate-photos

Support for AI Generated Photos

Author: tsolakoua

README.rst

@@ -276,6 +276,9 @@ List of supported endpoints
     # 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')
+
 Development & Contributing
 --------------------------
 

amadeus/media/__init__.py

@@ -0,0 +1,3 @@
+from ._files import GeneratedPhotos
+
+__all__ = ['GeneratedPhotos']

amadeus/media/_files.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from .files import GeneratedPhotos
+
+
+class Files(Decorator, object):
+    def __init__(self, client):
+        Decorator.__init__(self, client)
+        self.generated_photos = GeneratedPhotos(client)

amadeus/media/files/__init__.py

@@ -0,0 +1,4 @@
+from ._generated_photos import GeneratedPhotos
+
+
+__all__ = ['GeneratedPhotos']

amadeus/media/files/_generated_photos.py

@@ -0,0 +1,19 @@
+from amadeus.client.decorator import Decorator
+
+
+class GeneratedPhotos(Decorator, object):
+    def get(self, **params):
+        '''
+        Returns a link with a rendered image of a landscape
+
+        .. code-block:: python
+
+        amadeus.media.files.generated_photos.get(category='MOUNTAIN')
+
+        :param category: the type of landscape to be generated,
+            ``"MOUNTAIN"`` or ``"BEACH"``
+
+        :rtype: amadeus.Response
+        :raises amadeus.ResponseError: if the request could not be completed
+        '''
+        return self.client.get('/v2/media/files/generated-photos', **params)

amadeus/namespaces/_media.py

@@ -0,0 +1,8 @@
+from amadeus.client.decorator import Decorator
+from amadeus.media._files import Files
+
+
+class Media(Decorator, object):
+    def __init__(self, client):
+        Decorator.__init__(self, client)
+        self.files = Files(client)

amadeus/namespaces/core.py

@@ -3,6 +3,7 @@
 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
 
 
 class Core(object):
@@ -12,3 +13,4 @@ def __init__(self):
         self.shopping = Shopping(self)
         self.e_reputation = EReputation(self)
         self.airport = Airport(self)
+        self.media = Media(self)

docs/index.rst

@@ -108,3 +108,9 @@ Airport/Predictions
 
 .. autoclass:: amadeus.airport.predictions.AirportOnTime
   :members: get
+
+Media/Files
+================
+
+.. autoclass:: amadeus.media.files.GeneratedPhotos
+  :members: get

specs/namespaces/namespaces_spec.py

@@ -54,6 +54,10 @@
         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)
+
     with it('should define all expected .get methods'):
         client = self.client
         expect(client.reference_data.urls.checkin_links.get).not_to(be_none)
@@ -93,6 +97,8 @@
 
         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'):
         with before.each:
             self.client.get = method_returning(None)
@@ -232,3 +238,9 @@
             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(
+                '/v2/media/files/generated-photos', a='b'
+            ))