feat: add search api

Author: Anonymous Committer

justoneapi/apis/search.py

@@ -0,0 +1,33 @@
+from justoneapi.apis import request_util
+from justoneapi.log import logger
+
+
+class SearchAPI:
+    def __init__(self, token: str, base_url: str):
+        self.token = token
+        self.base_url = base_url
+
+    def search_v1(self, keyword: str = None, source: str = None, start: str = None, end: str = None, next_cursor: str = None):
+        url = f"{self.base_url}/api/search/v1"
+        params = {
+            "token": self.token,
+        }
+        if keyword:
+            params["keyword"] = keyword
+        if source:
+            params["source"] = source
+        if start:
+            params["start"] = start
+        if end:
+            params["end"] = end
+        if next_cursor:
+            params["nextCursor"] = next_cursor
+
+        has_next_page = False
+        result, data, message = request_util.get_request_page(url, params)
+        try:
+            if data.get("nextCursor"):
+                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

@@ -2,6 +2,7 @@
 from justoneapi.apis.bilibili import BilibiliAPI
 from justoneapi.apis.douyin import DouyinAPI
 from justoneapi.apis.kuaishou import KuaishouAPI
+from justoneapi.apis.search import SearchAPI
 from justoneapi.apis.taobao import TaobaoAPI
 from justoneapi.apis.user import UserAPI
 from justoneapi.apis.weibo import WeiboAPI
@@ -26,3 +27,4 @@ def __init__(self, token: str, env: str = "cn"):
         self.kuaishou = KuaishouAPI(self.token, self.base_url)
         self.weibo = WeiboAPI(self.token, self.base_url)
         self.bilibili = BilibiliAPI(self.token, self.base_url)
+        self.search = SearchAPI(self.token, self.base_url)

tests/test_search.py

@@ -0,0 +1,14 @@
+import json
+import os
+from unittest import TestCase
+
+from justoneapi.client import JustOneAPIClient
+
+
+class TestSearchAPI(TestCase):
+    client = JustOneAPIClient(token=os.environ.get("JUSTONEAPI_TOKEN"))
+
+    def test_search_v1(self):
+        result, data, message, has_next_page = self.client.search.search_v1(keyword="deepseek", source="XIAOHONGSHU", start="2025-07-10 00:00:00", end="2025-08-10 00:00:00")
+        if result:
+            print(json.dumps(data, ensure_ascii=False))