main.py 546 B

12345678910111213141516171819
  1. from typing import Optional
  2. from urllib.parse import urlparse
  3. from pydantic import BaseModel
  4. def get_filtered_results(results, filter_list):
  5. if not filter_list:
  6. return results
  7. filtered_results = []
  8. for result in results:
  9. domain = urlparse(result["url"]).netloc
  10. if any(domain.endswith(filtered_domain) for filtered_domain in filter_list):
  11. filtered_results.append(result)
  12. return filtered_results
  13. class SearchResult(BaseModel):
  14. link: str
  15. title: Optional[str]
  16. snippet: Optional[str]