arpakitlib 1.8.257__py3-none-any.whl → 1.8.259__py3-none-any.whl

arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py

@@ -1,48 +1,41 @@
 # arpakit
-import datetime as dt
-from typing import Any, Optional, get_type_hints, get_origin, Union, Annotated, get_args
+from typing import Any, get_type_hints, get_origin, Union, Annotated, get_args
 
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, ConfigDict
 from sqlalchemy import inspect
-from sqlalchemy.orm import ColumnProperty
-from sqlalchemy.sql.sqltypes import (
-    Boolean, Integer, BigInteger, SmallInteger,
-    String, Text, Unicode, UnicodeText,
-    DateTime, Date, Time,
-    Float, Numeric, DECIMAL, LargeBinary, JSON
-)
+from sqlalchemy.orm import ColumnProperty, Mapped
+from sqlalchemy.util import get_annotations
+
+from project.sqlalchemy_db_.sqlalchemy_model import UserDBM
 
 _ARPAKIT_LIB_MODULE_VERSION = "3.0"
 
-_SQLA_TYPE_MAP = {
-    Boolean: bool,
-    Integer: int,
-    BigInteger: int,
-    SmallInteger: int,
-    Float: float,
-    Numeric: float,
-    DECIMAL: float,
-    String: str,
-    Unicode: str,
-    Text: str,
-    UnicodeText: str,
-    LargeBinary: bytes,
-    JSON: dict | list,
-    DateTime: dt.datetime,
-    Date: dt.date,
-    Time: dt.time,
-}
-
-
-def _python_type_from_col(col) -> type | str:
-    try:
-        return col.type.python_type
-    except Exception:
-        for sa_t, py_t in _SQLA_TYPE_MAP.items():
-            if isinstance(col.type, sa_t):
-                return py_t
+
+def __declared_sqlalchemy_column_type(declared_type: Any) -> Any:
+    """
+    Возвращает тип колонки ИСКЛЮЧИТЕЛЬНО из аннотации поля.
+    Разворачивает оболочки:
+      - Annotated[T, ...] -> T
+      - Mapped[T] -> T
+    Если аннотации нет — возвращает Any.
+    """
+    if declared_type is None or declared_type is Any:
         return Any
 
