Source code for dirhunt.sources.base
from dirhunt.pool import Pool
[docs]class Source(Pool):
def __init__(self, result_callback, error_callback, max_workers=None):
super(Source, self).__init__(max_workers)
self.result_callback = result_callback
self.error_callback = error_callback
[docs] def add_domain(self, domain):
self.submit(domain)
[docs] def callback(self, domain):
raise NotImplementedError
[docs] def add_result(self, url):
if self.result_callback:
self.result_callback(url)
[docs] def add_error(self, message):
if self.error_callback:
self.error_callback(message)