arpakitlib 1.8.200__py3-none-any.whl → 1.8.202__py3-none-any.whl

arpakitlib/clone_pydantic_model_fields.py

@@ -0,0 +1,45 @@
+from typing import Any, Type
+
+from pydantic import BaseModel, create_model
+from pydantic_core import PydanticUndefined
+
+
+def clone_pydantic_model_fields(
+        *,
+        model_cls: Type[BaseModel],
+        fields_to_remove: set[str] | None = None,
+        new_class_name: str | None = None,
+) -> Type[BaseModel]:
+    if fields_to_remove is None:
+        fields_to_remove = set()
+    if new_class_name is None:
+        new_class_name = f"{model_cls.__name__}Cloned"
+
+    field_defs: dict[str, tuple[type[Any], Any]] = {}
+
+    for field_name, field_ in model_cls.model_fields.items():
+        if field_name in fields_to_remove:
+            continue
+
+        if field_.default_factory is not None and field_.default is PydanticUndefined:
+            default = field_
+        elif field_.default is not PydanticUndefined:
+            default = field_.default
+        else:
+            default = field_
+
+        field_defs[field_name] = ((field_.annotation or Any), default)
+
+    return create_model(
+        new_class_name,
+        __base__=BaseModel,
+        **field_defs,
+    )
+
+
+def __example():
+    pass
+
+
+if __name__ == '__main__':
+    __example()

{arpakitlib-1.8.200.dist-info → arpakitlib-1.8.202.dist-info}/METADATA

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

{arpakitlib-1.8.200.dist-info → arpakitlib-1.8.202.dist-info}/RECORD

@@ -413,9 +413,9 @@ arpakitlib/ar_sqlalchemy_util.py,sha256=hiDh1GrFHmnqa6lJPMq4fb9m3_fs-eDKuRQzbFxI
 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/prune_model.py,sha256=GOW6BJGd40hQo8CxaQD0PDs8-nb67aDDEufDT_4BwX4,2745
-arpakitlib-1.8.200.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.200.dist-info/METADATA,sha256=fcrNNEp3lekRKpBaKMDbcr_cokkXZYviLtaWpf74u6A,3741
-arpakitlib-1.8.200.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-arpakitlib-1.8.200.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.200.dist-info/RECORD,,
+arpakitlib/clone_pydantic_model_fields.py,sha256=k3IrYAI-lEHVSHYJHmKWiz8FNVMGS5vwmStitCGHG4E,1152
+arpakitlib-1.8.202.dist-info/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.202.dist-info/METADATA,sha256=65x0MDlwddjab8wNPxBRB9_0nEGiWMI2KLVQ-j4pB3s,3741
+arpakitlib-1.8.202.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+arpakitlib-1.8.202.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.202.dist-info/RECORD,,

arpakitlib/prune_model.py

@@ -1,79 +0,0 @@
-import copy
-from typing import Type
-
-from pydantic import BaseModel
-from pydantic.fields import FieldInfo, Field
-
-
-def prune_model(
-        *,
-        model_cls: Type[BaseModel],
-        fields_to_remove: set[str],
-        new_class_name: str,
-) -> Type[BaseModel]:
-    """
-    Создаёт новый класс-модель (Pydantic v2), который:
-      - наследуется от тех же баз, что и model_cls (в том же порядке);
-      - содержит только поля, объявленные В САМОМ model_cls (а не у его баз),
-        за вычетом fields_to_remove;
-      - копирует каждое поле целиком через deepcopy(FieldInfo), сохраняя ВСЕ атрибуты
-        (alias, constraints, repr, json_schema_extra и т.д.).
-      - НЕ копирует валидаторы, конфиг, docstring и прочее — только поля.
-
-    Наследованные от баз поля сохранятся за счёт наследования. Удалить наследуемое поле
-    таким способом нельзя — нужно менять базовые классы / MRO.
-    """
-    if not (isinstance(model_cls, type) and issubclass(model_cls, BaseModel)):
-        raise TypeError("model_cls должен быть подклассом pydantic.BaseModel (v2).")
-
-    namespace: dict = {
-        "__module__": getattr(model_cls, "__module__", "__main__"),
-        "__annotations__": {},
-    }
-
-    for name, annotation in dict(getattr(model_cls, "__annotations__", {})).items():
-        if name in fields_to_remove:
-            continue
-
-        field_info: FieldInfo | None = getattr(model_cls, "model_fields", {}).get(name)
-
-        if isinstance(field_info, FieldInfo):
-            namespace["__annotations__"][name] = annotation
-            namespace[name] = copy.deepcopy(field_info)
-
-    new_model = type(new_class_name, model_cls.__bases__, namespace)
-
-    if not issubclass(new_model, BaseModel):
-        raise RuntimeError("not issubclass(new_model, BaseModel)")
-
-    return new_model
-
-
-def __example():
-    from typing import Optional
-    from pydantic import BaseModel
-
-    class Timestamped(BaseModel):
-        created_at: int = 1
-        updated_at: int = 1
-
-        class Names:
-            arsen = "arse"
-
-    class User(Timestamped):
-        id: int = Field(default=1)
-        email: str = "asasf"
-        password_hash: str = "asasf"
-        nickname: Optional[str] = None
-
-    PublicUser = prune_model(
-        model_cls=User,
-        fields_to_remove={"id"},
-        new_class_name="PublicUser",
-    )
-
-    print(PublicUser())
-
-
-if __name__ == '__main__':
-    __example()