Code for How to Create a Custom Wordlist in Python Tutorial

wordlist_gen.py

# Import the argparse module for handling command line arguments.
# Import the itertools module for generating combinations.
import argparse, itertools


# Define a function to generate a wordlist based on given parameters.
def generate_wordlist(characters, min_length, max_length, output_file):
    # Open the output file in write mode.
    with open(output_file, 'w') as file:
        # Iterate over the range of word lengths from min_length to max_length.
        for length in range(min_length, max_length + 1):
            # Generate all possible combinations of characters with the given length.
            for combination in itertools.product(characters, repeat=length):
                # Join the characters to form a word and write it to the file
                word = ''.join(combination)
                file.write(word + '\n')


# Create an ArgumentParser object for handling command line arguments.
parser = argparse.ArgumentParser(description="Generate a custom wordlist similar to crunch.")

# Define command line arguments.
parser.add_argument("-c", "--characters", type=str, default="abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789",
                    help="Set of characters to include in the wordlist")
parser.add_argument("-min", "--min_length", type=int, default=4, help="Minimum length of the words")
parser.add_argument("-max", "--max_length", type=int, default=6, help="Maximum length of the words")
parser.add_argument("-o", "--output_file", type=str, default="custom_wordlist.txt", help="Output file name")

# Parse the command line arguments.
args = parser.parse_args()

# Call the generate_wordlist function with the provided arguments.
generate_wordlist(args.characters, args.min_length, args.max_length, args.output_file)

# Print a message indicating the wordlist has been generated and saved.
print(f"[+] Wordlist generated and saved to {args.output_file}")