'''
如密鑰短語密碼為: university -> universty
明文: abcdefghijklmnopqrstuvwxyz
密文：jklmopqwxzuniverstyabcdfgh
'''

# 構造映射 asc ---> crypt
def dic(x):
  list_x =[]
  list_z = []
  for i in x:
    list_x.append(ord(i))
  for i in range(97,123):
    if i not in list_x:
      list_x.append(i)
  list_ = list_x[26-len(x)-1:]
  cr = list_+list_x[:26-len(list_)]
  for i in range(97,123):
    list_z.append(i)
  return dict(map(lambda x,y:[x,y],list_z,cr))


# 構造映射 crypt ---> asc
def dic_2(x):
  list_x =[]
  list_z = []
  for i in x:
    list_x.append(ord(i))
  for i in range(97,123):
    if i not in list_x:
      list_x.append(i)
  list_ = list_x[26-len(x)-1:]
  cr = list_+list_x[:26-len(list_)]
  for i in range(97,123):
    list_z.append(i)
  return dict(map(lambda x,y:[x,y],cr,list_z))

# 密鑰去重
def remove(x):
  unique_x = []
  for i in x:
    if i not in unique_x:
      unique_x.append(i)
  return unique_x

# 加密
def encode():
  x = input('請輸入密鑰字符：')
  if not x.isalpha():
    print('請輸入正確的密鑰格式！')
    exit(0)
  s = input('請輸入明文：')
  print('加密后字符：',end='')
  unique_x = remove(x)
  dic_ = dic(unique_x)
  for i in s:
    if i.isspace():
      print(' ', end='')
    else:
      print(chr(dic_[ord(i)]),end='')


# 解密
def decode():
  x = input('請輸入密鑰字符：')
  if not x.isalpha():
    print('請輸入正確的密鑰格式！')
    exit(0)
  s = input('請輸入密文：')
  print('解密后字符：',end='')
  unique_x = remove(x)
  dic_ = dic_2(unique_x)
  for i in s:
    if i.isspace():
      print(' ',end='')
    else:
      print(chr(dic_[ord(i)]),end='')

# 輸入指令
answer = input(f'請輸入所需的操作：編碼/E or 解碼/D: ')
try:
  if answer.upper() == 'E':
    encode()
  elif answer.upper() == 'D':
    decode()
  else:
    print('輸入錯誤！')
except KeyError:
  print('請正確輸入小寫字母！')