google_cloud_webrisk_uri_threat_detection_quickstart.py

1from google.cloud import webrisk_v1
2
3def search_uri(uri: str):
4    """Checks whether a URI is on any of the threat lists."""
5    client = webrisk_v1.WebRiskServiceServiceClient()
6
7    # The threat types we want to check for.
8    threat_types = [
9        webrisk_v1.ThreatType.MALWARE,
10        webrisk_v1.ThreatType.SOCIAL_ENGINEERING,
11        webrisk_v1.ThreatType.UNWANTED_SOFTWARE,
12    ]
13
14    request = webrisk_v1.SearchUrisRequest()
15    request.uri = uri
16    request.threat_types = threat_types
17
18    response = client.search_uris(request=request)
19
20    if response.threat:
21        print(f"The URI {uri} is unsafe.")
22        print(f"Threat type: {response.threat.threat_types}")
23    else:
24        print(f"The URI {uri} is safe.")
25
26if __name__ == "__main__":
27    # Example URI to check
28    search_uri("http://testsafebrowsing.appspot.com/s/malware.html")