Skip to content

Commit f2d5bdf

Browse files
committed
add support and include docs-tests
1 parent adb5773 commit f2d5bdf

File tree

9 files changed

+65
-0
lines changed

9 files changed

+65
-0
lines changed

README.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,6 +272,9 @@ List of supported endpoints
272272
# Flight Delay Prediction
273273
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')
274274
275+
# AI Generated Photos
276+
amadeus.media.files.generated_photos.get(category='MOUNTAIN')
277+
275278
Development & Contributing
276279
--------------------------
277280

amadeus/media/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from ._files import GeneratedPhotos
2+
3+
__all__ = ['GeneratedPhotos']

amadeus/media/_files.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from amadeus.client.decorator import Decorator
2+
from .files import GeneratedPhotos
3+
4+
5+
class Files(Decorator, object):
6+
def __init__(self, client):
7+
Decorator.__init__(self, client)
8+
self.generated_photos = GeneratedPhotos(client)

amadeus/media/files/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
from ._generated_photos import GeneratedPhotos
2+
3+
4+
__all__ = ['GeneratedPhotos']
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
from amadeus.client.decorator import Decorator
2+
3+
4+
class GeneratedPhotos(Decorator, object):
5+
def get(self, **params):
6+
'''
7+
Returns a link with a rendered image of a landscape
8+
9+
.. code-block:: python
10+
11+
amadeus.media.files.generated_photos.get(category='MOUNTAIN')
12+
13+
:param category: the type of landscape to be generated,
14+
``"MOUNTAIN"`` or ``"BEACH"``
15+
16+
:rtype: amadeus.Response
17+
:raises amadeus.ResponseError: if the request could not be completed
18+
'''
19+
return self.client.get('/v2/media/files/generated-photos', **params)

amadeus/namespaces/_media.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from amadeus.client.decorator import Decorator
2+
from amadeus.media._files import Files
3+
4+
5+
class Media(Decorator, object):
6+
def __init__(self, client):
7+
Decorator.__init__(self, client)
8+
self.files = Files(client)

amadeus/namespaces/core.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from amadeus.namespaces._travel import Travel
33
from amadeus.namespaces._shopping import Shopping
44
from amadeus.namespaces._e_reputation import EReputation
5+
from amadeus.namespaces._media import Media
56

67

78
class Core(object):
@@ -10,3 +11,4 @@ def __init__(self):
1011
self.travel = Travel(self)
1112
self.shopping = Shopping(self)
1213
self.e_reputation = EReputation(self)
14+
self.media = Media(self)

docs/index.rst

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,9 @@ Helper/Location
102102
==================
103103

104104
.. autoclass:: amadeus.Location
105+
106+
Media/Files
107+
================
108+
109+
.. autoclass:: amadeus.media.files.GeneratedPhotos
110+
:members: get

specs/namespaces/namespaces_spec.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@
5050

5151
expect(client.e_reputation.hotel_sentiments).not_to(be_none)
5252

53+
expect(client.media).not_to(be_none)
54+
expect(client.media.files).not_to(be_none)
55+
expect(client.media.files.generated_photos).not_to(be_none)
56+
5357
with it('should define all expected .get methods'):
5458
client = self.client
5559
expect(client.reference_data.urls.checkin_links.get).not_to(be_none)
@@ -87,6 +91,8 @@
8791

8892
expect(client.e_reputation.hotel_sentiments.get).not_to(be_none)
8993

94+
expect(client.media.files.generated_photos.get).not_to(be_none)
95+
9096
with context('testing all calls to the client'):
9197
with before.each:
9298
self.client.get = method_returning(None)
@@ -220,3 +226,9 @@
220226
expect(self.client.get).to(have_been_called_with(
221227
'/v2/e-reputation/hotel-sentiments', hotelIds='XKPARC12'
222228
))
229+
230+
with it('.media.files.generated_photos.get'):
231+
self.client.media.files.generated_photos.get(a='b')
232+
expect(self.client.get).to(have_been_called_with(
233+
'/v2/media/files/generated-photos', a='b'
234+
))

0 commit comments

Comments
 (0)