arpakitlib 1.8.251__py3-none-any.whl → 1.8.252__py3-none-any.whl

arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py

@@ -44,7 +44,7 @@ def _python_type_from_col(col) -> type | str:
         return Any
 
 
-def _collect_properties_with_types(model_class: type) -> dict[str, Any]:
+def _get_property_name_to_type_from_model_class(model_class: type) -> dict[str, Any]:
     """
     Находит все @property в классе и вытаскивает их возвращаемый тип.
     Если тип не удаётся получить — подставляем Any.
@@ -64,13 +64,15 @@ def _collect_properties_with_types(model_class: type) -> dict[str, Any]:
 def pydantic_schema_from_sqlalchemy_model(
         sqlalchemy_model: type,
         *,
-        name: str | None = None,
+        model_name: str | None = None,
         base_model: type[BaseModel] = BaseModel,
         include_column_defaults: bool = False,
         exclude_column_names: list[str] | None = None,
         include_properties: bool = False,
         include_property_names: list[str] | None = None,
         exclude_property_names: list[str] | None = None,
+        filter_property_prefixes: list[str] | None = None,
+        remove_prefixes: list[str] | None = None,
 ) -> type[BaseModel]:
     """
     Генерирует Pydantic-модель из колонок SQLAlchemy-модели и (опционально) из @property.
@@ -83,14 +85,16 @@ def pydantic_schema_from_sqlalchemy_model(
     - exclude_property_names: blacklist имён свойств (исключаются после whitelist'а).
     """
     mapper = inspect(sqlalchemy_model).mapper
-    model_name = name or f"{sqlalchemy_model.__name__}Schema"
+    model_name = model_name or f"{sqlalchemy_model.__name__}Schema"
 
     annotations: dict[str, Any] = {}
     attrs: dict[str, Any] = {}
 
-    exclude_column_names = set(exclude_column_names or [])
-    include_property_names = set(include_property_names or [])
-    exclude_property_names = set(exclude_property_names or [])
+    exclude_column_names = set(set(exclude_column_names) or [])
+    include_property_names = set(set(include_property_names) or [])
+    exclude_property_names = set(set(exclude_property_names) or [])
+    filter_property_prefixes = list(set(filter_property_prefixes) or [])
+    remove_prefixes = list(set(remove_prefixes) or [])
 
     # 1) Колонки
     for prop in mapper.attrs:
@@ -121,9 +125,17 @@ def pydantic_schema_from_sqlalchemy_model(
 
     # 2) Свойства (@property)
     if include_properties:
-        property_name_to_type = _collect_properties_with_types(sqlalchemy_model)
+        property_name_to_type = _get_property_name_to_type_from_model_class(sqlalchemy_model)
 
-        # whitelist (если задан)
+        # фильтр по префиксам (если задан и не пуст)
+        if filter_property_prefixes:
+            property_name_to_type = {
+                property_name: property_type
+                for property_name, property_type in property_name_to_type.items()
+                if any(property_name.startswith(v) for v in filter_property_prefixes)
+            }
+
+        # whitelist по именам
         if include_property_names:
             property_name_to_type = {
                 k: v
@@ -133,13 +145,38 @@ def pydantic_schema_from_sqlalchemy_model(
         else:
             property_name_to_type = dict(property_name_to_type)
 
-        # blacklist
+        # blacklist по именам
         if exclude_property_names:
             for property_name in list(property_name_to_type.keys()):
                 if property_name in exclude_property_names:
-                    property_name_to_type.pop(name, None)
-
-        # Добавляем аннотации свойств (колонки имеют приоритет)
+                    property_name_to_type.pop(property_name, None)
+
+        # удаляем префиксы
+        if remove_prefixes:
+            renamed_property_name_to_type = {}
+            for property_name, property_type in property_name_to_type.items():
+                new_property_name = property_name
+                for prefix in remove_prefixes:
+                    if new_property_name.startswith(prefix):
+                        new_property_name = new_property_name[len(prefix):].strip()
+                        break
+                if new_property_name != property_name:
+                    property_name_to_type.pop(property_name)
+                    # избегаем коллизий
+                    if (
+                            new_property_name not in annotations
+                            and new_property_name not in property_name_to_type
+                            and new_property_name not in renamed_property_name_to_type
+                    ):
+                        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}'"
+                        )
+            property_name_to_type.update(renamed_property_name_to_type)
+
+        # добавляем (колонки в приоритете)
         for property_name, property_type in property_name_to_type.items():
             if property_name in annotations:
                 continue

arpakitlib/ar_sqlalchemy_util.py

@@ -174,6 +174,7 @@ class BaseDBM(DeclarativeBase):
             exclude_sd_properties: Collection[str] | None = None,
             include_columns_and_sd_properties: Collection[str] | None = None,
             prefix: str = "sdp_",
+            remove_prefix: bool = True,
             kwargs: dict[str, Any] | None = None
     ) -> dict[str, Any]:
         if exclude_columns is None:
@@ -199,7 +200,9 @@ class BaseDBM(DeclarativeBase):
                 if not attr_name.startswith(prefix) or not isinstance(getattr(type(self), attr_name, None), property):
                     continue
 
-                sd_property_name = attr_name.removeprefix(prefix)
+                sd_property_name = attr_name
+                if remove_prefix:
+                    sd_property_name = sd_property_name.removeprefix(prefix)
 
                 if (
                         include_columns_and_sd_properties is not None

{arpakitlib-1.8.251.dist-info → arpakitlib-1.8.252.dist-info}/METADATA

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

{arpakitlib-1.8.251.dist-info → arpakitlib-1.8.252.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=NuvydQe7Cih6dW5H53GB6BsG2oFKgf27F3iJxuPO2J0,5364
+arpakitlib/ar_pydantic_schema_from_sqlalchemy_model.py,sha256=Dy53opHf_WjAMUC_PpvLkYXnrKOlqhbuNeEqlwIWJ8E,7369
 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
@@ -426,12 +426,12 @@ arpakitlib/ar_schedule_uust_api_client_util.py,sha256=rXI2_3OTaIBgR-GixM1Ti-Ue1f
 arpakitlib/ar_settings_util.py,sha256=Y5wi_cmsjDjfJpM0VJHjbo0NoVPKfypKaD1USowwDtQ,1327
 arpakitlib/ar_sleep_util.py,sha256=ggaj7ML6QK_ADsHMcyu6GUmUpQ_9B9n-SKYH17h-9lM,1045
 arpakitlib/ar_sqladmin_util.py,sha256=SEoaowAPF3lhxPsNjwmOymNJ55Ty9rmzvsDm7gD5Ceo,861
-arpakitlib/ar_sqlalchemy_util.py,sha256=5I7KsKCzJc9hxpn2_0r__RcbqcKpTs_PDw_f5_0P7Xw,16050
+arpakitlib/ar_sqlalchemy_util.py,sha256=FDva9onjtCPrYZYIHHb93NMwD1WlmaORjiWgPRJQKoU,16180
 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.251.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.251.dist-info/METADATA,sha256=qCRaVdN1qqbRnIqYG7bYL2J3i6_sDOIpxqO7SG1BLNA,3919
-arpakitlib-1.8.251.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.251.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.251.dist-info/RECORD,,
+arpakitlib-1.8.252.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.252.dist-info/METADATA,sha256=y2bDBx7FFGVOBgDvKBp20DIzFNfjhHBMqJJ5G9HUXf4,3919
+arpakitlib-1.8.252.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.252.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.252.dist-info/RECORD,,