Handle nullable columns in Encrypted

Author: maximebf

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sqlorm-py"
-version = "0.2.2"
+version = "0.2.3"
 description = "A new kind or ORM that do not abstract away your database or SQL queries."
 authors = ["Maxime Bouroumeau-Fuseau <[email protected]>"]
 readme = "README.md"

sqlorm/types/encrypted.py

@@ -13,13 +13,17 @@ def __init__(self, key):
         self.key = key
 
     def encrypt(self, raw):
+        if raw is None:
+            return None
         raw = self.pad(raw).encode()
         iv = Random.new().read(AES.block_size)
         key = self.key() if callable(self.key) else self.key
         cipher = AES.new(key, AES.MODE_CBC, iv)
         return base64.b64encode(iv + cipher.encrypt(raw))
 
     def decrypt(self, enc):
+        if enc is None:
+            return None
         enc = base64.b64decode(enc)
         iv = enc[:16]
         key = self.key() if callable(self.key) else self.key