mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-18 02:00:09 +08:00
21 lines
659 B
Python
21 lines
659 B
Python
import os
|
|
from googlesearch import search
|
|
|
|
from . import SearchEngine, SearchResult
|
|
|
|
from typing import List
|
|
|
|
class Google(SearchEngine):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.proxy = os.environ.get("https_proxy")
|
|
|
|
async def search(self, query: str, num_results: int) -> List[SearchResult]:
|
|
results = []
|
|
try:
|
|
ls = search(query, advanced=True, num_results=num_results, timeout=3, proxy=self.proxy)
|
|
for i in ls:
|
|
results.append(SearchResult(title=i.title, url=i.url, snippet=i.description))
|
|
except Exception as e:
|
|
raise e
|
|
return results |