ИБ 5 лаба

message = input("Enter message:\n")
key = input("Enter key:\n")

#message = 'люблю_криптографию' #18 символа
#key = '461523' #6 символов

des_message = ""
a = []
res_message = ""
a1 = []

for i in range(int(len(message)/len(key))):
  a.append(message[i * (len(key)):len(key) * (i + 1)])

for i in range(1, (len(key) + 1)):
  index = key.index(str(i))
  for word in a:
    res_message += word[index]

print("Coded message:", res_message)

for i in range(int(len(key))):
  a1.append(res_message[i * int(len(message)/len(key)):int(len(message)/len(key)) * (i + 1)])

for j in range(int(len(message)/len(key))):
  for i in range(len(key)):
    index = int(key[i])
    number_str = a1[index - 1]
    des_message += number_str[j]
print("Decoded message:", des_message)