python-base-toolkit 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

python_base_toolkit/base_structures/base_enum.py

@@ -21,6 +21,10 @@ class EnumMixin:
     def to_dict(cls: type[E]) -> dict:
         return {member.name: member.value for member in cls}
 
+    @classmethod
+    def to_json(cls) -> str:
+        return json.dumps(cls.to_dict())
+
     @classmethod
     def names(cls: type[E]) -> list[str]:
         return [c.name for c in cls]
@@ -41,10 +45,6 @@ class EnumMixin:
     def has_value(cls: type[E], value: Any) -> bool:
         return any(member.value == value for member in cls)
 
-    @classmethod
-    def to_json(cls) -> str:
-        return json.dumps(cls.to_dict())
-
 
 class BaseEnum(EnumMixin, Enum):
     pass

python_base_toolkit/base_structures/base_pydantic_model.py

@@ -0,0 +1,38 @@
+from typing import Any, TypeVar
+from pydantic import BaseModel
+
+T = TypeVar("T", bound="PydanticMixin")
+
+
+class BasePydanticModel(BaseModel):
+    """Mixin to add Enum-like utility methods to Pydantic BaseModels."""
+
+    def __str__(self) -> str:
+        return f"{self.__class__.__name__}({self.model_dump()})"
+
+    def to_dict(self, **kwargs) -> dict[str, Any]:
+        """Wrapper for model_dump."""
+        return self.model_dump(**kwargs)
+
+    def to_json(self, **kwargs) -> str:
+        """Wrapper for model_dump_json."""
+        return self.model_dump_json(**kwargs)
+
+    def values(self) -> list[Any]:
+        """Returns a list of the current instance's field values."""
+        return list(self.model_dump().values())
+
+    @classmethod
+    def field_names(cls) -> list[str]:
+        """Returns a list of all field names defined in the model."""
+        return list(cls.model_fields.keys())
+
+    @classmethod
+    def has_field(cls, name: str) -> bool:
+        """Checks if a field name exists in the model definition."""
+        return name in cls.model_fields
+
+    @classmethod
+    def to_schema_dict(cls) -> dict[str, Any]:
+        """Returns the JSON schema of the model as a dictionary."""
+        return cls.model_json_schema()

python_base_toolkit/utils/data_serialization.py

@@ -8,11 +8,14 @@ from typing import Any
 
 import pandas as pd
 from custom_python_logger import get_logger
+from pydantic import BaseModel
 
 logger = get_logger(__name__)
 
 
 def default_serialize(obj: object) -> object:
+    if isinstance(obj, BaseModel):
+        return obj.model_dump()
     if isinstance(obj, type):
         return obj.__name__
     if is_dataclass(obj) and not isinstance(obj, type):

