check if data binding not work only in sqlite

ormar/fields/base.py

@@ -5,6 +5,9 @@ from pydantic import Field, Json, typing
 from pydantic.fields import FieldInfo, Required, Undefined
 
 import ormar  # noqa I101
+from ormar import ModelDefinitionError
+from ormar.fields.sqlalchemy_encrypted import EncryptBackend, EncryptBackends, \
+    EncryptedString
 
 if TYPE_CHECKING:  # pragma no cover
     from ormar.models import Model
@@ -49,6 +52,11 @@ class BaseField(FieldInfo):
     self_reference: bool = False
     self_reference_primary: Optional[str] = None
 
+    encrypt_secret: str
+    encrypt_backend: EncryptBackends = EncryptBackends.NONE
+    encrypt_custom_backend: Type[EncryptBackend] = None
+    encrypt_max_length: int = 5000
+
     default: Any
     server_default: Any
 
@@ -93,10 +101,11 @@ class BaseField(FieldInfo):
         :rtype: bool
         """
         return (
-            field_name not in ["default", "default_factory", "alias", "allow_mutation"]
-            and not field_name.startswith("__")
-            and hasattr(cls, field_name)
-            and not callable(getattr(cls, field_name))
+                field_name not in ["default", "default_factory", "alias",
+                                   "allow_mutation"]
+                and not field_name.startswith("__")
+                and hasattr(cls, field_name)
+                and not callable(getattr(cls, field_name))
         )
 
     @classmethod
@@ -205,7 +214,7 @@ class BaseField(FieldInfo):
         :rtype: bool
         """
         return cls.default is not None or (
-            cls.server_default is not None and use_server
+                cls.server_default is not None and use_server
         )
 
     @classmethod
@@ -238,7 +247,7 @@ class BaseField(FieldInfo):
                 ondelete=con.ondelete,
                 onupdate=con.onupdate,
                 name=f"fk_{cls.owner.Meta.tablename}_{cls.to.Meta.tablename}"
-                f"_{cls.to.get_column_alias(cls.to.Meta.pkname)}_{cls.name}",
+                     f"_{cls.to.get_column_alias(cls.to.Meta.pkname)}_{cls.name}",
             )
             for con in cls.constraints
         ]
@@ -256,25 +265,46 @@ class BaseField(FieldInfo):
         :return: actual definition of the database column as sqlalchemy requires.
         :rtype: sqlalchemy.Column
         """
-        column = sqlalchemy.Column(
-            cls.alias or name,
-            cls.column_type,
-            *cls.construct_constraints(),
-            primary_key=cls.primary_key,
-            nullable=cls.nullable and not cls.primary_key,
-            index=cls.index,
-            unique=cls.unique,
-            default=cls.default,
-            server_default=cls.server_default,
-        )
+        if cls.encrypt_backend == EncryptBackends.NONE:
+            column = sqlalchemy.Column(
+                cls.alias or name,
+                cls.column_type,
+                *cls.construct_constraints(),
+                primary_key=cls.primary_key,
+                nullable=cls.nullable and not cls.primary_key,
+                index=cls.index,
+                unique=cls.unique,
+                default=cls.default,
+                server_default=cls.server_default,
+            )
+        else:
+            if cls.primary_key or cls.is_relation:
+                raise ModelDefinitionError("Primary key field and relations fields"
+                                           "cannot be encrypted!")
+            column = sqlalchemy.Column(
+                cls.alias or name,
+                EncryptedString(
+                    _field_type=cls,
+                    encrypt_secret=cls.encrypt_secret,
+                    encrypt_backend=cls.encrypt_backend,
+                    encrypt_custom_backend=cls.encrypt_custom_backend,
+                    encrypt_max_length=cls.encrypt_max_length
+                ),
+                nullable=cls.nullable,
+                index=cls.index,
+                unique=cls.unique,
+                default=cls.default,
+                server_default=cls.server_default,
+            )
+
         return column
 
     @classmethod
     def expand_relationship(
-        cls,
-        value: Any,
-        child: Union["Model", "NewBaseModel"],
-        to_register: bool = True,
+            cls,
+            value: Any,
+            child: Union["Model", "NewBaseModel"],
+            to_register: bool = True,
     ) -> Any:
         """
         Function overwritten for relations, in basic field the value is returned as is.
@@ -302,7 +332,7 @@ class BaseField(FieldInfo):
         :rtype: None
         """
         if cls.owner is not None and (
-            cls.owner == cls.to or cls.owner.Meta == cls.to.Meta
+                cls.owner == cls.to or cls.owner.Meta == cls.to.Meta
         ):
             cls.self_reference = True
             cls.self_reference_primary = cls.name