Merge pull request #1 from Airbase/SM-psycopg3

Add support for psycopg3
Airbase · May 9, 2024 · c4740d5 · c4740d5
2 parents 9056d4b + 523e51f
commit c4740d5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fernet_fields/fields.py b/fernet_fields/fields.py
@@ -120,7 +120,10 @@ class EncryptedEmailField(EncryptedField, models.EmailField):
 
 
 class EncryptedIntegerField(EncryptedField, models.IntegerField):
-    pass
+    def get_db_prep_save(self, value, connection):
+        if value is not None:
+            retval = self.fernet.encrypt(force_bytes(value))
+            return connection.Database.Binary(retval)
 
 
 class EncryptedDateField(EncryptedField, models.DateField):