feat: add toutiao.search_v1 with API integration, client support, and test

Author: Anonymous Committer

justoneapi/apis/toutiao.py

@@ -0,0 +1,28 @@
+from justoneapi.apis import request_util
+from justoneapi.log import logger
+
+
+class ToutiaoAPI:
+    def __init__(self, token: str, base_url: str):
+        self.token = token
+        self.base_url = base_url
+
+    def search_v1(self, keyword: str, page: int, search_id: str = None):
+        url = f"{self.base_url}/api/toutiao/search/v1"
+        params = {
+            "token": self.token,
+            "keyword": keyword,
+            "page": page,
+        }
+        if search_id:
+            params["searchId"] = search_id
+
+        has_next_page = False
+        result, data, message = request_util.get_request_page(url, params)
+        try:
+            if data:
+                if data.get("search_id"):
+                    has_next_page = True
+        except Exception as e:
+            logger.warning(f"Pagination parse error at {url}. Contact us to fix it.")
+        return result, data, message, has_next_page

justoneapi/client.py

@@ -6,6 +6,7 @@
 from justoneapi.apis.search import SearchAPI
 from justoneapi.apis.taobao import TaobaoAPI
 from justoneapi.apis.tiktok import TikTokAPI
+from justoneapi.apis.toutiao import ToutiaoAPI
 from justoneapi.apis.user import UserAPI
 from justoneapi.apis.weibo import WeiboAPI
 from justoneapi.apis.xiaohongshu import XiaohongshuAPI
@@ -32,3 +33,4 @@ def __init__(self, token: str, env: str = "cn"):
         self.search = SearchAPI(self.token, self.base_url)
         self.jd = JdAPI(self.token, self.base_url)
         self.tiktok = TikTokAPI(self.token, self.base_url)
+        self.toutiao = ToutiaoAPI(self.token, self.base_url)

justoneapi/config.py

@@ -1,2 +1,2 @@
-BASE_URL_CN = "http://47.117.133.51:30015"
+BASE_URL_CN = "http://localhost:8008"
 BASE_URL_GLOBAL = "https://api.justoneapi.com"

tests/test_toutiao.py

@@ -0,0 +1,14 @@
+import json
+import os
+from unittest import TestCase
+
+from justoneapi.client import JustOneAPIClient
+
+
+class TestToutiaoAPI(TestCase):
+    client = JustOneAPIClient(token=os.environ.get("JUSTONEAPI_TOKEN"))
+
+    def test_search_v1(self):
+        result, data, message, has_next_page = self.client.toutiao.search_v1(keyword="个人公积金", page=1)
+        if result:
+            print(json.dumps(data, ensure_ascii=False))