kosmeticke upravy

diff --git a/jakobsen.py b/jakobsen.py
index 40a1d1c..7e858f0 100644 (file)
--- a/jakobsen.py
+++ b/jakobsen.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+"""Modul pro praci s Jakobsenovym algoritmem."""

 
 from itertools import combinations
 import pickle
 
 from itertools import combinations
 import pickle


@@ -7,6 +7,7 @@ from ocesavac import ocesat
 MABECEDA = ' ABCDEFGHIJKLMNOPQRSTUVWXYZ'
 
 def nova_tabulka(zprava, abc=MABECEDA):
 MABECEDA = ' ABCDEFGHIJKLMNOPQRSTUVWXYZ'
 
 def nova_tabulka(zprava, abc=MABECEDA):

+    """Vraci tabulku cetnosti bigramu ve zprave nad danou abecedou."""

     tabulka = dict()
     for i in abc:
         tabulka[i] = dict()
     tabulka = dict()
     for i in abc:
         tabulka[i] = dict()


@@ -21,6 +22,8 @@ def nova_tabulka(zprava, abc=MABECEDA):
     return tabulka
 
 def vzdalenost(tab1, tab2, abc=MABECEDA):
     return tabulka
 
 def vzdalenost(tab1, tab2, abc=MABECEDA):

+    """Vraci soucet strednich kvadratickych odchylek pro dve tabulky cetnosti
+    bigramu nad danou abecedou."""

     rozdil = 0
     for i in abc:
         for j in abc:
     rozdil = 0
     for i in abc:
         for j in abc:


@@ -28,15 +31,18 @@ def vzdalenost(tab1, tab2, abc=MABECEDA):
     return rozdil
 
 def substituce(zprava, slovnik, abc=MABECEDA):
     return rozdil
 
 def substituce(zprava, slovnik, abc=MABECEDA):

+    """Vrati zpravu, ve ktere jsou znaky dane abecedy nahrazenypodle daneho
+    slovniku."""

     pole = []
     pole = []

-    for c in zprava:
-        if c in abc:
-            pole.append(slovnik[c])
+    for char in zprava:
+        if char in abc:
+            pole.append(slovnik[char])

         else:
         else:

-            pole.append(c)
+            pole.append(char)

     return pole
 
 def poradi_dle_frekvence(zprava, abc=MABECEDA):
     return pole
 
 def poradi_dle_frekvence(zprava, abc=MABECEDA):

+    """Vrati znaky dane abecedy v poradi podle frekvence v dane zprave."""

     freq = dict()
     for char in abc:
         freq[char] = 0
     freq = dict()
     for char in abc:
         freq[char] = 0


@@ -49,8 +55,13 @@ class reference:
     pass
 
 def jakobsen(zprava, ref):
     pass
 
 def jakobsen(zprava, ref):

+    """Pro danou sifrovanou zpravu vrati substitucni slovnik odvozeny s pomoci
+    Jakobsenova algoritmu s danou referencni tabulkou."""

     slovnik = dict(zip(poradi_dle_frekvence(zprava, ref.abeceda), ref.poradi))
     slovnik = dict(zip(poradi_dle_frekvence(zprava, ref.abeceda), ref.poradi))

-    tabulka = nova_tabulka(substituce(zprava, slovnik, ref.abeceda), ref.abeceda)
+    tabulka = nova_tabulka(
+                  substituce(zprava, slovnik, ref.abeceda),
+                  ref.abeceda
+                  )

     vzdal = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
 
     vzdal_old = vzdal + 1
     vzdal = vzdalenost(tabulka, ref.tabulka, ref.abeceda)
 
     vzdal_old = vzdal + 1


@@ -58,22 +69,27 @@ def jakobsen(zprava, ref):
         vzdal_old = vzdal
         for (x, y) in combinations(ref.abeceda, 2):
             slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
         vzdal_old = vzdal
         for (x, y) in combinations(ref.abeceda, 2):
             slovnik[x], slovnik[y] = slovnik[y], slovnik[x]

