pyrit_cli.py

#!/usr/bin/env python3

import sys

class PyritRuntimeError(Exception):
    """Custom exception for Pyrit CLI errors."""
    pass

class Pyrit_CLI:
    def __init__(self):
        pass

    def initFromArgv(self):
        if len(sys.argv) <= 1 or sys.argv[1] in ("-h", "--help"):
            self.print_help()
        else:
            print(f"Unrecognized option: {sys.argv[1]}")
            raise PyritRuntimeError("Invalid argument")

    def print_help(self):
        print("""\
Pyrit 0.4.0 - WPA/WPA2-PSK attack tool

Usage: pyrit <command> [options]

Commands:
  analyze             Analyze captured data
  batch               Batch-process packets
  benchmark           Run CPU benchmarks
  help                Show this help message

For more help on a command, run: pyrit help <command>
""")


# Entry point for /usr/local/bin/pyrit
if __name__ == "__main__":
    try:
        Pyrit_CLI().initFromArgv()
    except PyritRuntimeError as e:
        print(f"Error: {e}")
        sys.exit(1)