import sys


def checkMatriz(colunaMatrizA, linhaMatrizB):
    # print(colunaMatrizA, linhaMatrizB)
    if int(colunaMatrizA) != int(linhaMatrizB):
        print(f'\033[1;31mErro: As dimensões das matrizes não são compatíveis para multiplicação.\033[m')
        print(f'\033[1;91mPrograma Finalizado!!!!!!.\033[m')
        return sys.exit()
    else:
        print()
        print(f'\033[1;96mAs dimensões das matrizes são compatíveis para multiplicação. '
              f'\nVamos continuar!!!!!.\033[m')
        print()
        return


def criar_matriz(num_linhas, num_colunas):
    matriz = []
    for i in range(num_linhas):
        linha = []
        for j in range(num_colunas):
            while True:
                try:
                    valor = int(input(f'Digite o {j + 1}º elemento da {i + 1}ª linha: '))
                    linha.append(valor)
                    break
                except ValueError:
                    print(f'\033[1;31mValor INVÁLIDO! Digite apenas números!\033[m')
        matriz.append(linha)
    return matriz


def getLinha(matriz, n):
    return [i for i in matriz[n]]


def getColuna(matriz, n):
    return [i[n] for i in matriz]


print(f'\033[1;34mIniciando o programa de multiplicação de matrizes.\033[m')
print(f'\033[1;33mDeterminando as ordens das matrizes.\033[m')
print(f'\033[1;33mPrimeira Matriz.\033[m')

while True:
    try:
        numero_linhas_1 = int(input('Digite o número de linhas da matriz: '))
        numero_colunas_1 = int(input('Digite o número de colunas da matriz: '))
        break
    except ValueError:
        print(f'\033[1;31mValor INVÁLIDO! Digite apenas números!\033[m')

print()
print(f'\033[1;33mSegunda Matriz.\033[m')

while True:
    try:
        numero_linhas_2 = int(input('Digite o número de linhas da matriz: '))
        numero_colunas_2 = int(input('Digite o número de colunas da matriz: '))
        break
    except ValueError:
        print(f'\033[1;31mValor INVÁLIDO! Digite apenas números!\033[m')

checkMatriz(numero_colunas_1, numero_linhas_2)

print(f'\033[1;33mCriando a primeira Matriz.\033[m')
matriz_A = criar_matriz(numero_linhas_1, numero_colunas_1)
print()
print(f'\033[1;33mCriando a segunda Matriz.\033[m')
matriz_B = criar_matriz(numero_linhas_2, numero_colunas_2)

matrizResultante = []
for i in range(len(matriz_A)):
    matrizResultante.append([])

    for j in range(len(matriz_B[0])):
        print(f'getLinha(matriz_A, i): {getLinha(matriz_A, i)}')
        print(f'getColuna(matriz_B, j): {getColuna(matriz_B, j)}')
        # multiplica cada linha de matriz_A por cada coluna de matriz_B
        listMult = [x * y for x, y in zip(getLinha(matriz_A, i), getColuna(matriz_B, j))]
        # print(listMult)

        # e em seguida adiciona a matrizResultante a soma das multiplicações
        matrizResultante[i].append(sum(listMult))

# Print results.
print()
print("\033[1;31mMatriz original A\033[m")
for linha in matriz_A:
    print(linha)
print()
print("\033[1;31mMatriz original B\033[m")
for linha in matriz_B:
    print(linha)
print()
print(f'\033[1;33mO Resulto da multiplicação das matrizes:\033[m')
for linha in matrizResultante:
    print(f'\033[1;31m{linha}\033[m')