Make backwards compatible with nacl 1.0.x

[securemail.git] / cryptonize.py

diff --git a/cryptonize.py b/cryptonize.py
index 9c1b02fdcfd17b8f179dc4228b98b0a421befe87..7a598d32cfd77cb672415d30d908d76b31b84814 100644 (file)
--- a/cryptonize.py
+++ b/cryptonize.py
@@ -50,6 +50,8 @@ class Cryptonize:
         return plain
         
     def asymmetric_encrypt(self, privkey, pubkey, plain):
+        if not isinstance(plain, bytes):
+            plain = plain.encode('utf-8')
         box = Box(privkey, pubkey)
         if NACL_VERSION < "1.1.0":
             nonce = random(Box.NONCE_SIZE)
@@ -61,6 +63,8 @@ class Cryptonize:
         return cipher
         
     def asymmetric_decrypt(self, privkey, pubkey, cipher):
+        if not isinstance(cipher, bytes):
+            cipher = cipher.encode('utf-8')
         box = Box(privkey, pubkey)
         plain = box.decrypt(cipher)
         box = None