made changes for ruff test

Signed-off-by: grpathak22 <[email protected]>

Author: grpathak22

Diff for: ciphers/aes_128.py

@@ -15,6 +15,7 @@
 '''
 
 
+
 def aes_encrypt(plaintext: str, key: str) -> str:
     """
     AES-128 Encryption using CBC mode and PKCS7 padding.
@@ -30,9 +31,10 @@ def aes_encrypt(plaintext: str, key: str) -> str:
     >>> aes_decrypt(enc, key) == msg
     True
     """
-    cipher = AES.new(key.encode('utf-8'), AES.MODE_CBC)
-    ciphertext = cipher.encrypt(pad(plaintext.encode('utf-8'), AES.block_size))
-    return base64.b64encode(cipher.iv + ciphertext).decode('utf-8')
+    cipher = AES.new(key.encode("utf-8"), AES.MODE_CBC)
+    ciphertext = cipher.encrypt(pad(plaintext.encode("utf-8"), AES.block_size))
+    return base64.b64encode(cipher.iv + ciphertext).decode("utf-8")
+
 
 def aes_decrypt(ciphertext: str, key: str) -> str:
     """
@@ -50,8 +52,9 @@ def aes_decrypt(ciphertext: str, key: str) -> str:
     True
     """
     raw = base64.b64decode(ciphertext)
-    cipher = AES.new(key.encode('utf-8'), AES.MODE_CBC, iv=raw[:AES.block_size])
-    return unpad(cipher.decrypt(raw[AES.block_size:]), AES.block_size).decode('utf-8')
+    cipher = AES.new(key.encode("utf-8"), AES.MODE_CBC, iv=raw[: AES.block_size])
+    return unpad(cipher.decrypt(raw[AES.block_size :]), AES.block_size).decode("utf-8")
+
 
 def main() -> None:
     key = input("Enter 16-character key (AES-128): ")
@@ -68,7 +71,9 @@ def main() -> None:
     decrypted_message = aes_decrypt(encrypted_message, key)
     print("Decrypted message:", decrypted_message)
 
+
 if __name__ == "__main__":
     import doctest
+
     doctest.testmod()
     main()