Python decrypt/encrypt aes with iv

aes.py

import base64
import hashlib
from Crypto.Cipher import AES

class Aes(object):
    def __init__(self, key: str, iv: str):
        self.bs = AES.block_size
        self.key = hashlib.sha256(key.encode()).digest()
        self.iv = iv.encode()

    def encrypt(self, raw):
        return base64.b64encode(
            AES.new(self.key, AES.MODE_CBC, self.iv).encrypt(
                self._pad_pkcs7(raw).encode()
            )
        )

    def decrypt(self, enc: str or bytes):
        
        if not isinstance(enc, (bytes, bytearray)):
            enc = enc.encode()
        
        return self._unpad_pkcs7(
            AES.new(self.key, AES.MODE_CBC, self.iv).decrypt(base64.b64decode(enc))
        )

    def _pad_pkcs7(self, source):
        return source + chr(AES.block_size - len(source) % AES.block_size) * (
            AES.block_size - len(source) % AES.block_size
        )

    def _unpad_pkcs7(self, source):
        return source[0 : len(source) - source[len(source) - 1]]


if __name__ == "__main__":

    iv   = "1234567890123456" # needs to be 16 bytes/chars long
    key  = "key"              
    text = "Hello World"
    
    encrypted = Aes(key, iv).encrypt(text)
    print('Encrypted:', encrypted)
    
    decrypted = Aes(key, iv).decrypt(encrypted.decode()) # encrypted string can be str or bytes (there is safety check)
    print('Decrypted:', decrypted.decode())