Use With Statement for Temporary Directory

Signed-off-by: dusoleil <howcansocksbereal@gmail.com>

1 files changed, 4 insertions, 4 deletions

diff --git a/sploit/main.py b/sploit/main.py
index ebcbd41..6d181b8 100644
--- a/sploit/main.py
+++ b/sploit/main.py
@@ -25,10 +25,10 @@ def main():
 
 def daemon(script):
     print("Running in Pipe Daemon Mode...")
-    tmpdir = tempfile.TemporaryDirectory()
-    tmp = os.path.split(tmpdir.name)[1]
-    while(True):
-        runscript(script,Comm(Pipes(tmp)));
+    with tempfile.TemporaryDirectory() as tmpdir:
+        tmp = os.path.split(tmpdir)[1]
+        while(True):
+            runscript(script,Comm(Pipes(tmp)));
 
 def pipe(script):
     print("Running in Pipe Mode...");