main.py 547 B

1234567891011121314151617181920
  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]