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

arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/user.py

@@ -1,4 +1,3 @@
-from __future__ import annotations
 
 import datetime as dt
 from typing import TYPE_CHECKING, Any
@@ -92,14 +91,14 @@ class UserDBM(SimpleDBM):
     )
 
     # many to one
-    user_tokens: Mapped[list[UserTokenDBM]] = relationship(
+    user_tokens: Mapped[list["UserTokenDBM"]] = relationship(
         "UserTokenDBM",
         uselist=True,
         back_populates="user",
         foreign_keys="UserTokenDBM.user_id",
         cascade="all, delete-orphan"
     )
-    verification_codes: Mapped[list[VerificationCodeDBM]] = relationship(
+    verification_codes: Mapped[list["VerificationCodeDBM"]] = relationship(
         "VerificationCodeDBM",
         uselist=True,
         back_populates="user",

arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py

@@ -4,14 +4,13 @@ from typing import Any, get_type_hints, get_origin, Union, Annotated, get_args
 from pydantic import BaseModel, ConfigDict
 from sqlalchemy import inspect
 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"
 
 
-def __declared_sqlalchemy_column_type(declared_type: Any) -> Any:
+def _define_sqlalchemy_column_mapped_type(type_: Any) -> Any:
     """
     Возвращает тип колонки ИСКЛЮЧИТЕЛЬНО из аннотации поля.
     Разворачивает оболочки:
@@ -19,22 +18,13 @@ def __declared_sqlalchemy_column_type(declared_type: Any) -> Any:
       - 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
+    origin = get_origin(type_)
 
-    # Mapped[T] -> T
     if origin is Mapped:
-        args = get_args(declared_type)
-        return __declared_sqlalchemy_column_type(args[0]) if args else Any
+        return get_args(type_)[0] if get_args(type_) else Any
 
-    return declared_type
+    return type_
 
 
 def _get_property_name_to_type_from_model_class(
@@ -102,6 +92,16 @@ def _type_matches(*, type_: Any, allowed_types: list[type]) -> bool:
     return False
 
 
+def _get_sqlalchemy_mapped_types(sqlalchemy_model):
+    result = {}
+    for cls in reversed(sqlalchemy_model.__mro__):
+        annotations = getattr(cls, "__annotations__", {})
+        for field, annotation in annotations.items():
+            if get_origin(annotation) is Mapped:
+                result[field] = annotation
+    return result
+
+
 def pydantic_schema_from_sqlalchemy_model(
         sqlalchemy_model: type,
         *,
@@ -144,16 +144,13 @@ def pydantic_schema_from_sqlalchemy_model(
     # 1) Колонки
     if include_columns:
 
-        # читаем аннотации класса (include_extras=True нужно для Annotated/Mapped)
-        type_hints = get_annotations(sqlalchemy_model)
-
-        for prop in mapper.column_attrs:
-            if not isinstance(prop, ColumnProperty):
+        for column_attr in mapper.column_attrs:
+            if not isinstance(column_attr, ColumnProperty):
                 continue
-            if prop.key in exclude_column_names:
+            if column_attr.key in exclude_column_names:
                 continue
-
-            annotations[prop.key] = __declared_sqlalchemy_column_type(type_hints.get(prop.key, None))
+            mapped_type = _get_sqlalchemy_mapped_types(sqlalchemy_model=sqlalchemy_model)[column_attr.key]
+            annotations[column_attr.key] = _define_sqlalchemy_column_mapped_type(type_=mapped_type)
 
     # 2) Свойства (@property)
     if include_properties:
@@ -242,3 +239,5 @@ def pydantic_schema_from_sqlalchemy_model(
 
     return type(model_name, (base_model,), attrs)
 
+
+print(pydantic_schema_from_sqlalchemy_model(sqlalchemy_model=UserDBM))

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

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

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

@@ -275,7 +275,7 @@ arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/common.py,sha256=NsZbGGJvsn1IgfUQ9J4_6qBQyNBADDmt2Q8waRePD0c,5851
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/operation.py,sha256=6mDtD20H0bQEoSwReLPzaqNp6vLHgOQ7D1DD7KC7zIU,7111
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/story_log.py,sha256=Xou8XvEGQfLuEHH5RKdq6pXCZWMhmtyshQvvohT-Go8,2507
-arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/user.py,sha256=22TE8y2Vm9byLLkIE6RVAMa29d9ILCVg1e1QYi4ONIQ,7898
+arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/user.py,sha256=rLR4uWTS-tCX2SMVy8YrkMI-01b5kXEhaBHDbrLmmFI,7867
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/user_token.py,sha256=Yy4XZLHJqNEt_SlQOcVCDwHqxpuBwtOmyOfGKbYqKeo,1988
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/sqlalchemy_model/verification_code.py,sha256=xM5e9C8ELI1RJBAaSXZPn9Qpwl4DT2TgRvbYOxsKZsM,3137
 arpakitlib/_arpakit_project_template_v_5/project/sqlalchemy_db_/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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=MBIP8VDohDB5Z3awbhJehNNRt0xIO4s6sEBezsPHP3s,10413
+arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py,sha256=0OXyyN0f08N5k5SwZOhqZnRnkU38ReCBDIDd3qDarlY,10403
 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.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,,
+arpakitlib-1.8.260.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.260.dist-info/METADATA,sha256=8iFJMcmzoK7qqnHIMzadbyvpFOZIWiF6xdYSj9wuA2o,3919
+arpakitlib-1.8.260.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.260.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.260.dist-info/RECORD,,