Question

Can i get Playfair Cipher for python 3 that encrypts a message and decrypts it, could you possibly make it as simple as you can without losing functionality.

please include comments, that would help me better understand

Example of PlayFair Cipher:

https://en.wikipedia.org/wiki/Playfair_cipher

The Playfair cipher uses a 5 by 5 table containing a key word or phrase. Memorization of the keyword and 4 simple rules was all that was required to create the 5 by 5 table and use the cipher.

To generate the key table, one would first fill in the spaces in the table (a modified Polybius square) with the letters of the keyword (dropping any duplicate letters), then fill the remaining spaces with the rest of the letters of the alphabet in order (usually omitting "J" or "Q" to reduce the alphabet to fit; other versions put both "I" and "J" in the same space). The key can be written in the top rows of the table, from left to right, or in some other pattern, such as a spiral beginning in the upper-left-hand corner and ending in the center. The keyword together with the conventions for filling in the 5 by 5 table constitute the cipher key.

To encrypt a message, one would break the message into digrams (groups of 2 letters) such that, for example, "HelloWorld" becomes "HE LL OW OR LD". These digrams will be substituted using the key table. Since encryption requires pairs of letters, messages with an odd number of characters usually append an uncommon letter, such as "X", to complete the final digram. The two letters of the digram are considered opposite corners of a rectangle in the key table. To perform the substitution, apply the following 4 rules, in order, to each pair of letters in the plaintext:

1. If both letters are the same (or only one letter is left), add an "X" after the first letter. Encrypt the new pair and continue. Some variants of Playfair use "Q" instead of "X", but any letter, itself uncommon as a repeated pair, will do.
2. If the letters appear on the same row of your table, replace them with the letters to their immediate right respectively (wrapping around to the left side of the row if a letter in the original pair was on the right side of the row).
3. If the letters appear on the same column of your table, replace them with the letters immediately below respectively (wrapping around to the top side of the column if a letter in the original pair was on the bottom side of the column).
4. If the letters are not on the same row or column, replace them with the letters on the same row respectively but at the other pair of corners of the rectangle defined by the original pair. The order is important – the first letter of the encrypted pair is the one that lies on the same row as the first letter of the plaintext pair.

Answer 1

new=input("Enter new key")
new=new.replace(" ", "")
new=new.upper()
def matrix(a,b,initialvalue):
return [[initialvalue for i in range(a)] for j in range(b)]
  
result=list()
for c in new: #storing new key
if c not in result:
if c=='J':
result.append('I')
else:
result.append(c)
flag=0
for i in range(65,91): #storing other character
if chr(i) not in result:
if i==73 and chr(74) not in result:
result.append("I")
flag=1
elif flag==0 and i==73 or i==74:
pass
else:
result.append(chr(i))
k=0
new_matrix=matrix(5,5,0) #initialize matrix
for i in range(0,5): #making matrix
for j in range(0,5):
new_matrix[i][j]=result[k]
k+=1

def locindex(c): #get location of each character
loc=list()
if c=='J':
c='I'
for i ,j in enumerate(new_matrix):
for k,l in enumerate(j):
if c==l:
loc.append(i)
loc.append(k)
return loc
  
def encrypt(): #Encrypt
message=str(input("ENTER MSG:"))
message=message.upper()
message=message.replace(" ", "")   
i=0
for s in range(0,len(message)+1,2):
if s<len(message)-1:
if message[s]==message[s+1]:
message=message[:s+1]+'X'+message[s+1:]
if len(message)%2!=0:
message=message[:]+'X'
print("CIPHER TEXT:",end=' ')
while i<len(message):
loc=list()
loc=locindex(message[i])
loc1=list()
loc1=locindex(message[i+1])
if loc[1]==loc1[1]:
print("{}{}".format(new_matrix[(loc[0]+1)%5][loc[1]],new_matrix[(loc1[0]+1)%5][loc1[1]]),end=' ')
elif loc[0]==loc1[0]:
print("{}{}".format(new_matrix[loc[0]][(loc[1]+1)%5],new_matrix[loc1[0]][(loc1[1]+1)%5]),end=' ')
else:
print("{}{}".format(new_matrix[loc[0]][loc1[1]],new_matrix[loc1[0]][loc[1]]),end=' ')
i=i+2

def decrypt(): #decrypt
message=str(input("ENTER CIPHER TEXT:"))
message=message.upper()
message=message.replace(" ", "")
print("PLAIN TEXT:",end=' ')
i=0
while i<len(message):
loc=list()
loc=locindex(message[i])
loc1=list()
loc1=locindex(message[i+1])
if loc[1]==loc1[1]:
print("{}{}".format(new_matrix[(loc[0]-1)%5][loc[1]],new_matrix[(loc1[0]-1)%5][loc1[1]]),end=' ')
elif loc[0]==loc1[0]:
print("{}{}".format(new_matrix[loc[0]][(loc[1]-1)%5],new_matrix[loc1[0]][(loc1[1]-1)%5]),end=' ')
else:
print("{}{}".format(new_matrix[loc[0]][loc1[1]],new_matrix[loc1[0]][loc[1]]),end=' ')
i=i+2

while(1):
choice=int(input("\n 1.Encrypt \n 2.Decrypt: \n 3.close"))
if choice==1:
encrypt()
elif choice==2:
decrypt()
elif choice==3:
exit()
else:
print("Choose your option")