diff --git a/api/index.py b/api/index.py index 1eacb3e..ce45672 100644 --- a/api/index.py +++ b/api/index.py @@ -6,10 +6,12 @@ @app.route('/search') def search(): - # 从请求参数中获取关键词 + # 从请求参数中获取搜索的关键词 keywords = request.args.get('q') # 从请求参数中获取最大结果数,如果未指定,则默认为10 max_results = int(request.args.get('max_results', 10)) + # 从请求参数中获取 href + href = request.args.get('href') results = [] with DDGS() as ddgs: @@ -17,11 +19,12 @@ def search(): ddgs_gen = ddgs.text(keywords, safesearch='Off', timelimit='y', backend="lite") # 从搜索结果中获取最大结果数 for r in islice(ddgs_gen, max_results): - results.append(r) + # 如果请求参数中包含 href,检查每个结果的 href 是否包含指定的 href + if href is None or href in r.get('href', ''): + results.append(r) # 返回一个json响应,包含搜索结果 return {'results': results} if __name__ == '__main__': app.run(host='0.0.0.0') -