Merge pull request #118 from surbhishah/cleanup-reduce-disk-usage

Adding clean up function to reduce disk usage

Author: dxa4481

truffleHog/truffleHog.py

@@ -281,7 +281,16 @@ def find_strings(git_url, since_commit=None, max_depth=1000000, printJson=False,
         output = handle_results(output, output_dir, foundIssues)
     output["project_path"] = project_path
     output["clone_uri"] = git_url
+    output["issues_path"] = output_dir
     return output
 
+def clean_up(output):
+    project_path = output.get("project_path", None)
+    if project_path and os.path.isdir(project_path):
+        shutil.rmtree(output["project_path"])
+    issues_path = output.get("issues_path", None)
+    if issues_path and os.path.isdir(issues_path):
+        shutil.rmtree(output["issues_path"])
+
 if __name__ == "__main__":
     main()