python/sqlmodel
1
0
Fork 0
mirror of https://github.com/fastapi/sqlmodel.git synced 2025-08-26 10:01:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix enum type checks ordering in get_sqlalchemy_type (#669)

Browse Source 
Co-authored-by: Pierre Cheynier <p.cheynier@criteo.com>
This commit is contained in:
Sebastián Ramírez
2023-10-23 13:22:44 +04:00
committed by GitHub
parent 40c1af9202
commit d3261cab59
2 changed files with 45 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sqlmodel/main.py
View File

@ -384,6 +384,9 @@ class SQLModelMetaclass(ModelMetaclass, DeclarativeMeta):
def get_sqlalchemy_type(field: ModelField) -> Any:
if isinstance(field.type_, type) and field.shape == SHAPE_SINGLETON:
# Check enums first as an enum can also be a str, needed by Pydantic/FastAPI
if issubclass(field.type_, Enum):
return sa_Enum(field.type_)
if issubclass(field.type_, str):
if field.field_info.max_length:
return AutoString(length=field.field_info.max_length)
@ -402,8 +405,6 @@ def get_sqlalchemy_type(field: ModelField) -> Any:
return Interval
if issubclass(field.type_, time):
return Time
if issubclass(field.type_, Enum):
return sa_Enum(field.type_)
if issubclass(field.type_, bytes):
return LargeBinary
if issubclass(field.type_, Decimal):