Make associated_data optional.

Ayrx · Dec 13, 2014 · 7c0cd82 · 7c0cd82
1 parent 1f4fb2f
commit 7c0cd82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aead/__init__.py b/aead/__init__.py
@@ -28,7 +28,7 @@ def __init__(self, key, backend=default_backend()):
     def generate_key(cls):
         return base64.urlsafe_b64encode(os.urandom(32))
 
-    def encrypt(self, data, associated_data):
+    def encrypt(self, data, associated_data=b""):
         iv = os.urandom(16)
         return base64.urlsafe_b64encode(
             self._encrypt_from_parts(data, associated_data, iv)
@@ -55,7 +55,7 @@ def _encrypt_from_parts(self, data, associated_data, iv):
 
         return cipher_text + mac[:16]
 
-    def decrypt(self, data, associated_data):
+    def decrypt(self, data, associated_data=b""):
         decoded_data = base64.urlsafe_b64decode(data)
         mac = decoded_data[-16:]
         iv = decoded_data[0:16]