bugfix

[krypto.git] / vigenere.py

diff --git a/vigenere.py b/vigenere.py
index b276331..3197d07 100644 (file)
--- a/vigenere.py
+++ b/vigenere.py
@@ -10,9 +10,9 @@ def soucet(x, y):
 
 def vsechno(zprava, delka_h):
     """Najde heslo dane delky, pri jehoz aplikaci budou frekvence znaku nejlepe
 
 def vsechno(zprava, delka_h):
     """Najde heslo dane delky, pri jehoz aplikaci budou frekvence znaku nejlepe

-    odpovidat cestine."""
+    odpovidat cestine a patricne posunuty text."""

     rozklady = rozklad(ocesat(zprava, False), delka_h)
     posuny = [nej_posun(text) for text in rozklady]
     posunute = [''.join([posun(c, posuny[i]) for c in rozklady[i]]) for i in range(len(posuny))]
     rozklady = rozklad(ocesat(zprava, False), delka_h)
     posuny = [nej_posun(text) for text in rozklady]
     posunute = [''.join([posun(c, posuny[i]) for c in rozklady[i]]) for i in range(len(posuny))]

-    vysledek = [x for x  in reduce(soucet, (map(None, *posunute)[i] for i in range(len(rozklady[0])))) if x != None]
+    vysledek = [x for x  in reduce(soucet, (map(None, *posunute)[i] for i in range(len(rozklady[0])))) if x]

     return cisla_na_text(posuny), ''.join(vysledek)
     return cisla_na_text(posuny), ''.join(vysledek)