main.py 546 Bytes
Newer Older
1
from typing import Optional
2
from urllib.parse import urlparse
3
4
5
from pydantic import BaseModel


Que Nguyen's avatar
Que Nguyen committed
6
7
def get_filtered_results(results, filter_list):
    if not filter_list:
8
9
10
11
        return results
    filtered_results = []
    for result in results:
        domain = urlparse(result["url"]).netloc
Que Nguyen's avatar
Que Nguyen committed
12
        if any(domain.endswith(filtered_domain) for filtered_domain in filter_list):
13
14
15
            filtered_results.append(result)
    return filtered_results

16
17
18
19
class SearchResult(BaseModel):
    link: str
    title: Optional[str]
    snippet: Optional[str]