python playfair

#########################Playfair密码#########################  
  
#约定1：若明文字母数量为奇数，在明文末尾添加一个‘Z‘  
#约定2：‘I‘作为‘J‘来处理  
  
#字母表  
letter_list=‘ABCDEFGHJKLMNOPQRSTUVWXYZ‘  
  
#密码表  
T_letter=[‘‘,‘‘,‘‘,‘‘,‘‘]  
  
#根据密钥建立密码表  
def Create_Matrix(key):  
  key=Remove_Duplicates(key)  #移除密钥中的重复字母  
  key=key.replace(‘ ‘,‘‘) #去除密钥中的空格  
    
  for ch in letter_list:  #根据密钥获取新组合的字母表  
    if ch not in key:  
      key+=ch  
    
  j=0  
  for i in range(len(key)): #将新的字母表里的字母逐个填入密码表中，组成5*5的矩阵  
    T_letter[j]+=key[i]     #j用来定位字母表的行  
    if 0==(i+1)%5:  
      j+=1  
  
#移除字符串中重复的字母  
def Remove_Duplicates(key):  
  key=key.upper() #转成大写字母组成的字符串  
  _key=‘‘  
  for ch in key:  
    if ch==‘I‘:  
      ch=‘J‘  
    if ch in _key:  
      continue  
    else:  
      _key+=ch  
  return _key  
  
#获取字符在密码表中的位置  
def Get_MatrixIndex(ch):  
  for i in range(len(T_letter)):  
    for j in range(len(T_letter)):  
      if ch==T_letter[i][j]:  
        return i,j #i为行，j为列  
  
#加密  
def Encrypt(plaintext,T_letter):  
  ciphertext=‘‘  
    
  if len(plaintext) % 2 !=0:  #如果新的明文长度为奇数，在其末尾添上‘Z‘  
    plaintext+=‘Z‘  
    
  i=0  
  while i<len(plaintext): #对明文进行遍历  
    if True==plaintext[i].isalpha():  #如果是明文是字母的话，  
      j=i+1                           #则开始对该字母之后的明文进行遍历，  
      while j<len(plaintext):         #直到遍历到字母，进行加密  
        if True==plaintext[j].isalpha():  
          if ‘I‘==plaintext[i].upper():             #  
            x=Get_MatrixIndex(‘J‘)                  #  
          else:                                     #  
            x=Get_MatrixIndex(plaintext[i].upper()) #对字符在密码表中的坐标  
          if ‘I‘==plaintext[j].upper():             #进行定位,同时将‘I‘作为  
              y=Get_MatrixIndex(‘J‘)                #‘J‘来处理  
          else:                                     #  
            y=Get_MatrixIndex(plaintext[j].upper()) #  
            
          if x[0]==y[0]:    #如果在同一行  
            ciphertext+=T_letter[x[0]][(x[1]+1)%5]+T_letter[y[0]][(y[1]+1)%5]  
          elif x[1]==y[1]:  #如果在同一列  
            ciphertext+=T_letter[(x[1]+1)%5][x[0]]+T_letter[(y[1]+1)%5][y[0]]  
          else:             #如果不同行不同列  
            ciphertext+=T_letter[x[0]][y[1]]+T_letter[y[0]][x[1]]  
          break;  #每组明文对加密完成后，结束本次对明文的遍历  
        j+=1  
      i=j+1  #每次对明文的遍历是从加密过后的明文的后一个明文开始的,结束本次循环  
      continue  
    else:  
      ciphertext+=plaintext[i]  #如果明文不是字母，直接加到密文上  
    i+=1  
      
  return ciphertext  
  
#解密  
def Decrypt(ciphertext,T_letter):  
  plaintext=‘‘  
  if len(ciphertext) % 2 !=0:  #如果新的密文长度为奇数，在其末尾添上‘Z‘  
    ciphertext+=‘Z‘  
    
  i=0  
  while i<len(ciphertext): #对密文进行遍历  
    if True==ciphertext[i].isalpha():  #如果是密文是字母的话，  
      j=i+1                            #则开始对该字母之后的密文进行遍历，  
      while j<len(ciphertext):         #直到遍历到字母，进行解密  
        if True==ciphertext[j].isalpha():  
          if ‘I‘==ciphertext[i].upper():              #  
            x=Get_MatrixIndex(‘J‘)                    #  
          else:                                       #  
            x=Get_MatrixIndex(ciphertext[i].upper())  #对字符在密码表中的坐标  
          if ‘I‘==ciphertext[j].upper():              #进行定位,同时将‘I‘作为  
              y=Get_MatrixIndex(‘J‘)                  #‘J‘来处理  
          else:                                       #  
            y=Get_MatrixIndex(ciphertext[j].upper())  #  
            
          if x[0]==y[0]:    #如果在同一行  
            plaintext+=T_letter[x[0]][(x[1]-1)%5]+T_letter[y[0]][(y[1]-1)%5]  
          elif x[1]==y[1]:  #如果在同一列  
            plaintext+=T_letter[(x[1]-1)%5][x[0]]+T_letter[(y[1]-1)%5][y[0]]  
          else:             #如果不同行不同列  
            plaintext+=T_letter[x[0]][y[1]]+T_letter[y[0]][x[1]]  
          break;  #每组密文对解密完成后，结束本次对密文的遍历  
        j+=1  
      i=j+1  #每次对密文的遍历是从解密过后的密文的后一个密文开始的,结束本次循环  
      continue  
    else:  
      plaintext+=ciphertext[i]  #如果密文不是字母，直接加到明文上  
    i+=1  
      
  return plaintext  
  
#主函数  
if __name__==‘__main__‘:  
  print("加密请按D,解密请按E:")  
  user_input=input();  
  while(user_input!=‘D‘ and user_input!=‘E‘):#输入合法性检测  
    print("输入有误!请重新输入:")  
    user_input=input()  
    
  print(‘请输入密钥，密钥由英文字母组成:‘)  
  key=input()  
    
  Create_Matrix(key)  #建立密码表  
  
  if user_input==‘D‘: #加密  
    print(‘请输入明文:‘)  
    plaintext=input()  
    print("密文为:\n%s" % Encrypt(plaintext,T_letter))  
  else:               #解密  
    print(‘请输入密文:‘)  
    ciphertext=input()  
    print(‘明文为:\n%s‘ % Decrypt(ciphertext,T_letter))