get shodan api key from /home/josue environment variable

Josue87 · Josue87 · commit 24248c994adf · 2019-10-28T10:25:55.000+01:00
diff --git a/modules/shodan/_shodan.py b/modules/shodan/_shodan.py
@@ -3,6 +3,7 @@
 from utildata.dataset_options import Option
 from utils.shodan_search import shodan_search
 from utils.shell_options import ShellOptions
+from os import getenv
 
 
 class ShodanModule(Module):
@@ -17,6 +18,16 @@ def __init__(self, information, opts=None, default_file_name="./files/shodan.txt
             options.update(opts)
         # Constructor of the parent class
         super(ShodanModule, self).__init__(information, options)
+        # Set Shodan key if exists in $HOME/.shodan/api_key and it doesn't have a global value.
+        if not self.args["apishodan"]:
+            try:
+                home = getenv("HOME")
+                shodan_key_path = home + "/.shodan/api_key"
+                with open(shodan_key_path, "r") as f:
+                    key = f.read().strip()
+                    self.set_value(["apishodan", key])
+            except:
+                pass
 
     # This function must be always implemented, it is called by the run option
     def run(self): 