{python_base_toolkit-1.0.0.dist-info → python_base_toolkit-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-base-toolkit
-Version: 1.0.0
+Version: 1.0.2
 Summary: A Python package for managing pytest plugins.
 Author: Avi Zaguri
 License: MIT
@@ -22,6 +22,7 @@ Requires-Dist: custom-python-logger>=2.0.11
 Requires-Dist: pandas>=2.3.3
 Requires-Dist: pathlib>=1.0.1
 Requires-Dist: pre-commit>=4.5.0
+Requires-Dist: pydantic>=2.12.5
 Requires-Dist: pyfiglet>=1.0.4
 Requires-Dist: python-dotenv>=1.2.1
 Requires-Dist: pyyaml>=6.0.3

{python_base_toolkit-1.0.0.dist-info → python_base_toolkit-1.0.2.dist-info}/RECORD

@@ -1,6 +1,7 @@
 python_base_toolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 python_base_toolkit/base_structures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_base_toolkit/base_structures/base_enum.py,sha256=AzVRP8X6W1M2GoYQD3Cm9hpZE-uGnu9YoY9wX9WuFI8,1324
+python_base_toolkit/base_structures/base_enum.py,sha256=pnKIBvGGudkwAbbE7C2dITVdZGclXWOvvVuat3erLMo,1324
+python_base_toolkit/base_structures/base_pydantic_model.py,sha256=kS_cXzlXFxJUbMnxANqSvxiRo0hfp0575KHTw1RP2t0,1239
 python_base_toolkit/ci/task_files/default.yml,sha256=7NKbXXZP366t-K34MfRj1XJqhhm19YaRpraV1Na4c90,145
 python_base_toolkit/ci/task_files/dependencies.yml,sha256=jqdNAugfDWoiH7YDBD-5QWMX9z4tZkSGGV9PNp6wpFA,891
 python_base_toolkit/ci/task_files/deploy.yml,sha256=uKvRJjNImWGPk_xqKV08HlIhDoPepEvf5M54l7Bk_6I,537
@@ -22,7 +23,7 @@ python_base_toolkit/instances/instance_manager.py,sha256=iq72kcPlbc7-AQTuKcBOeqK
 python_base_toolkit/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 python_base_toolkit/scripts/remove_build_files.py,sha256=461h7BkuwhaCBK17fbxDc3n_nQhL5aCXm3VP52s8sW8,694
 python_base_toolkit/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_base_toolkit/utils/data_serialization.py,sha256=Xf_JTMQPAQfCDM4b0CJxHK7CjmbmVassYViDa65B97A,1197
+python_base_toolkit/utils/data_serialization.py,sha256=Wwi9DQfymNgPj3_fJDUpKoA6U-BE8cJyf1-3qLD7ngE,1295
 python_base_toolkit/utils/date_time.py,sha256=U1P_8qPnEQzI4DguHVEofYLzlyuWfK7L6VsOKPbMa7c,429
 python_base_toolkit/utils/execute.py,sha256=6QYQWd1Z15Ofb4gcAtXCx90l1D-REwTWDZNybU6OibM,1560
 python_base_toolkit/utils/file_utils.py,sha256=XGiFocSNpb77GeX_hEuR_7juNsSwA-RikpH1CvEG4TM,8793
@@ -35,8 +36,8 @@ python_base_toolkit/utils/pycache.py,sha256=X-pWRjAd2xcBtMkcyCuKz34SbHWc2leBHBIj
 python_base_toolkit/utils/random_utils.py,sha256=gTH6k1MKKGNZvBTLc5H-eSmtEGvXEdLqtzXN9aHuu-A,2730
 python_base_toolkit/utils/shorten_url.py,sha256=zsyW3wloFJdVb9s3fhR-olL6ekT03FDlFm2xbFY4viw,359
 python_base_toolkit/utils/venv_details.py,sha256=d1WlZr1LuWz2SXdFP0Hqh6VHXXXSIuXB0oTmWXJrGqM,210
-python_base_toolkit-1.0.0.dist-info/licenses/LICENSE,sha256=cSikHY6SZFsPZSBizCDAJ0-Bjjzxt-JtX6TVbKxwimo,1067
-python_base_toolkit-1.0.0.dist-info/METADATA,sha256=hTpBVqTsaFCPiy520S95ZOfGD-3IACgGqPrApuJAkfI,2963
-python_base_toolkit-1.0.0.dist-info/WHEEL,sha256=YLJXdYXQ2FQ0Uqn2J-6iEIC-3iOey8lH3xCtvFLkd8Q,91
-python_base_toolkit-1.0.0.dist-info/top_level.txt,sha256=I9kK6FGKq_6z3aGySoWamY68GRgg0xsL6sS-uyaxJ2k,20
-python_base_toolkit-1.0.0.dist-info/RECORD,,
+python_base_toolkit-1.0.2.dist-info/licenses/LICENSE,sha256=cSikHY6SZFsPZSBizCDAJ0-Bjjzxt-JtX6TVbKxwimo,1067
+python_base_toolkit-1.0.2.dist-info/METADATA,sha256=Xbb2WDCzDa-FGH6tRZA9kDm6uXEgGmMVSB3Mss1G6CU,2995
+python_base_toolkit-1.0.2.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+python_base_toolkit-1.0.2.dist-info/top_level.txt,sha256=I9kK6FGKq_6z3aGySoWamY68GRgg0xsL6sS-uyaxJ2k,20
+python_base_toolkit-1.0.2.dist-info/RECORD,,

{python_base_toolkit-1.0.0.dist-info → python_base_toolkit-1.0.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (81.0.0)
+Generator: setuptools (82.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any