ℹ️

ℹ️ fix
securisec · Nov 30, 2019 · 24475ae · 24475ae
1 parent a8d840d
commit 24475ae
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/chepy/modules/forensics.py b/chepy/modules/forensics.py
@@ -2,7 +2,6 @@
 import os
 import logging
 import pathlib
-import shutil
 import tempfile
 import pprint
 
@@ -54,7 +53,7 @@ def get_mime(self, set_state: bool = False):
             logging.info(mimetype)
         if set_state:
             self.state = mimetype
-        shutil.rmtree(filename)
+        # pathlib.Path(filename).unlink()
         return self
 
     @ChepyDecorators.call_stack
@@ -81,7 +80,7 @@ def get_metadata(self, set_state: bool = False):
                 self.state = meta
             else:  # pragma: no cover
                 logging.info(pprint.pformat(meta))
-        shutil.rmtree(filename)
+        # pathlib.Path(filename).unlink()
         return self
 
     @ChepyDecorators.call_stack
@@ -103,7 +102,7 @@ def embedded_files(self, extract_path: str = None):
             subfile.setOutput(extract_path)
         subfile.loadParsers()
         subfile.main()
-        shutil.rmtree(filename)
+        # pathlib.Path(filename).unlink()
         return self
 
     @ChepyDecorators.call_stack
@@ -118,5 +117,5 @@ def get_exif(self):  # pragma: no cover
             exif = et.get_metadata(filename)
             if exif:
                 self.state = exif
-        shutil.rmtree(filename)
+        # pathlib.Path(filename).unlink()
         return self