RSA implementation

rsa.py

#!/usr/bin/env python3
from typing import List

class RSAConfig:
    def __init__(self, p, q, e, d):
        self.p = p
        self.q = q
        self.n = p * q
        self.phi = (p-1) * (q-1)
        self.e = e
        self.d = d
        # self.alphabet = alphabet

    def encrypt(self, nums: List[int]) -> List[int]:
        return [
            (chunk ** self.e) % self.n
            for chunk in nums
        ]

    def decrypt(self, nums: List[int]) -> List[int]:
        return [
            (chunk ** self.d) % self.n
            for chunk in nums
        ]


if __name__ == '__main__':
    rsa = RSAConfig(11, 17, 19, 59)
    message = [18, 91, 24, 2]
    
    # Just to confirm it works as expected
    assert rsa.decrypt(rsa.encrypt(message)) == message
    
    print(f"Original message: {message}")
    print(f"Encrypted: {(encrypted := rsa.encrypt(message))}")
    print(f"Decrypted encrypted: {rsa.decrypt(encrypted)}")
    print(f"Decrypted original: {rsa.decrypt(message)}")