From f5dd9e26d418ac7fdf517aedc6bd90b1d0d3bfc2 Mon Sep 17 00:00:00 2001
From: Rowan Seymour <rowanseymour@gmail.com>
Date: Tue, 6 May 2025 15:28:18 -0500
Subject: [PATCH] Remove fetching messages by broadcast, contact and label

---
 temba_client/v2/__init__.py | 9 ++-------
 temba_client/v2/tests.py    | 6 ------
 2 files changed, 2 insertions(+), 13 deletions(-)

diff --git a/temba_client/v2/__init__.py b/temba_client/v2/__init__.py
index 85f6563..4973717 100644
--- a/temba_client/v2/__init__.py
+++ b/temba_client/v2/__init__.py
@@ -186,21 +186,16 @@ def get_labels(self, uuid=None, name=None):
         """
         return self._get_query("labels", self._build_params(uuid=uuid, name=name), Label)
 
-    def get_messages(self, id=None, broadcast=None, contact=None, folder=None, label=None, before=None, after=None):
+    def get_messages(self, id=None, folder=None, before=None, after=None):
         """
         Gets all matching messages
         :param id: message id
-        :param broadcast: broadcast id
-        :param contact: contact object or UUID
         :param folder: folder name
-        :param label: message label name or UUID
         :param datetime before: created before
         :param datetime after: created after
         :return: message query
         """
-        params = self._build_params(
-            id=id, broadcast=broadcast, contact=contact, folder=folder, label=label, before=before, after=after
-        )
+        params = self._build_params(id=id, folder=folder, before=before, after=after)
         return self._get_query("messages", params, Message)
 
     def get_org(self, retry_on_rate_exceed=False):
diff --git a/temba_client/v2/tests.py b/temba_client/v2/tests.py
index 7db1fb4..d77b776 100644
--- a/temba_client/v2/tests.py
+++ b/temba_client/v2/tests.py
@@ -629,10 +629,7 @@ def test_get_messages(self, mock_request):
         # check with all params
         self.client.get_messages(
             id=123456,
-            broadcast=234567,
-            contact="d33e9ad5-5c35-414c-abd4-e7451c69ff1d",
             folder="inbox",
-            label="Spam",
             after=datetime(2014, 12, 12, 22, 34, 36, 978123, tzone.utc),
             before=datetime(2014, 12, 12, 22, 56, 58, 917123, tzone.utc),
         ).all()
@@ -643,10 +640,7 @@ def test_get_messages(self, mock_request):
             "messages",
             params={
                 "id": 123456,
-                "broadcast": 234567,
-                "contact": "d33e9ad5-5c35-414c-abd4-e7451c69ff1d",
                 "folder": "inbox",
-                "label": "Spam",
                 "after": "2014-12-12T22:34:36.978123Z",
                 "before": "2014-12-12T22:56:58.917123Z",
             },