Merge pull request #165 from siddydutta/travel-restrictions-v2

Add support of Travel Restrictions V2

Author: tsolakoua

README.rst

@@ -352,8 +352,10 @@ List of supported endpoints
     amadeus.analytics.itinerary_price_metrics.get(originIataCode='MAD', destinationIataCode='CDG',
                                                 departureDate='2021-03-21')
 
-    # Covid-19 Area Report
+    # Travel Restrictions v1
     amadeus.duty_of_care.diseases.covid19_area_report.get(countryCode="US")
+    # Travel Restrictions v2
+    amadeus.duty_of_care.diseases.covid19_report.get(countryCode='US')
 
     # Airline Routes
     amadeus.airline.destinations.get(airlineCode='BA')

amadeus/duty_of_care/_diseases.py

@@ -1,8 +1,10 @@
 from amadeus.client.decorator import Decorator
 from amadeus.duty_of_care.diseases import Covid19AreaReport
+from amadeus.duty_of_care.diseases import Covid19Report
 
 
 class Diseases(Decorator, object):
     def __init__(self, client):
         Decorator.__init__(self, client)
         self.covid19_area_report = Covid19AreaReport(client)
+        self.covid19_report = Covid19Report(client)

amadeus/duty_of_care/diseases/__init__.py

@@ -1,3 +1,4 @@
 from ._covid19_area_report import Covid19AreaReport
+from ._covid19_report import Covid19Report
 
-__all__ = ['Covid19AreaReport']
+__all__ = ['Covid19AreaReport', 'Covid19Report']

amadeus/duty_of_care/diseases/_covid19_report.py

@@ -0,0 +1,22 @@
+from amadeus.client.decorator import Decorator
+
+
+class Covid19Report(Decorator, object):
+    def get(self, **params):
+        '''
+        Returns the Covid-19 restrictions on targeted area.
+
+        .. code-block:: python
+
+            amadeus.duty_of_care.diseases.covid19_report.get(
+                countryCode='US'
+            )
+
+        :param countryCode: Country code following ISO 3166 Alpha-2
+            standard, for example ``"US"`` for United States of America
+
+        :rtype: amadeus.Response
+        :raises amadeus.ResponseError: if the request could not be completed
+        '''
+        return self.client.get(
+            '/v2/duty-of-care/diseases/covid19-area-report', **params)

docs/index.rst

@@ -257,6 +257,9 @@ DutyOfCare/Diseases
 .. autoclass:: amadeus.duty_of_care.diseases.Covid19AreaReport
   :members: get
 
+.. autoclass:: amadeus.duty_of_care.diseases.Covid19Report
+  :members: get
+
 Airline/Destinations
 ================
 

specs/namespaces/namespaces_spec.py

@@ -92,6 +92,7 @@
 
         expect(client.duty_of_care).not_to(be_none)
         expect(client.duty_of_care.diseases.covid19_area_report).not_to(be_none)
+        expect(client.duty_of_care.diseases.covid19_report).not_to(be_none)
 
         expect(client.airline.destinations).not_to(be_none)
 
@@ -159,6 +160,7 @@
 
         expect(client.duty_of_care.diseases.covid19_area_report.get).not_to(
             be_none)
+        expect(client.duty_of_care.diseases.covid19_report.get).not_to(be_none)
 
         expect(client.airline.destinations.get).not_to(be_none)
 
@@ -525,6 +527,12 @@
                 '/v1/duty-of-care/diseases/covid19-area-report', a='b'
             ))
 
+        with it('.duty_of_care.diseases.covid19_report.get'):
+            self.client.duty_of_care.diseases.covid19_report.get(a='b')
+            expect(self.client.get).to(have_been_called_with(
+                '/v2/duty-of-care/diseases/covid19-area-report', a='b'
+            ))
+
         with it('.reference_data.locations.hotels.by_hotels.get'):
             self.client.reference_data.locations.hotels.by_hotels.get(
                 a='b')