def getDomainStatus(domain): #key is linked to my Gmail account and will limit to # 10,000 lookups per day. api_key = 'ABQIAAAA_ZP3Gp39cTLiKArCd3K0GhQs0x-2rr_UO6hjtMF0uHW-GkhMSA' gsb = SafebrowsinglookupClient(api_key, 0, 0) return gsb.lookup(domain)
def test_errors(self): client = SafebrowsinglookupClient("AAAAAAAAaaAAAaAAAaA0a0AaaAAAAAAa0AaAAAaAa0aaaAaAa0Aa0AaaaA") results = client.lookup(*["http://www.google.com/", "http://www.google.org/"]) self.assertEquals(2, len(results)) self.assertEquals("error", results["http://www.google.com/"]) self.assertEquals("error", results["http://www.google.org/"])
def test_errors(self): client = SafebrowsinglookupClient( 'AAAAAAAAaaAAAaAAAaA0a0AaaAAAAAAa0AaAAAaAa0aaaAaAa0Aa0AaaaA') results = client.lookup( *['http://www.google.com/', 'http://www.google.org/']) self.assertEquals(2, len(results)) self.assertEquals('error', results['http://www.google.com/']) self.assertEquals('error', results['http://www.google.org/'])
class SafebrowsinglookupClient_ParseTestCase(unittest.TestCase): def setUp(self): self.client = SafebrowsinglookupClient('insert_API_key_here') def test_errors(self): client = SafebrowsinglookupClient( 'AAAAAAAAaaAAAaAAAaA0a0AaaAAAAAAa0AaAAAaAa0aaaAaAa0Aa0AaaaA') results = client.lookup( *['http://www.google.com/', 'http://www.google.org/']) self.assertEquals(2, len(results)) self.assertEquals('error', results['http://www.google.com/']) self.assertEquals('error', results['http://www.google.org/']) def test_no_match(self): if self.client.key == 'insert_API_key_here': return results = self.client.lookup( *['http://www.google.com/', 'http://www.google.org/']) self.assertEquals(2, len(results)) self.assertEquals('ok', results['http://www.google.com/']) self.assertEquals('ok', results['http://www.google.org/']) def test_match(self): if self.client.key == 'insert_API_key_here': return results = self.client.lookup('http://www.gumblar.cn/') self.assertEquals(1, len(results)) self.assertEquals('malware', results['http://www.gumblar.cn/']) def test_match_many(self): if self.client.key == 'insert_API_key_here': return urls = [] for i in range(1, 600): urls.append("http://www.gumblar.cn/" + str(i)) results = self.client.lookup(*urls) self.assertEquals(len(urls), len(results)) self.assertEquals('malware', results['http://www.gumblar.cn/1']) self.assertEquals('malware', results['http://www.gumblar.cn/500']) self.assertEquals('malware', results['http://www.gumblar.cn/599'])
class SafebrowsinglookupClient_ParseTestCase(unittest.TestCase): def setUp(self): self.client = SafebrowsinglookupClient("insert_API_key_here") def test_errors(self): client = SafebrowsinglookupClient("AAAAAAAAaaAAAaAAAaA0a0AaaAAAAAAa0AaAAAaAa0aaaAaAa0Aa0AaaaA") results = client.lookup(*["http://www.google.com/", "http://www.google.org/"]) self.assertEquals(2, len(results)) self.assertEquals("error", results["http://www.google.com/"]) self.assertEquals("error", results["http://www.google.org/"]) def test_no_match(self): if self.client.key == "insert_API_key_here": return results = self.client.lookup(*["http://www.google.com/", "http://www.google.org/"]) self.assertEquals(2, len(results)) self.assertEquals("ok", results["http://www.google.com/"]) self.assertEquals("ok", results["http://www.google.org/"]) def test_match(self): if self.client.key == "insert_API_key_here": return results = self.client.lookup("http://www.gumblar.cn/") self.assertEquals(1, len(results)) self.assertEquals("malware", results["http://www.gumblar.cn/"]) def test_match_many(self): if self.client.key == "insert_API_key_here": return urls = [] for i in range(1, 600): urls.append("http://www.gumblar.cn/" + str(i)) results = self.client.lookup(*urls) self.assertEquals(len(urls), len(results)) self.assertEquals("malware", results["http://www.gumblar.cn/1"]) self.assertEquals("malware", results["http://www.gumblar.cn/500"]) self.assertEquals("malware", results["http://www.gumblar.cn/599"])
def retrieveSafeBrowseringResutls(): print("Connecting to SafebrowsinglookupClient...") #key = 'ABQIAAAAMW5c1P9_7qTcpx8drxoM8xQXc-3IJoTDvJwbFxcncj5ENdNsVw' key = 'AIzaSyAWHi27ImtGlIc2APppC4lGlU1N01fiE1c' client = SafebrowsinglookupClient(key) if(client): print("Connected successfully to SafebrowsinglookupClient!") print("Running lookup ...") results = client.lookup(*domains.keys()); for url in results: if (results[url] != 'ok'): print url + ',' + results[url] if (results[url] == "malware" or results[url] == "phishing" or results[url] == "malware,phishing"): malwareDomains.append(url) print("Number of Malware Domains : " + str(len(malwareDomains)) + " Out of: " + str(len(domains)))
def setUp(self): self.client = SafebrowsinglookupClient('insert_API_key_here')
def setUp(self): self.client = SafebrowsinglookupClient("insert_API_key_here")