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)