-            v1 = vzdalenost(nova_tabulka(substituce(zprava, slovnik, ref.abeceda), ref.abeceda),
-                            ref.tabulka, ref.abeceda)
-            if  v1 < vzdal:
-                vzdal = v1
+            nova_vzdalenost = vzdalenost(
+                     nova_tabulka(
+                         substituce(zprava, slovnik, ref.abeceda),
+                         ref.abeceda),
+                     ref.tabulka,
+                     ref.abeceda)
+            if  nova_vzdalenost < vzdal:
+                vzdal = nova_vzdalenost

             else:
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
     return slovnik
     
 
 def desifruj(zprava, mezery=True):
             else:
                 slovnik[x], slovnik[y] = slovnik[y], slovnik[x]
     return slovnik
     
 
 def desifruj(zprava, mezery=True):

+    """Vrati vysledek pokusu dekodovat zpravu Jakobsenovym algoritmem."""

     if mezery:
     if mezery:

-        f = open('ref')
+        soubor = open('ref')

     else:
     else:

-        f = open('bref')
-    ref = pickle.load(f)
-    f.close()
+        soubor = open('bref')
+    ref = pickle.load(soubor)
+    soubor.close()

     slovnik = jakobsen(ocesat(zprava, mezery), ref)
     return '\n'.join([''.join(substituce(zprava, slovnik, ref.abeceda)),
                       ' '.join(ref.abeceda),
     slovnik = jakobsen(ocesat(zprava, mezery), ref)
     return '\n'.join([''.join(substituce(zprava, slovnik, ref.abeceda)),
                       ' '.join(ref.abeceda),

diff --git a/robot.py b/robot.py
index 21d297a..7992a53 100755 (executable)
--- a/robot.py
+++ b/robot.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 
 #!/usr/bin/env python
 

+"""Mailove rozhrani pro sadu kryptoanalytickych nastroju."""
+

 import sys
 import smtplib
 import email
 import sys
 import smtplib
 import email


@@ -22,16 +24,16 @@ class settings:
 settings = pickle.load(open('./robot_settings'))
 os.chdir(settings.path)
 
 settings = pickle.load(open('./robot_settings'))
 os.chdir(settings.path)
 

-def posli_mail(komu, co, text, loguj=None):
+def posli_mail(komu, predmet, text, loguj=None):

     msg = MIMEText(text)
     msg = MIMEText(text)

-    msg['Subject'] = co
+    msg['Subject'] = predmet

     msg['From'] = settings.name + " <" + settings.mail  + ">"
     msg['To'] = komu
     smtplib.SMTP('localhost').sendmail(settings.mail, komu, msg.as_string())
     if loguj:
     msg['From'] = settings.name + " <" + settings.mail  + ">"
     msg['To'] = komu
     smtplib.SMTP('localhost').sendmail(settings.mail, komu, msg.as_string())
     if loguj:

-        f = open(loguj, 'w')
-        f.write(msg.as_string())
-        f.close()
+        soubor = open(loguj, 'w')
+        soubor.write(msg.as_string())
+        soubor.close()

 
 # Cteni mailu
 mail = email.message_from_file(sys.stdin)
 
 # Cteni mailu
 mail = email.message_from_file(sys.stdin)


@@ -41,11 +43,13 @@ if predmet != settings.subject:
     sys.exit()
 if opsny == 'H':
     posli_mail(odesilatel, "Napoveda", "Napoveda:\n"
     sys.exit()
 if opsny == 'H':
     posli_mail(odesilatel, "Napoveda", "Napoveda:\n"

-               "Prikazy robotovi vkladejte do predmetu zpravy zaslane na adresu " + 
+               "Prikazy robotovi vkladejte do predmetu zpravy "
+               "zaslane na adresu " + 

                settings.mail + ".\nNa velikosti pismen nezalezi."
                "\n\nSeznam prikazu:\n"
                "\"" + settings.subject + "\"   Analyza zpravy z tela mailu.\n" 
                settings.mail + ".\nNa velikosti pismen nezalezi."
                "\n\nSeznam prikazu:\n"
                "\"" + settings.subject + "\"   Analyza zpravy z tela mailu.\n" 

-               "\"" + settings.subject + " L\" Opakovana analyza posledni zpravy.\n" +
+               "\"" + settings.subject + " L\" Opakovana analyza "
+               "posledni zpravy.\n" +

                "\"" + settings.subject + " X\" Vynuti analyzu bez mezer.\n"
                "\"" + settings.subject + " M\" Vynuti analyzu s mezerami.\n"
                "\"" + settings.subject + " B\" Vypise cetnost vsech bigramu.\n"
                "\"" + settings.subject + " X\" Vynuti analyzu bez mezer.\n"
                "\"" + settings.subject + " M\" Vynuti analyzu s mezerami.\n"
                "\"" + settings.subject + " B\" Vypise cetnost vsech bigramu.\n"


@@ -108,14 +112,17 @@ if not mod_m and not mod_x:
 # Zakladni udaje
 analyza.append("\nZakladni udaje:")
 if mod_m:
 # Zakladni udaje
 analyza.append("\nZakladni udaje:")
 if mod_m:

-    analyza.append("Delka zpravy (vcetne mezer a zvlastnich znaku): " + str(len(zprava)))
-analyza.append("Delka zpravy (bez mezer a zvlastnich znaku): " + str(len(ocesat(zprava, False))))
+    analyza.append("Delka zpravy (vcetne mezer a zvlastnich znaku): " +
+                   str(len(zprava)))
+analyza.append("Delka zpravy (bez mezer a zvlastnich znaku): " +
+               str(len(ocesat(zprava, False))))

 
 # Prumerna delka slova
 if mod_m:
     analyza.append("\nPrumerna delka slova:")
     slova = ocesat(zprava, True).split()
 
 # Prumerna delka slova
 if mod_m:
     analyza.append("\nPrumerna delka slova:")
     slova = ocesat(zprava, True).split()

-    analyza.append('{0:.3}'.format(float(sum([len(word) for word in slova]))/len(slova)))
+    analyza.append('{0:.3}'.format(
+                   float(sum([len(word) for word in slova]))/len(slova)))

 
 # Frekvence znaku 
 analyza.append("\nFrekvencni analyza:")
 
 # Frekvence znaku 
 analyza.append("\nFrekvencni analyza:")


@@ -133,7 +140,8 @@ for char in sorted(freq, key=freq.get, reverse=True):
 # Index koincidence
 if mod_m:
     analyza.append("\nIndex koincidence (s mezerami):")
 # Index koincidence
 if mod_m:
     analyza.append("\nIndex koincidence (s mezerami):")

-    analyza.append('{0:.2%}'.format(koincidence.index_koincidence(zprava, True)))
+    analyza.append('{0:.2%}'.format(
+                   koincidence.index_koincidence(zprava, True)))

 if mod_x:
     analyza.append("\nIndexy koincidence (bez mezer):")
     analyza.append(koincidence.tabulka_indexu(zprava))
 if mod_x:
     analyza.append("\nIndexy koincidence (bez mezer):")
     analyza.append(koincidence.tabulka_indexu(zprava))


@@ -150,7 +158,8 @@ if mod_x:
 (posun, posunuty_text) = posuny.nejlepsi(zprava)
 if 'P' in opsny or 'A' in opsny:
     analyza.append("\nPosuny v abecede (nejlepsi posun {0}):".format(posun))
 (posun, posunuty_text) = posuny.nejlepsi(zprava)
 if 'P' in opsny or 'A' in opsny:
     analyza.append("\nPosuny v abecede (nejlepsi posun {0}):".format(posun))

-    analyza.append("\n".join("{0:>2} {1}".format(i + 1, posuny.vsechny(zprava)[i]) for i in range(26)))
+    analyza.append("\n".join("{0:>2} {1}".format(
+                   i + 1, posuny.vsechny(zprava)[i]) for i in range(26)))

 else:
     analyza.append("\nNejlepsi posun ({0}):".format(posun))
     analyza.append(posunuty_text)
 else:
     analyza.append("\nNejlepsi posun ({0}):".format(posun))
     analyza.append(posunuty_text)


@@ -178,7 +187,10 @@ analyza.append(''.join(["\nS pozdravem\nVas Robot\n"
                         settings.subject," H\".\n"]))
 
 # Odeslani odpovedi
                         settings.subject," H\".\n"]))
 
 # Odeslani odpovedi

-posli_mail(odesilatel, "Analyza Vasi zpravy", "\n".join(analyza), './robot_last.tmp')
+posli_mail(odesilatel,
+           "Analyza Vasi zpravy",
+           "\n".join(analyza),
+           './robot_last.tmp')

 
 f = open('./msg_last.tmp', 'w')
 pickle.dump(mail, f)
 
 f = open('./msg_last.tmp', 'w')
 pickle.dump(mail, f)