python實現凱撒密碼

def ask():
 while True:
  print("Welcome to you coming!")
  print("you can choose mode : encrypt(e) or decrypt(d)")  #有解密和加密模式
  print("If you choose encrypt ,you can lock the message!") #加密提示
  print("If you choose decrypt ,you can unlock the message!") #解密提示
  print("If you wanna exit , input 'q'!!")     #退出提示
  mode = input("choose:").lower()        #將輸入的模式進行變換（從大寫變小寫，小寫部分不變）
  if mode in 'encrypt e decrypt d q'.split():     #當模式是被要求的encrypt e（加密模式）decrypt d（解密模式） q（退出）時進行下一步操作
   #print(mode)           #打印輸入的模式
   return mode            #將mode的值作為返回值
  else:
   print('Please input right option!!')     #輸出提示


def getKey(mode):
 key = 0               #設置默認的key
 while key <= 0 or key >= 26:         #限制key的范圍在（1-25以內的數）
  try:              #這裡進行異常處理，將非整數類型的輸入進行錯誤提示打印
   key = int(input("Please input your key:(1-26)"))
  except:
   print("Please input correct number!!")
 #對解密的密匙進行變換
 if mode == 'd' or mode == 'decrypt':
  key = -key
 return key

def getMessage(key):
 #輸入信息
 while True:
  informetion = input("Please input message!!")    #輸入要解密或者加密的信息
  if informetion.isalpha():         #判斷輸入的字符串是否為純字母
   break
  else:
   print("Please input continuous character!!!")   #輸錯提示
 message = ''             #設置輸出的初始值
 for x in informetion:           #將輸入信息裡的進行逐一字母加密/解密
  num = ord(x)            #將單一字符通過ascii表進行轉換，將字母轉換為數字
  num += key             #加上key的值進行下列運算
  if x.isupper():            #判斷是否是大寫字母
   if num > ord('Z'):          #對超出ascii對應數值的范圍進行處理
    num -=26
    print(message)
   elif num < ord('A'):
    num +=26
    print(message)
  elif x.islower():           #判斷是否小寫字母
   if num > ord('z'):
    num -=26
   elif num < ord('a'):
    num +=26
  message += chr(num)           #將單一字符通過ascii表進行轉換，將數字轉換為字母
 return message             #返回message的值
  
if __name__=="__main__":           #主程序
 mode = ask()             #將ask()返回值存於mode變量中
 if mode == 'q':             #進行退出判斷
  print('welcome!!')
 else:
  key = getKey(mode)            #將mode變量的值帶入getKey函數中運行，運行後將key的值存入到key變量中
  last = getMessage(key)          #將key變量的值帶入到getMessage函數中，運行後將message的值存入到last變量中
  print(last)