torbot 3.0.6__tar.gz → 3.0.8__tar.gz

{torbot-3.0.6 → torbot-3.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: torbot
-Version: 3.0.6
+Version: 3.0.8
 Summary: OSINT for the dark web
 License: GNU GPL
 Author: Akeem King

{torbot-3.0.6 → torbot-3.0.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "torbot"
-version = "3.0.6"
+version = "3.0.8"
 description = "OSINT for the dark web"
 authors = ["Akeem King <akeemtlking@gmail.com>"]
 license = "GNU GPL"

{torbot-3.0.6 → torbot-3.0.8}/torbot/modules/config.py

@@ -3,11 +3,11 @@ import logging
 import sys
 
 from dotenv import load_dotenv
+from inspect import getsourcefile
 
-script_directory = os.path.dirname(os.path.abspath(sys.argv[0]))
-print(script_directory)
 
-load_dotenv(verbose=True)
+config_file_path = (os.path.abspath(getsourcefile(lambda:0)))
+load_dotenv(config_file_path, verbose=True)
 port = os.getenv("PORT")
 host = os.getenv("HOST")
 
@@ -24,11 +24,14 @@ def get_log_level() -> int:
 
 def get_data_directory(): 
     data_directory = os.getenv('TORBOT_DATA_DIR')
+
+    # if a path is not set, write data to the config directory
+    config_directory = os.path.dirname(config_file_path)
     if not data_directory:
-        data_directory = 'data'
+        data_directory = config_directory
 
     if data_directory.strip() == "":
-        data_directory = 'data'    
+        data_directory = config_directory 
 
     # create directory if it doesn't exist
     if not os.path.exists(data_directory):