Files
AstrBot/util/search_engine_scraper/google.py
T
2024-05-19 12:46:07 +08:00

23 lines
747 B
Python

from googlesearch import search
try:
from util.search_engine_scraper.engine import SearchEngine, SearchResult
from util.search_engine_scraper.config import HEADERS, USER_AGENTS
except ImportError:
from engine import SearchEngine, SearchResult
from config import HEADERS, USER_AGENTS
from typing import List
class Google(SearchEngine):
async def search(self, query: str, num_results: int) -> List[SearchResult]:
index = 1
results = []
try:
ls = search(query, advanced=True, num_results=num_results, timeout=3)
for i in ls:
results.append(SearchResult(title=i.title, url=i.url, snippet=i.description))
except:
pass
return results