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

####################################
#File: codevigenere.py             #
#Author: D. Lugiez                 #
#Date: 06/01/09                    #
####################################

#codage par le code de Vigenère
#limité aux lettres minuscules
#Pas de verification que les arguments sont corrects (chaine non vide de lettres
#minuscules pour le secret, chaine qcq, seules les minuscules sont codées.
#fonctionne aussi avec les majuscules si on modifie la taille de l'alphabet
#mise à 26 ici et le test dans la fonction code 'a'<=c and c<='z'


def codelettre(c,numalphabet,taillealphabet):
    "calcule le caractère codant le caractère c dans l'alphabet numéro numalphabet\
     de taille taillealphabet                                         \
     l'alphabet sera a,...,z de taille 26"
    code=ord(c)-ord('a')+numalphabet
    return chr(code%taillealphabet+ord('a'))


#on evite la construction for .. in pour le moment
def code(ch, secret,taillealphabet):
    "code la chaine ch avec un mot secret secret (une chaine non vide)\
    tout les lettres sont dans l'alphabet de taille taillealphabet"
    taillesecret=len(secret)
    taillemsg=len(ch)
    chcode=""
    i=0
    while i<len(ch):
        c=ch[i]
        if ('a'<=c and c<='z'):
            lettredusecret=secret[i%taillesecret]
            numalphabet=(ord(lettredusecret)-ord('a'))%taillealphabet
            chcode=chcode+codelettre(ch[i],numalphabet,taillealphabet)
        else:
            chcode=chcode+c
        i=i+1
    return chcode

#script de tests

#test pour verifier

#test1: le secret donne toujours l'alphabet usuel
secret1="a"
msg1="un test"
res1=code(msg1,secret1,26)
if res1!=msg1:
    print "test1:failed"
else:
    print "test1:sucess"

#test2: le secret donne toujours l'alphabet usuel
secret2="aa"
msg2="un test"
res2=code(msg2,secret2,26)
if res2!=msg2:
    print "test2:failed"
else:
    print "test2:sucess"


#test3: le secret donne toujours le decalage de César à une lettre
secret3="b"
msg3="cesar"
res3=code(msg3,secret3,26)
if res3!="dftbs":
    print "test3:failed"
else:
    print "test3:sucess"

#test 4: un message vide est toujours code par la chaine vide
secret4="cs"
msg4=""
res4=code(msg4,secret4,26)
if res2!=msg2:
    print "test4:failed"
else:
    print "test4:sucess"

#test 5: un exemple normal 
secret5="python"
msg5="messagesecret"
res5=code(msg5,secret5,26)
if res5!="bclzottqxjfri":
    print "test5:failed"
else:
    print "test5:sucess"


#test 6:  exemple du livre de Singh
secret6="rouge"
msg6="appelernordtroupesville"
res6=code(msg6,secret6,26)
if res6!="rdjkpvfhuvuhluygsmbmczy":
    print "test6:failed"
else:
    print "test6:sucess"
 
#script d'utilisation manuelle
answer='o'
while answer=='o':
    print "entrer secret (une chaine de caracteres non vide entre 'a' et 'z')"
    secret=input()
    print "entrer message (une chaine de caractères, seules les minuscules seront codées)"
    msg=input()
    msgcode=code(msg,secret,26)
    print "message codé",msgcode
    print("continuer o/n")
    anwer=input()