diff --git a/plugins/censys/plugin.py b/plugins/censys/plugin.py index 46ca5bd..6ec3664 100644 --- a/plugins/censys/plugin.py +++ b/plugins/censys/plugin.py @@ -1,24 +1,27 @@ -from censys.search import CensysCertificates +from censys.search import CensysCerts from core.env import SIGINT_handler import signal from core.logger import Output -NAME = 'censys' -ARG_HELP = 'censys subdomain certificates' +NAME = 'censys' +ARG_HELP = 'censys subdomain certificates' handler = SIGINT_handler() signal.signal(signal.SIGINT, handler.signal_handler) + class CENSYSError(Exception): - """Base class for censys exceptions""" - pass + """Base class for censys exceptions""" + pass + def execute(domain, config, **kwargs): if handler.SIGINT: Output().warn("Aborted plugin: %s" % NAME, False) return None try: - c = CensysCertificates(api_id=config['uid'], api_secret=config['secret']) + c = CensysCertificates( + api_id=config['uid'], api_secret=config['secret']) fields = [ "parsed.names", "parsed.subject.common_name", @@ -27,15 +30,15 @@ def execute(domain, config, **kwargs): results = [] for page in c.search(domain, fields, max_records=config['max_records']): results.append(page) - #Flatten json to array + # Flatten json to array list = [] for x in results: if x.get('parsed.namesn'): list += (x.get('parsed.names')) - + if x.get('parsed.subject.common_name'): list += x.get('parsed.subject.common_name') - + if x.get('parsed.extensions.subject_alt_name.dns_names'): list += x.get('parsed.extensions.subject_alt_name.dns_names') @@ -45,5 +48,5 @@ def execute(domain, config, **kwargs): subdomains = sorted(set(subdomains)) return subdomains except Exception as E: - print (E) + print(E) raise