robot.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import smtplib
   5 import email
   6 from email.mime.text import MIMEText
   7 from time import asctime
   8 from ocesavac import ocesat
   9 import jakobsen
  10 from jakobsen import reference
  11 import koincidence
  12 import pickle
  13 import os
  14
  15 ABECEDA = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
  16
  17 class settings:
  18     pass
  19
  20 settings = pickle.load(open('./robot_settings'))
  21 os.chdir(settings.path)
  22
  23 # Cteni mailu
  24 mail = email.message_from_file(sys.stdin)
  25 predmet, _, opsny = mail['Subject'].strip().upper().partition(' ')
  26 if predmet != settings.subject:
  27     sys.exit()
  28 if (opsny == 'L'):
  29     mail = pickle.load(open('./msg_last.tmp'))
  30 odesilatel = mail['From']
  31 telo = mail.get_payload().strip()
  32 if len(telo) < 2:
  33     sys.exit()
  34 zprava = telo.upper()
  35 analyza = []
  36
  37 # Zakladni udaje
  38 analyza.append("Delka zpravy (bez mezer a zvlastnich znaku):")
  39 analyza.append(str(len(ocesat(zprava, False))))
  40
  41 # Index koincidence
  42 analyza.append("\nIndexy koincidence:")
  43 analyza.append(koincidence.tabulka_indexu(zprava))
  44
  45 # Jakobsen
  46 analyza.append("\nJakobsen:")
  47 analyza.append(jakobsen.desifruj(zprava, './ref'))
  48
  49 # Posuny v abecede
  50 analyza.append("\nPosuny v abecede:")
  51
  52 def posun(char):
  53     if 65 <= ord(char) < 90:
  54         return chr(ord(char) + 1)
  55     if ord(char) == 90:
  56         return 'A'
  57     else:
  58         return char
  59
  60 posunuty = zprava
  61 for i in range(26):
  62     posunuty = map(posun, posunuty)
  63     analyza.append(''.join(posunuty))
  64
  65 # Frekvence znaku
  66 analyza.append("\nFrekvencni analyza:")
  67 freq = dict()
  68 for char in ABECEDA:
  69     freq[char] = 0
  70 for char in zprava:
  71     if char in freq:
  72         freq[char] += 1
  73 celkem = sum(freq.values())
  74 for char in sorted(freq, key=freq.get, reverse=True):
  75     analyza.append('{0} {1:>3} {2:>6.2%}'.format(char, freq[char],
  76                    float(freq[char])/celkem))
  77
  78 # Bigramy
  79 analyza.append("\nBigramy podle cetnosti:")
  80 bigramy = dict()
  81 for i in ABECEDA:
  82     for j in ABECEDA:
  83         bigramy[i+j] = 0
  84 for i in range(0, len(zprava)-1):
  85     if zprava[i:i+2] in bigramy:
  86         bigramy[zprava[i:i+2]] += 1
  87 celkem = sum(bigramy.values())
  88 for char in sorted(bigramy, key=bigramy.get, reverse=True):
  89     if bigramy[char] == 0:
  90         break
  91     analyza.append('{0} {1:>3} {2:>6.2%}'.format(char, bigramy[char],
  92                    float(bigramy[char])/celkem))
  93
  94 # Kompozice odpovedi
  95 msg = MIMEText("Puvodni zprava:\n" +
  96                telo + "\n\n" + "\n".join(analyza) +
  97                "\n\nS pozdravem\nVas Robot\n")
  98 msg['Subject'] = "Analyza Vasi zpravy"
  99 msg['From'] = settings.name + " <" + settings.mail  + ">"
 100 msg['To'] = odesilatel
 101
 102 # Odeslani odpovedi
 103 s = smtplib.SMTP('localhost')
 104 s.sendmail(settings.mail, odesilatel, msg.as_string())
 105 s.quit()
 106
 107 # Zalogovani posledni zpravy
 108 f = open('./robot_last.tmp', 'w')
 109 f.write(msg.as_string())
 110 f.close()
 111
 112 f = open('./msg_last.tmp', 'w')
 113 pickle.dump(mail, f)
 114 f.close()
 115
 116 f = open('./robot.log', 'a')
 117 f.write(asctime() + " " +  odesilatel + "\n")
 118 f.close()