PaiGramTeam/sqlmodel
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/sqlmodel.git synced 2024-11-25 09:27:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

python version for types.UnionType/Annotated

Browse Source
This commit is contained in:
honglei 2023-08-15 00:28:01 +08:00
parent a22dac8113
commit ce0064b286
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
sqlmodel/main.py
View File

@ -52,9 +52,14 @@ from .sql.sqltypes import GUID, AutoString
from .typing import SQLModelConfig
if sys.version_info >= (3, 8):
from typing import get_args, get_origin, Annotated
from typing import get_args, get_origin
else:
from typing_extensions import get_args, get_origin, Annotated
from typing_extensions import get_args, get_origin
if sys.version_info >= (3, 9):
from typing import Annotated
else:
from typing_extensions import Annotated
_T = TypeVar("_T")
NoArgAnyCallable = Callable[[], Any]
@ -437,7 +442,12 @@ def get_sqlalchemy_type(field: FieldInfo) -> Any:
type_: Optional[type] = field.annotation
# Resolve Optional/Union fields
if type_ is not None and get_origin(type_) in (types.UnionType, Union):
def is_optional_or_union(type_):
if sys.version_info >= (3, 10):
return get_origin(type_) in (types.UnionType, Union)
else:
return get_origin(type_) is Union
if type_ is not None and is_optional_or_union(type_):
bases = get_args(type_)
if len(bases) > 2:
raise RuntimeError(