Added EncryptedBooleanField

Author: Gagaro

pgcrypto/fields.py

@@ -255,6 +255,22 @@ def formfield(self, **kwargs):
         return super().formfield(**defaults)
 
 
+class EncryptedBooleanField(BaseEncryptedField):
+    description = _("Boolean")
+    field_cast = "::boolean"
+
+    def formfield(self, **kwargs):
+        defaults = {"form_class": forms.BooleanField}
+        defaults.update(kwargs)
+        return super().formfield(**defaults)
+
+    def to_python(self, value):
+        if value in self.empty_values:
+            return None
+        unencrypted_value = super().to_python(value)
+        return unencrypted_value == 'True'
+
+
 class EncryptedLookup(Lookup):
     patterns = {
         "contains": "%%%s%%",

testapp/fixtures/employees.json

@@ -4,7 +4,8 @@
         "name": "John Smith",
         "ssn": "999-05-6728",
         "salary": "52000.00",
-        "date_hired": "1999-01-23"
+        "date_hired": "1999-01-23",
+        "is_vip": false
     },
     "model": "testapp.employee",
     "pk": 1
@@ -15,7 +16,8 @@
         "ssn": "666-27-9811",
         "salary": "75248.77",
         "date_hired": "1996-02-28",
-        "email": "[email protected]"
+        "email": "[email protected]",
+        "is_vip": true
     },
     "model": "testapp.employee",
     "pk": 2

testapp/models.py

@@ -13,6 +13,7 @@ class Employee(models.Model):
     )
     email = pgcrypto.EncryptedEmailField(unique=True, null=True)
     date_modified = pgcrypto.EncryptedDateTimeField(auto_now=True)
+    is_vip = pgcrypto.EncryptedBooleanField()
 
     def __str__(self):
         return self.name
@@ -30,6 +31,7 @@ class RawEmployee(models.Model):
     date_hired = models.TextField()
     email = models.TextField(null=True)
     date_modified = models.TextField()
+    is_vip = models.BooleanField()
 
     class Meta:
         db_table = "testapp_employee"

testapp/tests.py

@@ -196,6 +196,7 @@ def test_formfields(self):
             "date_hired": forms.DateField,
             "email": forms.EmailField,
             "date_modified": forms.DateTimeField,
+            "is_vip": forms.BooleanField,
         }
         actual = {
             f.name: type(f.formfield())