Fixing __init__.py files to make imports easier, enabling running Osint scripts using import, fix virustotal module to prevent breaking on import

Author: KunalAggarwal

__init__.py

@@ -0,0 +1,11 @@
+from __future__ import absolute_import
+
+from . import username, emails, domain, ip, domainOsint, emailOsint, ipOsint, usernameOsint, datasploit
+
+__all__ = ["username", "emails", "domain", "ip", "domainOsint", "emailOsint", "ipOsint", "usernameOsint", "datasploit"]
+__title__ = 'datasploit'
+__version__ = '1.0'
+__author__ = 'Shubham Mittal, Sudhanshu Chauhan, Kunal Aggarwal'
+__license__ = 'GPL-3.0'
+
+del absolute_import

domain/__init__.py

@@ -1,5 +1,10 @@
-from os.path import dirname, basename, isfile
-import glob
+from os.path import dirname, basename, isfile, abspath
+import glob, importlib, sys
 
 modules = glob.glob(dirname(__file__) + "/domain_*.py")
 __all__ = [basename(f)[:-3] for f in modules if isfile(f)]
+sys.path.append(dirname(abspath(__file__)))
+
+for m in __all__:
+        __import__(m, locals(), globals())
+del m, f, dirname, basename, isfile, abspath, glob, importlib, sys, modules

emails/__init__.py

@@ -1,5 +1,10 @@
-from os.path import dirname, basename, isfile
-import glob
+from os.path import dirname, basename, isfile, abspath
+import glob, importlib, sys
 
 modules = glob.glob(dirname(__file__) + "/email_*.py")
 __all__ = [basename(f)[:-3] for f in modules if isfile(f)]
+sys.path.append(dirname(abspath(__file__)))
+
+for m in __all__:
+        __import__(m, locals(), globals())
+del m, f, dirname, basename, isfile, abspath, glob, importlib, sys, modules

ip/__init__.py

@@ -1,5 +1,10 @@
-from os.path import dirname, basename, isfile
-import glob
+from os.path import dirname, basename, isfile, abspath
+import glob, importlib, sys
 
 modules = glob.glob(dirname(__file__) + "/ip_*.py")
 __all__ = [basename(f)[:-3] for f in modules if isfile(f)]
+sys.path.append(dirname(abspath(__file__)))
+
+for m in __all__:
+        __import__(m, locals(), globals())
+del m, f, dirname, basename, isfile, abspath, glob, importlib, sys, modules

ip/ip_virustotal.py

@@ -9,7 +9,6 @@
 
 # Control whether the module is enabled or not
 ENABLED = True
-api = cfg.virustotal_public_api
 
 class style:
     BOLD = '\033[1m'
@@ -24,6 +23,7 @@ def banner():
 def main(ip):
     # Use the ip variable to do some stuff and return the data
     print ip
+    api = cfg.virustotal_public_api
     params = "{'ip': '%s', 'apikey': '%s'}" % (ip, api)
     url = "http://www.virustotal.com/vtapi/v2/ip-address/report?ip=%s&apikey=%s" % (ip, api)
     req = requests.get(url, params)

osint_runner.py

@@ -5,9 +5,9 @@
 
 
 def run(component, module_dir, m_input):
-    dir_path = os.path.abspath(module_dir)
+    dir_path = "%s/%s" % (os.path.dirname(os.path.abspath(__file__)), module_dir)
     sys.path.insert(0, dir_path)
-    domain_files = glob("%s/%s_*.py" % (module_dir, component))
+    domain_files = glob("%s/%s_*.py" % (dir_path, component))
     active_modules = []
     for index, i in enumerate(domain_files):
         module_name = os.path.basename(os.path.splitext(i)[0])

username/__init__.py

@@ -1,5 +1,10 @@
-from os.path import dirname, basename, isfile
-import glob
+from os.path import dirname, basename, isfile, abspath
+import glob, importlib, sys
 
 modules = glob.glob(dirname(__file__) + "/username_*.py")
 __all__ = [basename(f)[:-3] for f in modules if isfile(f)]
+sys.path.append(dirname(abspath(__file__)))
+
+for m in __all__:
+	__import__(m, locals(), globals())
+del m, f, dirname, basename, isfile, abspath, glob, importlib, sys, modules