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)