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 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")