52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
# Шифр Цезаря
|
||
SYMBOLS = 'АБВГДЕЕЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯабвгдеежзийклмнопрстуфхцчшщъыьэюя'
|
||
MAX_KEY_SIZE = len(SYMBOLS)
|
||
|
||
def getMode():
|
||
while True:
|
||
print('Вы хотите зашифровать или расшифровать текст?')
|
||
mode = input().lower()
|
||
if mode in ['зашифровать', 'з', 'расшифровать', 'р']:
|
||
return mode
|
||
else:
|
||
print('Введите "зашифровать" или "з" для зашифровки или "расшифровать" или "р" для расшифровки.')
|
||
|
||
def getMessage():
|
||
print('Введите текст:')
|
||
return input()
|
||
|
||
def getKey():
|
||
key = 0
|
||
while True:
|
||
print('Введите ключ шифрования (1-%s)' % (MAX_KEY_SIZE))
|
||
key = int(input())
|
||
if (key >= 1 and key <= MAX_KEY_SIZE):
|
||
return key
|
||
|
||
def getTranslatedMessage(mode, message, key):
|
||
if mode[0] == 'р':
|
||
key = -key
|
||
translated = ''
|
||
|
||
for symbol in message:
|
||
symbolIndex = SYMBOLS.find(symbol)
|
||
if symbolIndex == -1: # Символ не найден в SYMBOLS.
|
||
# Просто добавить этот символ без изменений.
|
||
translated += symbol
|
||
else:
|
||
# Зашифровать или расшифровать
|
||
symbolIndex += key
|
||
|
||
if symbolIndex >= len(SYMBOLS):
|
||
symbolIndex -= len(SYMBOLS)
|
||
elif symbolIndex < 0:
|
||
symbolIndex += len(SYMBOLS)
|
||
|
||
translated += SYMBOLS[symbolIndex]
|
||
return translated
|
||
|
||
mode = getMode()
|
||
message = getMessage()
|
||
key = getKey()
|
||
print('Преобразованный текст:')
|
||
print(getTranslatedMessage(mode, message, key)) |