Шифр Виженера, как можно сделать лучше код?

Question

[yoyyy]

В программировании я новичок, но захотелось сделать собственный код для шифра Виженера. Видел другие варианты кода, но мало что в них понял, поэтому сделал по-своему. Скажите, что можно исправить здесь, чтобы сократить или оптимизировать сам код?

import textwrap
#c-шиф символ n-алфавит k-ключ m-исх номер
alf_ru = 'АБВГДЕЁЖЗИЙКЛМНОПРСТУФХЦЧШЩЪЫЬЭЮЯ'
alf_rum = 'абвгдеёжзийклмнопрстуфхцчшщъыьэюя'
a=input(('Ввееди фразу: '))
key=input(('Слово: '))
act=str(input('1 - шиф, 2 - дешиф: '))

def shifr(sh):
    n=33
    result=''
    for i in range (len(a)):
        if a[i] in alf_rum:
            ish=a[i]
            kl=key[i]
            m=alf_rum.find(ish)
            k=alf_rum.find(kl)
            c=(m+k)%n
            result+=alf_rum[c]
        elif a[i] in alf_ru:
            ish = a[i]
            kl = key[i]
            m = alf_ru.find(ish)
            k = alf_rum.find(kl)
            c = (m + k) % n
            result+=alf_ru[c]
        else:
            result+=a[i]
    return result
def deshifr(sh):
    n=33
    result=''
    for i in range (len(a)):
        if a[i] in alf_rum:
            ish=a[i]
            kl=key[i]
            c=alf_rum.find(ish)
            k=alf_rum.find(kl)
            m=(c-k)%n
            result+=alf_rum[m]
        elif a[i] in alf_ru:
            ish = a[i]
            kl = key[i]
            c = alf_ru.find(ish)
            k = alf_rum.find(kl)
            m=(c-k)%n
            result+=alf_ru[m]
        else:
            result+=a[i]
    return result

empt=[]
for i in range(len(a)):
    if a[i]==' ':
        empt.append(i)
a=a.replace(' ','')
#print('1)', a, empt )
simb=[]
znaki=[]

for i in range(len(a)):
    if a[i] not in alf_rum and a[i] not in alf_ru:
        simb.append(i)
        znaki.append(a[i])
    else:
        continue
#print('2)', a, simb, znaki)

while len(a)!=len(key):
    for i in range(0,len(key)):
        if len(a)!=len(key):
            key+=key[i]

for i in range(len(simb)):
    pos=simb[i]
    key=key[:pos]+znaki[i]+key[pos:]
key=key[:len(a)]
#print('3)', key)



move={
    '1': shifr,
    '2': deshifr,
}
res=move.get(act)(act)
for i in range(len(empt)):
    pro=empt[i]
    res=res[:pro]+' '+res[pro:]

print(textwrap.fill(res, 157))

Comments

[Ланской Кирилл]

Спроси у ChatGPT4o

[Алан Гибизов]

Я не большой мастер в этом деле, но вот что мне думается стоит сделать:

• Во-первых, стоит переименовать все эти n, c, k, m, kl, a, sh в нормальные имена переменных. Иначе смотреть на код больно. Сокращение кода это не столько уменьшение количества букв, сколько сокращение повторений кода. А имена наоборот должны быть максимально длинные и читаемые и осмысленные.
  
• Во-вторых, вообще соблюсти PEP8 для читабельности (отступы между функциями и т.п.)
  
• В-третьих, посмотреть, где код повторяется, и подумать, как сделать, чтобы не повторялся (варианты - вынести в отдельную функцию, переделать if/elif/else в другой порядок, и т.п.)
  
• В-четвертых, код выглядит как лапша, его хорошо бы организовать. Можно в классы и методы, можно по функциям, более мелким и переиспользуемым... Тогда можно более удобно в дальнейшем его причесывать, а также тестировать.
  

[Алан Гибизов]

Да, еще - переменная sh как используется в функциях? Судя по коду, никак. Но является обязательным параметром функций. Зачем?

[yoyyy]

Алан Гибизов, спасибо! Постараюсь сделать лучше)
Переменные n, c, k, m я просто взял из математических формул, которые были на Википедии. Фактически, я эти формулы просто перепечатал. Переменную sh я не нашел, наверное, вы имели в виду ish, она скорее была для моего удобства, ей присваивалась исходная буква слова. Большое спасибо за ответ, не знал про PEP8, буду осваивать)

[Алан Гибизов]

yoyyy, переменная sh это обязательный параметр ваших функций. Он создается при вызове функции, наполняется значением переданного параметра и виден внутри функции. Но вы его не используете.
И вообще ощущение, что вы где-то этот код взяли - в нем встречаются довольно специфические действия (в части словаря с функциями, например).

Answer 1

[Руслан .]

Если передать на вход действие не 1 и не 2, то все упадет с ошибкой, т.к. функция в словаре move не будет найдена.

Использовать глобальные переменные это антипаттерн. Я про переменные key и a. Их можно было бы передать как параметры.

Для определения кода буквы можно использовать функцию ord и тогда не нужно будет пользоваться списком символов для вычисления индекса символа.