scanner.py

import subprocess
import datetime

def run_nmap(target):
    """Run Nmap scan on the target"""
    try:
        result = subprocess.getoutput(f"nmap -sV {target}")
        return result
    except Exception as e:
        return f"Error running Nmap: {e}"

def run_nikto(target):
    """Run Nikto scan on the target"""
    try:
        result = subprocess.getoutput(f"nikto -h {target}")
        return result
    except Exception as e:
        return f"Error running Nikto: {e}"

def save_report(target, nmap_result, nikto_result):
    """Save the scan results into a text report"""
    filename = f"report_{target}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.txt"
    with open(filename, "w") as f:
        f.write("==== Automated Vulnerability Scan Report ====\n")
        f.write(f"Target: {target}\n")
        f.write(f"Date: {datetime.datetime.now()}\n\n")

        f.write("---- Nmap Results ----\n")
        f.write(nmap_result + "\n\n")

        f.write("---- Nikto Results ----\n")
        f.write(nikto_result + "\n\n")

    print(f"\n[+] Report saved as {filename}")

def main():
    print("==== Automated Vulnerability Scanner (CLI Version) ====")
    target = input("Enter Target IP/Domain: ").strip()

    print("\n[+] Running Nmap Scan...")
    nmap_result = run_nmap(target)
    print(nmap_result)

    print("\n[+] Running Nikto Scan...")
    nikto_result = run_nikto(target)
    print(nikto_result)

    save_report(target, nmap_result, nikto_result)

if __name__ == "__main__":
    main()