__author__ = 'ipetrash'

"""Скрипт реализует простейшую реализацию алгоритма Код Цезаря."""

def caesar_code(text, shift):

"""Функция принимает текстовую строку и возвращает, новую строку

символы которой сдвинуты по алфавиту."""

alp = [chr(c) for c in range(ord('a'), ord('z') + 1)]

new_t = ''

for c in text.lower():

if c in alp:

i = (alp.index(c) + shift) % len(alp)

new_t += alp[i]

else:

new_t += c

return new_t

from string import ascii_uppercase

def shift(text, num):

shift_text = ''

for c in text:

if c not in ascii_uppercase:

shift_text += c

continue

i = (ascii_uppercase.index(c) + num) % len(ascii_uppercase)

shift_text += ascii_uppercase[i]

return shift_text

if __name__ == '__main__':

print(caesar_code("Hello World!", shift=2))

print(caesar_code("Hello World!", shift=-4))

print(caesar_code("Hello World!", shift=26))

print(caesar_code("Hello World!", shift=50))

print(caesar_code("Hello World!", shift=78))

print()

TEXT = 'VWDQ LV QRW ZKDW KH VHHPV'

for i in range(len(ascii_uppercase)):

print(i, shift(TEXT, i))