arsam
e2fbe90654
Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
try:
|
|
from ddgs import DDGS
|
|
except ImportError:
|
|
from duckduckgo_search import DDGS
|
|
|
|
from axon.models import SearchResult
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_TIMELIMIT = {"day": "d", "week": "w", "month": "m", "year": "y"}
|
|
|
|
|
|
async def search_ddg(
|
|
query: str,
|
|
count: int = 10,
|
|
topic: str = "general",
|
|
time_range: str | None = None,
|
|
) -> list[SearchResult]:
|
|
timelimit = _TIMELIMIT.get(time_range) if time_range else None
|
|
|
|
def _blocking_search() -> list[dict]:
|
|
with DDGS() as ddgs:
|
|
if topic == "news":
|
|
return list(ddgs.news(query, max_results=count, timelimit=timelimit))
|
|
return list(ddgs.text(query, max_results=count, timelimit=timelimit))
|
|
|
|
try:
|
|
raw = await asyncio.to_thread(_blocking_search)
|
|
results = []
|
|
for item in raw:
|
|
url = item.get("href") or item.get("url", "")
|
|
results.append(
|
|
SearchResult(
|
|
title=item.get("title", ""),
|
|
url=url,
|
|
content=item.get("body", ""),
|
|
score=0.5,
|
|
published_date=item.get("date"),
|
|
favicon=None,
|
|
)
|
|
)
|
|
return results
|
|
except Exception as exc:
|
|
logger.error("DuckDuckGo search error: %s", exc)
|
|
return []
|