From 2c31ad735d676b38a4dc4861adf1105b880f7f6d Mon Sep 17 00:00:00 2001
From: collerek <collerek@gmail.com>
Date: Tue, 9 Mar 2021 17:10:39 +0100
Subject: [PATCH] add crypto to reqs

---
 ormar/fields/sqlalchemy_encrypted.py | 10 ++++------
 requirements.txt                     |  1 +
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/ormar/fields/sqlalchemy_encrypted.py b/ormar/fields/sqlalchemy_encrypted.py
index 10a017b..cf2b6af 100644
--- a/ormar/fields/sqlalchemy_encrypted.py
+++ b/ormar/fields/sqlalchemy_encrypted.py
@@ -6,9 +6,7 @@ import json
 from enum import Enum
 from typing import Any, Callable, TYPE_CHECKING, Type, Union
 
-from sqlalchemy import String
-from sqlalchemy.engine.default import DefaultDialect
-from sqlalchemy.types import TypeDecorator
+import sqlalchemy.types as types
 
 from ormar import ModelDefinitionError
 
@@ -112,12 +110,12 @@ backends_map = {
 }
 
 
-class EncryptedString(TypeDecorator):  # pragma nocover
+class EncryptedString(types.TypeDecorator):  # pragma nocover
     """
     Used to store encrypted values in a database
     """
 
-    impl = String
+    impl = types.String
 
     def __init__(self,
                  *args: Any,
@@ -143,6 +141,7 @@ class EncryptedString(TypeDecorator):  # pragma nocover
 
     def __repr__(self) -> str:
         return f"String({self.max_length})"
+
     #
     # def load_dialect_impl(self, dialect: DefaultDialect) -> Any:
     #     dialect.type_descriptor(VARCHAR(self.max_length))
@@ -217,4 +216,3 @@ class EncryptedString(TypeDecorator):  # pragma nocover
 
     def _coerce(self, value):
         return self.underlying_type._coerce(value)
-
diff --git a/requirements.txt b/requirements.txt
index f47a05d..4d7610c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,6 +5,7 @@ pydantic
 sqlalchemy
 typing_extensions
 orjson
+cryptography
 
 # Async database drivers
 aiomysql