#-*- coding:utf-8 -*-

import string
import math
import time
from time import gmtime, strftime
import decimal




def encoding_calculus(liste, n, s): #une liste, un nombre, un autre nombre
	chaine_code = []
	for i in range(0, n):  
		k = (int(decimal.Decimal(i**3+i**2-13*i-11+liste[i])))%s
		chaine_code.append(k)
	return chaine_code

def decoding_calculus(liste, n, s): #une liste, un nombre, un nb
	chaine_decode = []
	for i in range(0, n):
		k = (int(decimal.Decimal(-i**3-i**2+13*i+11+liste[i])))%s #ou n'importe quelle fonction opérant sur entiers (polynômes...)
		chaine_decode.append(k)
	return chaine_decode

def encrypt(mot):
	
	
	alphabet = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'â', 'à', 'b', 'c', 'ç', 'd', 'e', 'é', 'è', 'ê', 'ë', 'f', 'g', 'h', 'i', 'ï', 'î', 'j', 'k', 'l', 'm', 'n', 'ñ', 'o', 'ô', 'ö', 'p', 'q', 'r', 's', 't', 'u', 'û', 'ù', 'ü', 'v', 'w', 'x', 'y', 'z', 'A', 'À', 'Â', 'Á', 'B', 'C', 'D', 'E', 'È', 'É', 'Ê', 'Ë', 'Ì', 'Í', 'Î', 'Ï', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '!', '"', '#', '$', '%', '&', "'", '(', ')', '*', '+', ',', '-', '.', '/', ':', ';', '<', '=', '>', '?', '@', '[', ']', '^', '_', '`', '{', '|', '}', '~', '-', '–', ' ', '\n', '’', '\t', '²']
	
	print("L'alphabet utilisé est :", alphabet)
	
	
	mot_nb = []
	
	for caractere in mot:
		position = alphabet.index(caractere) #position = alphabet.find(caractere)
		mot_nb.append(position) # On ajoute les rangs de chaque caractère du mot en caractères dans la liste du mot en numérique
		
	print("Votre mot : '", mot, "' comporte", len(mot), "caractères et correspond à :", mot_nb)
	mot_code_liste = []
	mot_code_nb = encoding_calculus(mot_nb, len(mot), len(alphabet)) # On encode le mot en nombres grâce à la fonction
	
	for i in mot_code_nb: # On met les valeurs du mot codé en nombres en caractères corrrespondants que l'on ajoute dansune liste
		letter_code = alphabet[int(i)] # On mémorise le rang du nombre codé
		mot_code_liste.append(letter_code) # On associe un caractère que l'on ajoute à une liste
	
	
	mot_code = ""
	for i in range(0, len(mot)): # Pour la liste des caractères codés
		mot_code += mot_code_liste[i] # On ajoute chaque élément de la liste dans une valeur chaîne de caractères
	print("Le mot codé : '", mot_code, "' correspond à :", mot_code_nb)
	return mot_code

def decrypt(mot_code):
		
	alphabet = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'â', 'à', 'b', 'c', 'ç', 'd', 'e', 'é', 'è', 'ê', 'ë', 'f', 'g', 'h', 'i', 'ï', 'î', 'j', 'k', 'l', 'm', 'n', 'ñ', 'o', 'ô', 'ö', 'p', 'q', 'r', 's', 't', 'u', 'û', 'ù', 'ü', 'v', 'w', 'x', 'y', 'z', 'A', 'À', 'Â', 'Á', 'B', 'C', 'D', 'E', 'È', 'É', 'Ê', 'Ë', 'Ì', 'Í', 'Î', 'Ï', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '!', '"', '#', '$', '%', '&', "'", '(', ')', '*', '+', ',', '-', '.', '/', ':', ';', '<', '=', '>', '?', '@', '[', ']', '^', '_', '`', '{', '|', '}', '~', '-', '–', ' ', '\n', '’', '\t', '²']
	
	
	mot_code_nb = []
	
	for caractere in mot_code:
		position = alphabet.index(caractere) #position = alphabet.find(caractere)
		mot_code_nb.append(position) # On ajoute les rangs de chaque caractère du mot codé en caractères dans la liste du même mot en numérique
		
	print("Votre mot : '", mot_code, "' comporte", len(mot_code), "caractères et correspond à :", mot_code_nb)
	
	# DEUXIEME PARTIE
	
	mot_nb = decoding_calculus(mot_code_nb, len(mot_code), len(alphabet)) # On décode le mot en nombres grâce à la fonction
	
	mot_liste = []
	for i in mot_nb: # On met les valeurs du mot décodé en nombres en caractères corrrespondants que l'on ajoute dansune liste
		letter = alphabet[int(i)] # On mémorise le rang du nombre codé
		mot_liste.append(letter) # On associe un caractère que l'on ajoute à une liste
	
	
	mot_decode = ""
	for i in range(0, len(mot_code)): # Pour la liste des caractères codés
		mot_decode += mot_liste[i] # On ajoute chaque élément de la liste dans une valeur chaîne de caractères
	print("Le mot décodé : '", mot_decode, "' correspond à :", mot_nb)
	return mot_decode

t1 = time.time()


# ENCODAGE
plaintext = open("plaintext.txt", "r")
mot = plaintext.read()
codage = encrypt(mot)

ciphertext = open("ciphertext.txt", "w") # 'a' pour "append" (ajouter) ou 'w' pour write (et écraser tout le contenu)
ciphertext.write(codage)
ciphertext.close()
plaintext.close()



# DECODAGE
ciphertext = open("ciphertext.txt", "r")
mot = ciphertext.read()
decodage = decrypt(mot)

decrypted_text = open("decrypted_text.txt", "w")
decrypted_text.write(decodage)
decrypted_text.close()
ciphertext.close()

t2 = time.time() - t1
print("Temps d'éxécution :", t2)
print(strftime("%a, %d %b %Y %H:%M:%S +0000", gmtime()))