pentestdir.py

from pathlib import Path

folders = {
    "documentation": ["reports", "notes", "screenshots"],
    "credentials": ["passwords", "keys", "tokens"],
    "logs": ["activity_logs", "error_logs", "audit_logs"],
    "targets": ["clients", "servers", "network"],
    "scans": ["nmap", "nessus", "burp", "custom"],
    "exploits": ["vulnerabilities", "privilege_escalation", "payloads"],
    "enum": ["network", "services", "web"],
    "tools": ["scripts", "utilities", "wordlists"],
    "reports": ["executive_summary", "technical_details"],
    "evidence": ["screenshots", "log_files", "captures"],
    "backup": ["source_code", "databases"],
    "sandbox": ["experiments"]
}

def create_subfolders(parent_path, subfolder_list):
    for subfolder in subfolder_list:
        (parent_path / subfolder).mkdir(parents=True, exist_ok=True)

def create_folders(base_path, folders_dict):
    for folder_name, subfolders in folders_dict.items():
        folder_path = Path(base_path) / folder_name
        folder_path.mkdir(parents=True, exist_ok=True)
        create_subfolders(folder_path, subfolders)

if __name__ == "__main__":
    base_directory = "pentest_project"  # Replace with the desired base directory
    create_folders(base_directory, folders)
    print("Penetration testing directory structure created successfully.")