From: Maruo <43961566+maru0123-2004@users.noreply.github.com>
Date: Thu, 6 Oct 2022 02:35:02 +0000 (+0900)
Subject: Update main.py
X-Git-Url: http://git.ipfire.org/gitweb/gitweb.cgi?a=commitdiff_plain;h=793af1a1f19638ce56673fed5428946e26059fe7;p=thirdparty%2Ffastapi%2Fsqlmodel.git

Update main.py
---

diff --git a/sqlmodel/main.py b/sqlmodel/main.py
index de097968..063d6703 100644
--- a/sqlmodel/main.py
+++ b/sqlmodel/main.py
@@ -376,8 +376,8 @@ class SQLModelMetaclass(ModelMetaclass, DeclarativeMeta):
 
 
 def get_sqlachemy_type(field: ModelField) -> Any:
-    if "sa_type" in field.field_info.extra:
-        return field.field_info.extra["sa_type"]
+    if not issubclass(type(field.field_info.sa_type), type(Undefined)):
+        return field.field_info.sa_type
     if issubclass(field.type_, str):
         if field.field_info.max_length:
             return AutoString(length=field.field_info.max_length)