Codepy - Python Online

def playfair(key, text):
    print(f"For {text}:", end=" ")
    ok = set(key)
    lol = key
    for c in "abcdefghiklmnopqrstuvwxyz":
        if c not in ok:
            lol += c
    table = []
    text = list(text)
    for i in range(len(text)):
        if text[i] == "j":
            text[i] = "i"
    text = ''.join(text)
    for i in range(0, 26, 5):
        table.append((lol[i:i+5]))
    # print(table)
    ans = ''
    if len(text)%2:
        text += 'x'
    for idx in range(0, len(text), 2):
        for i in range(len(table)):
            if text[idx] in table[i]:
                row1 = i
                column1 = table[i].index(text[idx])
                break
        for i in range(len(table)):
            if text[idx+1] in table[i]:
                row2 = i
                column2 = table[i].index(text[idx+1])
                break
        if row1 == row2:
            ans += table[row1][(column1 + 1)%5] + table[row1][(column2+1)%5]
        elif column1 == column2:
            ans += table[(row1+1)%5][column1] + table[(row2+1)%5][column1]
        else:
            ans += table[row1][column2] + table[row2][column1]
    print(ans)

key = input("Enter key: ")
text = input("Enter text: ")
playfair(key, text)

Share this code with others