python-base-toolkit 1.0.0__tar.gz → 1.0.2__tar.gz

{python_base_toolkit-1.0.0/python_base_toolkit.egg-info → python_base_toolkit-1.0.2}/PKG-INFO

@@ -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 → python_base_toolkit-1.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-base-toolkit"
-version = "1.0.0"
+version = "1.0.2"
 description = "A Python package for managing pytest plugins."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -27,6 +27,7 @@ dependencies = [
     "pandas>=2.3.3",
     "pathlib>=1.0.1",
     "pre-commit>=4.5.0",
+    "pydantic>=2.12.5",
     "pyfiglet>=1.0.4",
     "python-dotenv>=1.2.1",
     "pyyaml>=6.0.3",

python_base_toolkit-1.0.2/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-1.0.0 → python_base_toolkit-1.0.2}/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 → python_base_toolkit-1.0.2/python_base_toolkit.egg-info}/PKG-INFO

@@ -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 → python_base_toolkit-1.0.2}/python_base_toolkit.egg-info/SOURCES.txt

@@ -12,6 +12,7 @@ python_base_toolkit.egg-info/requires.txt
 python_base_toolkit.egg-info/top_level.txt
 python_base_toolkit/base_structures/__init__.py
 python_base_toolkit/base_structures/base_enum.py
+python_base_toolkit/base_structures/base_pydantic_model.py
 python_base_toolkit/ci/task_files/default.yml
 python_base_toolkit/ci/task_files/dependencies.yml
 python_base_toolkit/ci/task_files/deploy.yml

{python_base_toolkit-1.0.0 → python_base_toolkit-1.0.2}/python_base_toolkit.egg-info/requires.txt

@@ -4,6 +4,7 @@ custom-python-logger>=2.0.11
 pandas>=2.3.3
 pathlib>=1.0.1
 pre-commit>=4.5.0
+pydantic>=2.12.5
 pyfiglet>=1.0.4
 python-dotenv>=1.2.1
 pyyaml>=6.0.3

{python_base_toolkit-1.0.0 → python_base_toolkit-1.0.2}/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