[projet ISN] chiffrement de vigénère

# -*- coding:Latin-1 -*-

ALPHABET = '0123456789 ABCDEFGHIJKLMNOPQRSTUVWXYZ!'
CLE_SECRETE = 'LOLA'

def genere_cle_vigenere(msg_clair: str, cle_vig: str) -> str: 
	''' 
	Renvoie la clé vigenère ajustée à la longueur du message clair
	'''

	if (len(cle_vig) >= len(msg_clair)):
		cle_ajustee = cle_vig[:len(msg_clair)]
	else:
		cle_ajustee = cle_vig
		i = 0
		while(len(cle_ajustee) < len(msg_clair)):
			cle_ajustee = cle_ajustee + cle_vig[i%len(cle_vig)]
			i += 1

	return cle_ajustee

def assigne_un_entier_aux_lettres(message, cle_ajustee):
	''' Créé 2 dictionnaires:
	clé = une lettre du message, valeur = rang de cette lettre dans ALPHABET
	Idem avec la clé ajustée
	'''

	dico_msg = dico_cle = {}

	for k in range (len(message)):
			if (message[k].upper() in ALPHABET):
				dico_msg[message[k]] = ALPHABET.index(message[k].upper())

	for k in range (len(cle_ajustee)):
		if (cle_ajustee[k].upper() in ALPHABET):
			dico_cle[cle_ajustee[k]] = ALPHABET.index(cle_ajustee[k].upper())

	return (dico_msg, dico_cle)

def codage_decodage_vigenere(message, cle_ajustee, dico_msg, dico_cle, coeff):
	''' Codage ou decodage du message avec les dictionnaires '''

	message_resultant = ''

	for k in range (len(message)):
		message_resultant += ALPHABET[(dico_msg[message[k]] + dico_cle[cle_ajustee[k]]*coeff)%len(ALPHABET)].lower()

	return message_resultant


#Saisie du message clair
print('\n')
print('''Codage d'un message'''.center(50,'*'))
message_clair = input('\nEntrez le message à coder : ')

#Longueur de la clé = longueur du message clair
cle_ajustee = genere_cle_vigenere(message_clair, CLE_SECRETE)

#On assigne un entier à chaque lettre
dico_msg, dico_cle = assigne_un_entier_aux_lettres(message_clair, cle_ajustee)

#Codage
message_code = codage_decodage_vigenere(message_clair, cle_ajustee, dico_msg, dico_cle, 1)
print('>>>{}<<<'.format(message_code))


#Saisie du message codé
print('\n')
print('''Decodage d'un message'''.center(50,'*'))
message_code = input('\nEntrez le message à decoder : ')

#Longueur de la clé = longueur du message clair
cle_ajustee = genere_cle_vigenere(message_code, CLE_SECRETE)

#On assigne un entier à chaque lettre
dico_msg, dico_cle = assigne_un_entier_aux_lettres(message_code, cle_ajustee)

#Decodage
message_clair = codage_decodage_vigenere(message_code, cle_ajustee, dico_msg, dico_cle, -1)
print('>>>{}<<<'.format(message_clair))

input('\n\nAppuyer sur enter pour quitter ...')