FILE = "C:/Users/Asus/Desktop/GEDEFU.txt"
charset = ["A", "D", "G", "F","V", "X"]
french = ["E", " ", "A", "I", "S", "N", "R", "T", "O", "L", "U", "D", "C", "M", "P", "G", "B", "V", "H", "F", "Q", "Y", "X", "K", "W","Z", "1", "9", "6", "2", "8", "3", "4", "5", "7", "0"]
clef = "PAHMVIN"

def getblock(content): #découpe le texte en block
    solo, block, i = [], [], 0
    for elt in content:
        solo.append(elt)
    if len(solo)%2 != 0: #Padding si le texte n'est pas un chiffre pair.
        solo += "X"
    while i != len(solo):
        block.append(solo[i]+solo[i+1])
        i += 2
    return block

def refe(array): #Créer une liste de référence.
    ref, i, j = [], 0, 0
    for i in range(len(array)):
        for j in range(len(array)):
            ref.append(array[i]+array[j])
    return ref


def occurence(block, ref): #Calcule la fréquence d'apparition d'une pair de lettres.
    i, j, count, res = 0, 0, 0, []
    for i in range(len(ref)):
        for j in range(len(block)):
            if block[j] == ref[i]:
                count += 1
        count = round((count*100)/len(block), 2)
        res.append(count)
        count = 0
    return res

def liste(ref, frequence, french):  #Créer une liste avec correspondance.
    liste , i = [], 0
    for i in range(len(ref)):
        elt = ref[i]+":"+str(frequence[i])
        liste.append(elt)
    sub_liste, frequence, i = [], sorted(frequence, reverse=True), 0
    for i in range(len(french)):
        sub_liste.append(liste[i].replace(liste[i][3:], french[i]))
    return sub_liste

def substitution(block, liste, ref): #Remplace les lettres dans le texte par les lettres de la frequence de la langue (Etape 1 du chiffrement)
    for i in range(len(ref)):
        for j in range(len(block)):
            if ref[i] == block[j]:
                block[j] = block[j].replace(block[j], liste[i][3:])
    return block


def transposition(key, decomp_solve): #Transposition
    inter_solve = []
    solve = [[0]*len(key) for i in range(186)] #Créer une matrice 7 colonnes pour 186 lignes
    matrice = [[0]*186 for i in range(len(key))] #Créer une matrice de 186 lignes pour 7 colonnes
    tab = list(zip(*[iter(decomp_solve)]*len(key))) #Créer une liste de 7 listes composées de 186 éléménts
    i,j, dic = 0, 0, {}
    for i in range(len(key)):
        for j in range(186):
            matrice[i][j] = tab[j][i] #Remplissage via une intération
    decomp_key = [elt for elt in key] #Décomposition de la clé en une liste de lettre
    for i in range(len(matrice)):
        dic[decomp_key[i]] = matrice[i] #Création d'un dictionnaire avec pour clés les lettres de la clé et comme valeur les colonnes créent
    decomp_key = sorted(decomp_key) #Réorganisation dans l'odre alphabétique de la clé
    for elt in decomp_key:
        inter_solve.append(dic[elt]) #Remplissage de la liste du contenu des 7 colonnes dans l'ordre alphabétique
    for i in range(len(inter_solve)):
        for j in range(len(solve)):
            solve[j][i] = inter_solve[i][j] #Remplissage via une intération
    return solve



def convertString(array,n, p): #Convertit un array en string
    solve_str = ""
    for i in range((n)):
        for j in range((p)):
            solve_str += array[j][i]
    return solve_str


def main():
    with open(FILE, 'r') as content:
        raw = content.read()
        content.close()

    block = getblock(raw) #déclaration du block
    reference = refe(charset) #déclaration de notre charset pour le texte
    frequence = occurence(block, reference) #Analyse de la fréquence d'apparitions
    sub_liste = liste(reference, frequence, french) #Affiliation lettre réelle ==> pair de lettres du texte
    decomp_solve = substitution(block, sub_liste, reference) #Etape 1 réaliser (substitution)
    solve = transposition(clef, decomp_solve) #retourne la solution sous la forme d'un array
    solve_str = convertString(solve, 7, 186) #Convertir la solution en string
    print(sub_liste)
    print(solve_str)