Formas geométricas

import time
def desenhar_triangulo(linhas):
    print("\n")
    for i in range(1, linhas + 1):
        print(' ' * (linhas - i) + '*' * (2 * i - 1))
    time.sleep(2)
    print("\n")
def desenhar_quadrado(lado):
    for i in range(lado):
        print('*' * lado)
    print("\n")
    time.sleep(2)
def desenhar_retangulo(largura, altura):
    for i in range(altura):
        if i == 0 or i == altura - 1:
            print('*' * largura)
        else:
            print('*' + ' ' * (largura - 2) + '*')
    print("\n")
    time.sleep(2)
def desenhar_circulo(raio):
    for i in range(-raio, raio + 1):
        for j in range(-raio, raio + 1):
            if i ** 2 + j ** 2 <= raio ** 2:
                print('*', end='')
            else:
                print(' ', end='')
        print()
    time.sleep(2)

def main():
    while True:
        print("\033[94m=\033[m" * 60)
        print("\033[34m\t\t\t\tDesenhar Figuras Geométricas\033[m")
        print("\033[94m=\033[m"*60)
        print("\033[92m\t\t\t     1 - Triângulo\033[m")
        print("\033[92m\t\t\t     2 - Quadrado\033[m")
        print("\033[92m\t\t\t     3 - Retângulo\033[m")
        print("\033[92m\t\t\t     4 - Círculo\033[m")
        print("\033[31m\t\t\t     5 - Sair\033[m")
        opcao = str(input("\033[93mEscolha uma opção (1, 2, 3, 4 ou 5): \033[m"))
        if opcao == "1":
            try:
                print("\n")
                linhas = int(input("\033[96mDigite o número de linhas do triângulo: \033[m "))
                desenhar_triangulo(linhas)
            except ValueError:
                print("\033[31mErro! Deve digitar apenas números inteiros.\033[m")
        elif opcao == "2":
            try:
                print("\n")
                lado = int(input("\033[96mDigite o tamanho do lado do quadrado: \033[m "))
                desenhar_quadrado(lado)
            except ValueError:
                 print("\033[31mErro! Deve digitar apenas números inteiros.\033[m")
        elif opcao == "3":
            try:
                largura = int(input("\033[96mDigite a largura do retângulo: \033[m "))
                altura = int(input("\033[96mDigite a altura do retângulo: \033[m"))
                desenhar_retangulo(largura, altura)
            except ValueError:
                print("\033[31mErro! Deve digitar apenas números inteiros.\033[m")
        elif opcao == "4":
            try:
                print("\n")
                raio = int(input("\033[96mDigite o raio do círculo:  \033[m "))
                print("\n")
                desenhar_circulo(raio)

            except ValueError:
                print("\033[31mErro! Deve digitar apenas números inteiros.\033[m")
        elif opcao == "5":
            time.sleep(2)
            print("Sair do programa.")
if __name__ == "__main__":
    main()
Mundo em Python

Formas geométricas
