strideutils 0.2.3__tar.gz → 0.2.4__tar.gz

{strideutils-0.2.3 → strideutils-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.2.3
+Version: 0.2.4
 Summary: 
 Author: Your Name
 Author-email: you@example.com
@@ -17,6 +17,7 @@ Requires-Dist: google-auth-oauthlib (>=1.1.0)
 Requires-Dist: google-cloud-bigquery (>=3.11.4)
 Requires-Dist: gspread (>=6.0.2)
 Requires-Dist: pandas (>=2.1.1)
+Requires-Dist: pycryptodome (>=3.20.0)
 Requires-Dist: python-dateutil (>=2.8.2)
 Requires-Dist: pytz (>=2023.3)
 Requires-Dist: pyyaml (>=6.0.1)

{strideutils-0.2.3 → strideutils-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "strideutils"
-version = "0.2.3"
+version = "0.2.4"
 description = ""
 authors = ["Your Name <you@example.com>"]
 readme = "README.md"
@@ -21,9 +21,13 @@ pandas = ">=2.1.1"
 twilio = ">=8.10.0"
 tabulate = ">=0.9.0"
 pyyaml = ">=6.0.1"
-python-dateutil= ">=2.8.2"
+python-dateutil = ">=2.8.2"
+pycryptodome = ">=3.20.0"
 
 
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.1.1"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

strideutils-0.2.4/strideutils/cryptography.py

@@ -0,0 +1,37 @@
+import base64
+from typing import Tuple
+
+from Crypto.Cipher import AES
+from Crypto.Random import get_random_bytes
+from Crypto.Util.Padding import pad, unpad
+
+
+def encrypt_message(message: str, key: bytes) -> Tuple[str, str]:
+    """Encrypt the message using AES in CBC mode."""
+    # Generate a random initialization vector
+    iv = get_random_bytes(AES.block_size)
+
+    # Initialize the cipher for encryption
+    cipher_encrypt = AES.new(key, AES.MODE_CBC, iv)
+
+    # Encrypt the message
+    # The message needs to be padded to make sure its length is a multiple of the block size
+    encrypted = cipher_encrypt.encrypt(pad(message.encode(), AES.block_size))
+
+    # Return the encrypted data and the IV
+    return base64.b64encode(encrypted).decode(), base64.b64encode(iv).decode()
+
+
+def decrypt_message(encrypted: str, iv: str, key: bytes) -> str:
+    """Decrypt the message using AES in CBC mode."""
+    # Decode the encrypted message and IV from base64
+    encrypted = base64.b64decode(encrypted)
+    iv = base64.b64decode(iv)
+
+    # Initialize the cipher for decryption
+    cipher_decrypt = AES.new(key, AES.MODE_CBC, iv)
+
+    # Decrypt the message and unpad it
+    decrypted = unpad(cipher_decrypt.decrypt(encrypted), AES.block_size).decode()
+
+    return decrypted