Fixed Missing Working Directories

klezVirus · Dec 28, 2021 · 432c28f · 432c28f
1 parent f4c271c
commit 432c28f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/manage.py b/manage.py
@@ -1,6 +1,7 @@
 import argparse
 import configparser
 import os
+from pathlib import Path
 
 import requests
 from colorama import Fore
@@ -14,9 +15,13 @@ class Manager:
     def __init__(self):
         self.config = configparser.ConfigParser(allow_no_value=True, interpolation=configparser.ExtendedInterpolation())
            self.config.read(get_project_root().joinpath("config", "config.ini"))
+        workspaces = get_project_root().joinpath("data", "workspaces")
+        temp = get_project_root().joinpath("data", "temp")
+        workspaces.mkdir(exist_ok=True)
+        temp.mkdir(exist_ok=True)
 
     def run(self, _action, _workspace, **kwargs):
-        kwargs["config"] = get_project_root().joinpath("config", "zzz_doosan/config.ini")
+        kwargs["config"] = get_project_root().joinpath("config", "config.ini")
         action_class = Action.from_name(_action)
         if not action_class:
             error(f"Unknown action: {_action}")

diff --git a/scripts/sn0int-to-json.py b/scripts/sn0int-to-json.py
@@ -2,7 +2,7 @@
 
 leaks = []
 leak = {}
-with open("../zzz_doosan/leaked.txt") as infile:
+with open("../leaked.txt") as infile:
     uid = None
     mail = None
     creds = []
@@ -18,5 +18,5 @@
             cred = line.split(" ")[1].replace("\"", "").replace("\n", "").strip(" ()")
             creds.append(cred)
 
-with open("../zzz_doosan/leaked.json", "w") as outfile:
+with open("../leaked.json", "w") as outfile:
     json.dump(leaks, outfile)