Only check for Pydantic Types with Pydantic V2

Author: dereisele

sqlmodel/main.py

@@ -26,14 +26,8 @@
 )
 
 from pydantic import (
-    AwareDatetime,
     BaseModel,
     EmailStr,
-    FutureDate,
-    FutureDatetime,
-    NaiveDatetime,
-    PastDate,
-    PastDatetime,
 )
 from pydantic.fields import FieldInfo as PydanticFieldInfo
 from sqlalchemy import (
@@ -91,6 +85,16 @@
 )
 from .sql.sqltypes import AutoString
 
+if IS_PYDANTIC_V2:
+    from pydantic import (
+        AwareDatetime,
+        FutureDate,
+        FutureDatetime,
+        NaiveDatetime,
+        PastDate,
+        PastDatetime,
+    )
+
 if TYPE_CHECKING:
     from pydantic._internal._model_construction import ModelMetaclass as ModelMetaclass
     from pydantic._internal._repr import Representation as Representation
@@ -689,18 +693,23 @@ def get_sqlalchemy_type(field: Any) -> Any:
         return Boolean
     if issubclass(type_, int):
         return Integer
-    if issubclass(type_, (datetime, FutureDatetime, PastDatetime)):
+    if issubclass(type_, datetime):
         return DateTime
-    if issubclass(type_, (date, FutureDate, PastDate)):
+    if issubclass(type_, date):
         return Date
     if issubclass(type_, timedelta):
         return Interval
     if issubclass(type_, time):
         return Time
-    if issubclass(type_, AwareDatetime):
-        return DateTime(timezone=True)
-    if issubclass(type_, NaiveDatetime):
-        return DateTime(timezone=False)
+    if IS_PYDANTIC_V2:
+        if issubclass(type_, (FutureDate, PastDate)):
+            return DateTime
+        if issubclass(type_, (FutureDatetime, PastDatetime)):
+            return DateTime
+        if issubclass(type_, AwareDatetime):
+            return DateTime(timezone=True)
+        if issubclass(type_, NaiveDatetime):
+            return DateTime(timezone=False)
     if issubclass(type_, bytes):
         return LargeBinary
     if issubclass(type_, Decimal):