Merge pull request #172 from MerleLiuKun/feat-block-dm

Feat block dm

Author: MerleLiuKun

pytwitter/api.py

@@ -2035,6 +2035,32 @@ def unblock_user(self, user_id: str, target_user_id: str) -> dict:
         data = self._parse_response(resp)
         return data
 
+    def block_user_dm(self, target_user_id: str):
+        """
+        Allows the authenticated user to block direct messages (DMs) from the target user.
+        :param target_user_id: target user id
+        :return: blocked status
+        """
+        resp = self._request(
+            url=f"{self.BASE_URL_V2}/users/{target_user_id}/dm/block",
+            verb="POST",
+        )
+        data = self._parse_response(resp)
+        return data
+
+    def unblock_user_dm(self, target_user_id: str):
+        """
+        Allows the authenticated user to unblock direct messages (DMs) from the target user.
+        :param target_user_id: target user id
+        :return: unblocked status
+        """
+        resp = self._request(
+            url=f"{self.BASE_URL_V2}/users/{target_user_id}/dm/unblock",
+            verb="POST",
+        )
+        data = self._parse_response(resp)
+        return data
+
     def get_user_muting(
         self,
         user_id: str,

tests/apis/test_users.py

@@ -153,6 +153,27 @@ def test_block_and_unblock_user(api_with_user):
     assert not resp["data"]["blocking"]
 
 
+@responses.activate
+def test_block_and_unblock_user_dm(api_with_user):
+    target_user_id = "123456"
+
+    responses.add(
+        method=responses.POST,
+        url=f"https://api.twitter.com/2/users/{target_user_id}/dm/block",
+        json={"data": {"blocked": True}},
+    )
+    resp = api_with_user.block_user_dm(target_user_id=target_user_id)
+    assert resp["data"]["blocked"]
+
+    responses.add(
+        method=responses.POST,
+        url=f"https://api.twitter.com/2/users/{target_user_id}/dm/unblock",
+        json={"data": {"blocked": False}},
+    )
+    resp = api_with_user.unblock_user_dm(target_user_id=target_user_id)
+    assert not resp["data"]["blocked"]
+
+
 @responses.activate
 def test_get_blocking_users(api_with_user, helpers):
     users_data = helpers.load_json_data(