From 8d96a3fb840f180f046223ee6686c447ae3507d5 Mon Sep 17 00:00:00 2001 From: collerek Date: Tue, 9 Mar 2021 17:29:29 +0100 Subject: [PATCH] add length in imp --- ormar/fields/sqlalchemy_encrypted.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/ormar/fields/sqlalchemy_encrypted.py b/ormar/fields/sqlalchemy_encrypted.py index cf2b6af..a2e10b7 100644 --- a/ormar/fields/sqlalchemy_encrypted.py +++ b/ormar/fields/sqlalchemy_encrypted.py @@ -7,6 +7,7 @@ from enum import Enum from typing import Any, Callable, TYPE_CHECKING, Type, Union import sqlalchemy.types as types +from sqlalchemy.engine.default import DefaultDialect from ormar import ModelDefinitionError @@ -115,7 +116,7 @@ class EncryptedString(types.TypeDecorator): # pragma nocover Used to store encrypted values in a database """ - impl = types.String + impl = types.TypeEngine def __init__(self, *args: Any, @@ -140,11 +141,10 @@ class EncryptedString(types.TypeDecorator): # pragma nocover self.max_length = encrypt_max_length def __repr__(self) -> str: - return f"String({self.max_length})" + return f"VARCHAR({self.max_length})" - # - # def load_dialect_impl(self, dialect: DefaultDialect) -> Any: - # dialect.type_descriptor(VARCHAR(self.max_length)) + def load_dialect_impl(self, dialect: DefaultDialect) -> Any: + return dialect.type_descriptor(types.VARCHAR(self.max_length)) @property def key(self):