diff options
| -rw-r--r-- | searx/engines/iqiyi.py | 83 | ||||
| -rw-r--r-- | searx/settings.yml | 5 |
2 files changed, 88 insertions, 0 deletions
diff --git a/searx/engines/iqiyi.py b/searx/engines/iqiyi.py new file mode 100644 index 000000000..95d094ec6 --- /dev/null +++ b/searx/engines/iqiyi.py @@ -0,0 +1,83 @@ +# SPDX-License-Identifier: AGPL-3.0-or-later +"""iQiyi: A search engine for retrieving videos from iQiyi.""" + +from urllib.parse import urlencode +from datetime import datetime, timedelta + +from searx.exceptions import SearxEngineAPIException + +about = { + "website": "https://www.iqiyi.com/", + "wikidata_id": "Q15913890", + "use_official_api": False, + "require_api_key": False, + "results": "JSON", +} + +paging = True +time_range_support = True +categories = ["videos"] + +time_range_dict = {'day': '1', 'week': '2', 'month': '3'} + +base_url = "https://mesh.if.iqiyi.com" + + +def request(query, params): + query_params = {"key": query, "pageNum": params["pageno"], "pageSize": 25} + + if time_range_dict.get(params['time_range']): + query_params["sitePublishDate"] = time_range_dict[params['time_range']] + + params["url"] = f"{base_url}/portal/lw/search/homePageV3?{urlencode(query_params)}" + return params + + +def response(resp): + try: + data = resp.json() + except Exception as e: + raise SearxEngineAPIException(f"Invalid response: {e}") from e + results = [] + + if "data" not in data or "templates" not in data["data"]: + raise SearxEngineAPIException("Invalid response") + + for entry in data["data"]["templates"]: + album_info = entry.get("albumInfo", {}) + + published_date = None + release_time = album_info.get("releaseTime", {}).get("value") + if release_time: + try: + published_date = datetime.strptime(release_time, "%Y-%m-%d") + except (ValueError, TypeError): + pass + + length = None + subscript_content = album_info.get("subscriptContent") + if subscript_content: + try: + time_parts = subscript_content.split(":") + if len(time_parts) == 2: + minutes, seconds = map(int, time_parts) + length = timedelta(minutes=minutes, seconds=seconds) + elif len(time_parts) == 3: + hours, minutes, seconds = map(int, time_parts) + length = timedelta(hours=hours, minutes=minutes, seconds=seconds) + except (ValueError, TypeError): + pass + + results.append( + { + 'url': album_info.get("pageUrl", "").replace("http://", "https://"), + 'title': album_info.get("title", ""), + 'content': album_info.get("brief", {}).get("value", ""), + 'template': 'videos.html', + 'length': length, + 'publishedDate': published_date, + 'thumbnail': album_info.get("img", ""), + } + ) + + return results diff --git a/searx/settings.yml b/searx/settings.yml index 8a5069736..9662d2d09 100644 --- a/searx/settings.yml +++ b/searx/settings.yml @@ -1118,6 +1118,11 @@ engines: shortcut: ip disabled: true + - name: iqiyi + engine: iqiyi + shortcut: iq + disabled: true + - name: jisho engine: jisho shortcut: js |