шцкс

#!/usr/bin/python
# vim: fileencoding=utf-8 ts=4 expandtab

from lab1 import Substitution

abc1 = u'абвгдеёжзийклмнопрстуфхцчшщъыьэюя'
abc2 = u'абвгдеежзииклмнопрстуфхцчшщьыьэюя'
abc  = u'абвгдежзиклмнопрстуфхцчшщыьэюя'

assert(len(abc1) == len(abc2))
assert(set(abc2) == set(abc))
assert(len(abc) == len(set(abc)))

f = open('plain.txt', 'r')
line = f.read().decode('utf-8').lower()
f.close()
plain_text = u"".join([abc2[abc1.index(ch)] for ch in line if ch in abc1])

key_phrase = u"ключеваяфраза"
num_key = 3

s = Substitution(abc)

s.make_keyword(key_phrase, num_key)
cipher_text =  s.encrypt(plain_text)
print plain_text
print "====="
print cipher_text