Function encrypt_message

ciphers/transposition_cipher.py:25–36

>>> encrypt_message(6, 'Harshil Darji') 'Hlia rDsahrij'

(key: int, message: str)

def encrypt_message(key: int, message: str) -> str:
    """
    >>> encrypt_message(6, 'Harshil Darji')
    'Hlia rDsahrij'
    """
    cipher_text = [""] * key
    for col in range(key):
        pointer = col
        while pointer < len(message):
            cipher_text[col] += message[pointer]
            pointer += key
    return "".join(cipher_text)


def decrypt_message(key: int, message: str) -> str: