python 對于實現(xiàn)rsa加密算法

import base64

import rsa


class GenerateKey(object):
    d = "ascii"

    def generate_keys(self, bits=1024):
        (pubkey, privkey) = rsa.newkeys(bits)
        pem_pubkey = rsa.PublicKey.save_pkcs1(pubkey).decode(self.d)
        b64_pubkey = self.b64_encrypt(pem_pubkey).decode(self.d)

        pem_privkey = rsa.PrivateKey.save_pkcs1(privkey).decode(self.d)
        b64_privkey = self.b64_encrypt(pem_privkey).decode(self.d)

        return b64_pubkey, b64_privkey

    def decrypt(
        self, ciphertext: str | bytes, private_key_pem: bytes | str | rsa.PrivateKey
    ) -> str:
        """私鑰入?yún)⒈仨毷莗em格式， 不接受（模數(shù)，指數(shù)） 的類型

        Args:
            ciphertext (str): 字符串
            private_key_pem (str): pem字節(jié)串

        Returns:
            _type_: _description_
        """
        if isinstance(private_key_pem, (str, bytes)):
            private_key_pem = private_key_pem.strip()
            if isinstance(private_key_pem, str):
                private_key_pem = private_key_pem.encode()
            private_key = rsa.PrivateKey.load_pkcs1(
                private_key_pem, "PEM"
            )  # 解析成為 （模數(shù)，指數(shù)） 類型
        else:
            private_key = private_key_pem
        if isinstance(ciphertext, str):
            ciphertext = ciphertext.encode()

        decrypted_text = rsa.decrypt(ciphertext, private_key)

        return decrypted_text.decode()

    def encrypt(self, plaintext: str | bytes, public_key: bytes | str | rsa.PublicKey):
        # rsa 加密函數(shù)
        """僅接受pem格式數(shù)據(jù)，不支持（模數(shù)，指數(shù)） 的類型

        Args:
            plaintext (str): 需要加密的文本
            public_key (rsa.PublicKey): 使用的公鑰字節(jié)串

        Returns:
            _type_: _description_
        """
        if isinstance(plaintext, str):
            plaintext = plaintext.encode()

        if isinstance(public_key, (str, bytes)):
            public_key = public_key.strip()
            if isinstance(public_key, str):
                public_key = public_key.encode()
            public_key = rsa.PublicKey.load_pkcs1(public_key, "PEM")  # 解析成為 （模數(shù)，指數(shù)） 類型
        else:
            public_key = public_key

        ciphertext = rsa.encrypt(plaintext, public_key)
        return ciphertext.decode()

    def b64_encrypt(self, test: str):
        # b64編碼
        return base64.b64encode(test.encode(self.d))

    def b64_decrypt(self, test: str):
        # b64解碼
        return base64.b64decode(test.encode(self.d))