+    origin = get_origin(declared_type)
+
+    # Annotated[T, ...] -> T
+    if origin is Annotated:
+        args = get_args(declared_type)
+        return __declared_sqlalchemy_column_type(args[0]) if args else Any
+
+    # Mapped[T] -> T
+    if origin is Mapped:
+        args = get_args(declared_type)
+        return __declared_sqlalchemy_column_type(args[0]) if args else Any
+
+    return declared_type
+
 
 def _get_property_name_to_type_from_model_class(
         *,
@@ -64,8 +57,9 @@ def _get_property_name_to_type_from_model_class(
                 ret_type = hints.get("return", Any)
             except Exception:
                 if skip_property_if_cannot_define_type:
-                    continue
-                ret_type = Any
+                    ret_type = Any
+                else:
+                    raise
             if exclude_property_names:
                 if property_name in exclude_property_names:
                     continue
@@ -113,7 +107,7 @@ def pydantic_schema_from_sqlalchemy_model(
         *,
         model_name: str | None = None,
         base_model: type[BaseModel] = BaseModel,
-        include_column_defaults: bool = False,
+        include_columns: bool = True,
         exclude_column_names: list[str] | None = None,
         include_properties: bool = False,
         include_property_names: list[str] | None = None,
@@ -122,7 +116,8 @@ def pydantic_schema_from_sqlalchemy_model(
         exclude_property_types: list[type] | None = None,
         filter_property_prefixes: list[str] | None = None,
         remove_property_prefixes: list[str] | None = None,
-        skip_property_if_cannot_define_type: bool = True
+        skip_property_if_cannot_define_type: bool = True,
+        skip_property_name_if_exists: bool = True
 ) -> type[BaseModel]:
     """
     Генерирует Pydantic-модель из колонок SQLAlchemy-модели и (опционально) из @property.
@@ -147,31 +142,18 @@ def pydantic_schema_from_sqlalchemy_model(
     remove_property_prefixes = set(remove_property_prefixes or [])
 
     # 1) Колонки
-    for prop in mapper.attrs:
-        if not isinstance(prop, ColumnProperty):
-            continue
-        if prop.key in exclude_column_names:
-            continue
-
-        column = prop.columns[0]
-        column_type = _python_type_from_col(column)
-
-        # Аннотация типа
-        if column.nullable:
-            annotations[prop.key] = Optional[column_type]  # type: ignore[name-defined]
-        else:
-            annotations[prop.key] = column_type
+    if include_columns:
 
-        # Дефолты, если нужно
-        if include_column_defaults:
-            default_value = None
-            if column.default is not None and getattr(column.default, "is_scalar", False):
-                default_value = column.default.arg
-            elif column.server_default is not None and getattr(column.server_default.arg, "text", None):
-                default_value = column.server_default.arg.text
+        # читаем аннотации класса (include_extras=True нужно для Annotated/Mapped)
+        type_hints = get_annotations(sqlalchemy_model)
 
-            if default_value is not None:
-                attrs[prop.key] = Field(default=default_value)
+        for prop in mapper.column_attrs:
+            if not isinstance(prop, ColumnProperty):
+                continue
+            if prop.key in exclude_column_names:
+                continue
+
+            annotations[prop.key] = __declared_sqlalchemy_column_type(type_hints.get(prop.key, None))
 
     # 2) Свойства (@property)
     if include_properties:
@@ -240,10 +222,11 @@ def pydantic_schema_from_sqlalchemy_model(
                     ):
                         renamed_property_name_to_type[new_property_name] = property_type
                     else:
-                        raise ValueError(
-                            f"Property name '{property_name}' after removing prefix "
-                            f"conflicts with existing name '{new_property_name}'"
-                        )
+                        if not skip_property_name_if_exists:
+                            raise ValueError(
+                                f"Property name '{property_name}' after removing prefix "
+                                f"conflicts with existing name '{new_property_name}'"
+                            )
             property_name_to_type.update(renamed_property_name_to_type)
 
         # добавляем (колонки в приоритете)
@@ -258,3 +241,4 @@ def pydantic_schema_from_sqlalchemy_model(
         attrs["model_config"] = ConfigDict(extra="ignore", arbitrary_types_allowed=True, from_attributes=True)
 
     return type(model_name, (base_model,), attrs)
+

{arpakitlib-1.8.257.dist-info → arpakitlib-1.8.259.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arpakitlib
-Version: 1.8.257
+Version: 1.8.259
 Summary: arpakitlib
 License: Apache-2.0
 Keywords: arpakitlib,arpakit,arpakit-company,arpakitcompany,arpakit_company

{arpakitlib-1.8.257.dist-info → arpakitlib-1.8.259.dist-info}/RECORD

@@ -414,7 +414,7 @@ arpakitlib/ar_need_type_util.py,sha256=XmY1kswz8j9oo5f9CxRu0_zgfvxWrXPYKOj6MM04s
 arpakitlib/ar_openai_api_client_util.py,sha256=dWgsSPXtxNBxS5VRi_NharGQrUXF_YjIfhU3Bj5cW9M,5651
 arpakitlib/ar_parse_command.py,sha256=1WTdQoWVshoDZ1jDaKeTzajfqaYHP3FNO0-REyo1aMY,3003
 arpakitlib/ar_postgresql_util.py,sha256=1AuLjEaa1Lg4pzn-ukCVnDi35Eg1k91APRTqZhIJAdo,945
-arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py,sha256=3FXPiTdDTel8IsWyAmCK7ufzgbbB4WQPlfdEkz7ao3U,10673
+arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py,sha256=MBIP8VDohDB5Z3awbhJehNNRt0xIO4s6sEBezsPHP3s,10413
 arpakitlib/ar_raise_own_exception_if_exception.py,sha256=A6TuNSBk1pHaQ_qxnUmE2LgsNGA1IGqX26b1_HEA4Nc,5978
 arpakitlib/ar_rat_func_util.py,sha256=Ca10o3RJwyx_DJLxjTxgHDO6NU3M6CWgUR4bif67OE4,2006
 arpakitlib/ar_really_validate_email.py,sha256=HBfhyiDB3INI6Iq6hR2WOMKA5wVWWRl0Qun-x__OZ9o,1201
@@ -430,8 +430,8 @@ arpakitlib/ar_sqlalchemy_util.py,sha256=FDva9onjtCPrYZYIHHb93NMwD1WlmaORjiWgPRJQ
 arpakitlib/ar_str_util.py,sha256=2lGpnXDf2h1cBZpVf5i1tX_HCv5iBd6IGnrCw4QWWlY,4350
 arpakitlib/ar_type_util.py,sha256=Cs_tef-Fc5xeyAF54KgISCsP11NHyzIsglm4S3Xx7iM,4049
 arpakitlib/ar_yookassa_api_client_util.py,sha256=VozuZeCJjmLd1zj2BdC9WfiAQ3XYOrIMsdpNK-AUlm0,5347
-arpakitlib-1.8.257.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.257.dist-info/METADATA,sha256=zulZ5lal5ER2B_RxZIvOt0ucF8hbMu8ckkHVVsXjruo,3919
-arpakitlib-1.8.257.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.257.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.257.dist-info/RECORD,,
+arpakitlib-1.8.259.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.259.dist-info/METADATA,sha256=GEDZNXDfp08rmbipsZhX4WwXtmlOGrJbJEpIJ7Q-KDg,3919
+arpakitlib-1.8.259.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.259.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.259.dist-info/RECORD,,