Add support for query_slots

cronofy · Jan 18, 2024 · 08795ca · 08795ca
1 parent fe3dd4f
commit 08795ca
Show file tree

Hide file tree

Showing 4 changed files with 92 additions and 5 deletions.
diff --git a/PKG-INFO b/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: pycronofy
-Version: 2.0.3
+Version: 2.0.4
 Summary: Python wrapper for Cronofy
 Home-page: https://github.com/cronofy/pycronofy
 Author: VenueBook

diff --git a/pycronofy/__init__.py b/pycronofy/__init__.py
@@ -1,6 +1,6 @@
 from pycronofy.client import Client  # noqa: F401
 from pycronofy import settings
-__version__ = '2.0.3'
+__version__ = '2.0.4'
 __name__ = 'PyCronofy'
 
 """

diff --git a/pycronofy/client.py b/pycronofy/client.py
@@ -442,14 +442,24 @@ def read_free_busy(self,
 
         return Pages(self.request_handler, results, 'free_busy', automatic_pagination)
 
-    def availability(self, participants=(), required_duration=(), available_periods=(), start_interval=None, buffer=(), response_format=None):
+    def availability(
+        self,
+        participants=(),
+        required_duration=(),
+        available_periods=None,
+        start_interval=None,
+        buffer=(),
+        response_format=None,
+        query_slots=None
+    ):
         """ Performs an availability query.
         :param list participants: An Array of participant groups or a dict for a single participant group.
         :param dict or int required_duration - An Integer representing the minimum number of minutes of availability required.
         :param list available_periods - An Array of available time periods dicts, each must specify a start and end Time.
         :param dict or int start_interval - An Interger representing the start interval minutes for the event.
         :param dict buffer - An Dict representing the buffer to apply to the request.
         :param string response_format - periods, slots or overlapping_slots (Optional, default periods)
+        :param list query_slots - An Array of query slots, each much specify a start Time.
 
         :rtype: ``list``
         """
@@ -465,8 +475,13 @@ def availability(self, participants=(), required_duration=(), available_periods=
 
         response_element = 'available_periods'
 
-        self.translate_available_periods(available_periods)
-        options['available_periods'] = available_periods
+        if available_periods:
+            self.translate_available_periods(available_periods)
+            options['available_periods'] = available_periods
+
+        if query_slots:
+            self.translate_query_slots(query_slots)
+            options['query_slots'] = query_slots
 
         if response_format:
             options['response_format'] = response_format
@@ -850,6 +865,11 @@ def translate_available_periods(self, periods):
                 if params[tp]:
                     params[tp] = format_event_time(params[tp])
 
+    def translate_query_slots(self, query_slots):
+        for params in query_slots:
+            if params['start']:
+                params['start'] = format_event_time(params['start'])
+
     def map_availability_sequence(self, sequence):
         if isinstance(sequence, collections.abc.Iterable):
             return list(map(lambda item: self.map_sequence_item(item), sequence))

diff --git a/pycronofy/tests/test_availability.py b/pycronofy/tests/test_availability.py
@@ -398,3 +398,70 @@ def request_callback(request):
 
     result = client.sequenced_availability(sequence=sequence, available_periods=periods)
     assert len(result) == 2
+
+
+@responses.activate
+def test_availablity_with_query_slots(client):
+    """Test Client.availability().
+
+    :param Client client: Client instance with test data.
+    """
+
+    def request_callback(request):
+        payload = json.loads(request.body)
+        assert payload['required_duration'] == {'minutes': 30}
+        assert payload['start_interval'] == {'minutes': 30}
+        assert payload['buffer']['before'] == {'minutes': 30}
+        assert payload['buffer']['after'] == {'minutes': 45}
+        assert payload['query_slots'] == [
+            {'start': '2017-01-03T09:00:00Z'},
+            {'start': '2017-01-04T09:00:00Z'}
+        ]
+        assert payload['participants'] == [
+            {
+                'required': 'all',
+                'members': [
+                    {'sub': 'acc_567236000909002'},
+                    {'sub': 'acc_678347111010113'}
+                ]
+            }
+        ]
+        assert payload['response_format'] == 'slots'
+
+        return (200, {}, json.dumps(TEST_AVAILABLITY_RESPONSE_SLOTS))
+
+    responses.add_callback(
+        responses.POST,
+        url='%s/%s/availability' % (settings.API_BASE_URL, settings.API_VERSION),
+        callback=request_callback,
+        content_type='application/json',
+    )
+
+    query_slots = (
+        {'start': "2017-01-03T09:00:00Z"},
+        {'start': "2017-01-04T09:00:00Z"}
+    )
+
+    example_participants = ({
+        'members': [
+            "acc_567236000909002",
+            "acc_678347111010113",
+        ],
+    })
+
+    example_buffer = {
+        'before': 30,
+        'after': {'minutes': 45}
+    }
+
+    example_response_format = 'slots'
+
+    result = client.availability(
+        required_duration=30,
+        query_slots=query_slots,
+        participants=example_participants,
+        start_interval=30,
+        buffer=example_buffer,
+        response_format=example_response_format
+    )
+    assert len(result) == 2