diff --git a/lib/modules/crawler/crawler.py b/lib/modules/crawler/crawler.py index b142301..9f93d30 100644 --- a/lib/modules/crawler/crawler.py +++ b/lib/modules/crawler/crawler.py @@ -23,13 +23,13 @@ class SitadelSpider(CrawlSpider): ] # Method for parsing items - def parse_items(self, links): + @classmethod + def parse_items(cls, links): for link in links: if urlparse(link.url).netloc in allowed_domains: urls.add(link.url) yield link - def crawl(url, user_agent): try: output = Services.get("output") diff --git a/lib/utils/banner.py b/lib/utils/banner.py index 93801e3..c8c17c7 100644 --- a/lib/utils/banner.py +++ b/lib/utils/banner.py @@ -24,11 +24,13 @@ def banner(self): print(self.g + "~/#" + self.e + " https://github.com/shenril/Sitadel" + self.g + " #\\~" + self.e) print("\n") - def preamble(self, url): + @classmethod + def preamble(cls, url): print('URL: %s' % url) print('--------- Scan Started: %s ---------' % (time.strftime('%d/%m/%Y %H:%M:%S'))) - - def postscript(self): + + @classmethod + def postscript(cls): print('--------- Scan Finished: %s ---------' % (time.strftime('%d/%m/%Y %H:%M:%S'))) def version(self):