pattern_drawing.py

# 🖼️ Python Pattern Drawing Project
from colorama import Fore
# Step 1: Display a menu to the user
print("🌟 Welcome to the Python Pattern Drawing Program!")
print("Choose a pattern type:")
print("1. Lettered Right-angled Triangle")
print("2. Square with Hollow Center")
print("3. Diamond")
print("4. Left-angled Triangle")
print("5. Emoji Square")  # 5 seemed the same as 2, so I modified it
print("6. Christmas Pyramid")
print("7. Reverse Pyramid")
print("8. Rectangle with Hollow Center")
print("9. Butterfly")
print("10. Cross")

# Step 2: Get the user's choice
choice = int(input(Fore.LIGHTWHITE_EX + "Enter the number corresponding to your choice: "))

rows = 0
size = 0

quit_message = False

while not quit_message:

    # Step 3: Get dimensions based on choice
    if choice in [1, 3, 4, 6, 7]:  # Patterns that need the number of rows
        rows = int(input("Enter the number of rows: "))
    elif choice in [2, 5, 9, 10]:  # removed choice 8 since it doesn't need size
        size = int(input("Enter the size of the figure: "))

    # Step 4: Generate the selected pattern
    if choice == 1:  # Right-angled Triangle
        for row in range(1, rows + 1):
            color = ''
            if row % 2 == 0:
                color = f'{Fore.LIGHTRED_EX + chr(64 + row)} '
            else:
                color = f'{Fore.LIGHTBLUE_EX + chr(64 + row)} '
            print(color * row)

        """
        # a more basic solutions, w/o colors and letters:
        char = '*'
        for row in range(1, rows + 1):
            print(char * row)

        # or a solution w/ nested loops:
        char = '*'
        for row in range(1, rows + 1):
            for _ in range(row):
                print(char, end='')
            print()
        """

    elif choice == 2:  # Square with Hollow Center
        for s in range(1, size + 1):
            color = Fore.LIGHTMAGENTA_EX + '*  '
            # for the figure to look like a square rather than a rectangle,
            # there should be two extra spaces after the star,
            space = '   '  # and each 'space' needs to be three spaces long
            if s == 1 or s == size:
                print(color * size)
            else:
                print(f'{color}{(size - 2) * space}{color}')


    elif choice == 3:  # Diamond
        half_rows = int(rows / 2)
        color = ''
        space = '   '  # I think the extra spaces make a better looking diamond :)
        if rows % 2 == 0:
            for row in range(1, rows + 1):
                color = Fore.LIGHTGREEN_EX + ' $ '
                if row <= half_rows:
                    print(f"{(half_rows - row) * space}{color * (row + row - 1)}")
            for row in range(half_rows, -1, -1):
                color = Fore.LIGHTRED_EX + ' $ '
                print(f"{(half_rows - row) * space}{color * (row + row - 1)}")
        else:
            for row in range(1, rows + 1):
                if row < half_rows + 1:
                    color = Fore.RED + ' $ '
                    print(f"{(half_rows + 1 - row) * space}{color * (row + row - 1)}")
                elif row == half_rows + 1:
                    color = Fore.LIGHTYELLOW_EX + ' $ '
                    print(f"{(half_rows + 1 - row) * space}{color * (row + row - 1)}")
            for row in range(half_rows, -1, -1):
                color = Fore.GREEN + ' $ '
                print(f"{(half_rows + 1 - row) * space}{color * (row + row - 1)}")


    elif choice == 4:  # Left-angled Triangle
        for row in range(rows, -1, -1):
            if rows % 2 == 0:
                if row % 2 != 0:
                    print('♥️' * row)
                else:
                    print('♠️' * row)
            else:
                if row % 2 != 0:
                    print('♦️️' * row)
                else:
                    print('♣️️' * row)

    elif choice == 5:  # Hollow Square
        for s in range(1, size + 1):
            char = '😜 '
            # for the figure to look like a square rather than a rectangle,
            # there should be two extra spaces after the star,
            space = '🫥 '  # and each 'space' needs to be three spaces long
            if s == 1 or s == size:
                print(char * size)
            else:
                print(f'{char}{(size - 2) * space}{char}')

    elif choice == 6:  # Christmas Pyramid
        for row in range(rows):
            char1 = '  '
            if row == 0:
                char2 = Fore.LIGHTYELLOW_EX + '* '
            elif row % 2 != 0:
                char2 = Fore.LIGHTGREEN_EX + '* '
            else:
                char2 = Fore.LIGHTRED_EX + '* '
            print(char1 * (rows - (row + 1)), end='')
            print(char2 * (row + 1), end='')
            print(char2 * row)

    elif choice == 7:  # Reverse Pyramid
        char1 = '  '
        char2 = Fore.LIGHTYELLOW_EX + '* '
        for row in range(rows - 1, -1, -1):
            print(char1 * (rows - (row + 1)), end='')
            print(char2 * (row + 1), end='')
            print(char2 * row)

    elif choice == 8:  # Rectangle with Hollow Center
        width = int(input("Enter the width of the rectangle: "))
        height = int(input("Enter the height of the rectangle: "))
        char1 = Fore.CYAN + '*  '
        char2 = '   '
        for h in range(1, height + 1):
            if h == 1 or h == height:
                print(width * char1)
            else:
                print(f'{char1}{(width - 2) * char2}{char1}')

    elif choice == 9:  # Butterfly - an additional pattern
        char1 = Fore.LIGHTRED_EX + '* '
        char2 = '  '
        char3 = Fore.LIGHTYELLOW_EX + '* '
        for num in range(1, size):
            print(char3, end='')
            print((num - 1) * char1, end="")
            print(char2 * (2 * (size - num)), end="")
            print((num - 1) * char1, end='')
            print(char3)
        for num in range(size, 0, -1):
            print(num * char1, end="")
            print(char2 * (2 * (size - num)), end="")
            print(num * char1)

    elif choice == 10:  # Cross
        half_range = int(size / 2)
        char1 = '   '
        char2 = Fore.LIGHTYELLOW_EX + '✝  '
        if size % 2 == 0:
            for char in range(half_range - 1):
                print(char1 * (half_range - 1), end='')
                print(char2 * 2)
            print(char2 * size)
        else:
            for char in range(half_range):
                print(char1 * half_range, end='')
                print(char2)
            print(char2 * size)
        if size % 2 == 0:
            print(char2 * size)
            for char in range(half_range - 1):
                print(char1 * (half_range - 1), end='')
                print(char2 * 2)
        else:
            for char in range(half_range):
                print(char1 * half_range, end='')
                print(char2)

    else:
        print("❌ Invalid choice! Please restart the program.")

    # Step 5: Optional - Allow the user to restart or exit
    choice = int(input(Fore.LIGHTWHITE_EX + "Enter the number corresponding to your choice or enter 0 to quit: "))
    if choice == 0:
        quit_message = True