abs-utils 0.2.0__tar.gz → 0.2.2__tar.gz

{abs_utils-0.2.0 → abs_utils-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: abs-utils
-Version: 0.2.0
+Version: 0.2.2
 Summary: AutoBridge Systems Utility Library
 Author: AutoBridgeSystems
 Author-email: info@autobridgesystems.com

{abs_utils-0.2.0 → abs_utils-0.2.2}/abs_utils/schemas/fields_schema/common_schema.py

@@ -77,6 +77,7 @@ class FieldType(str, Enum):
     LIST = "list"
     ASSOCIATION = "association"
 
+
 class BaseFieldSchema(BaseModel):
     """Base schema for field definitions."""
     name: str = Field(description="Name of the field")

abs_utils-0.2.2/abs_utils/schemas/libs/extend_enum.py

@@ -0,0 +1,28 @@
+from enum import Enum
+from typing import Type, Union, Dict
+
+
+def extend_enum(
+    name: str,
+    base_enum: Type[Enum],
+    additional_fields: Union[Dict[str, str], Type[Enum]]
+) -> Type[Enum]:
+    """
+    Create a new Enum by extending an existing one with additional fields
+    provided either as a dict or another Enum class.
+
+    :param name: Name of the new Enum
+    :param base_enum: The base Enum class to extend
+    :param additional_fields: A dict or Enum class with new values
+    :return: A new Enum class with combined values
+    """
+    base_members = {e.name: e.value for e in base_enum}
+    extra_members = {}
+    if isinstance(additional_fields, dict):
+        extra_members = additional_fields
+    elif isinstance(additional_fields, type) and issubclass(additional_fields, Enum):
+        extra_members = {e.name: e.value for e in additional_fields}
+
+    merged_members = {**base_members, **extra_members}
+
+    return Enum(name, merged_members, type=str)

abs_utils-0.2.2/abs_utils/schemas/libs/make_optional.py

@@ -0,0 +1,45 @@
+from typing import Optional, Set, Type
+
+from pydantic import BaseModel, Field, create_model
+
+
+def make_optional(
+    model: Type[BaseModel], exclude_fields: Set[str] = None
+) -> Type[BaseModel]:
+    """Make all fields of a Pydantic model optional while preserving validation constraints, and allow field exclusion."""
+    fields = {}
+    exclude_fields = exclude_fields or set()
+
+    model_schema = model.model_json_schema()
+    properties = model_schema.get("properties", {})
+
+    for name, field in model.model_fields.items():
+        if name in exclude_fields:
+            continue  # Skip excluded fields
+
+        field_schema = properties.get(name, {})
+        constraints = {}
+
+        if "minimum" in field_schema:
+            constraints["ge"] = field_schema["minimum"]
+        if "maximum" in field_schema:
+            constraints["le"] = field_schema["maximum"]
+        if "exclusiveMinimum" in field_schema:
+            constraints["gt"] = field_schema["exclusiveMinimum"]
+        if "exclusiveMaximum" in field_schema:
+            constraints["lt"] = field_schema["exclusiveMaximum"]
+        if "minLength" in field_schema:
+            constraints["min_length"] = field_schema["minLength"]
+        if "maxLength" in field_schema:
+            constraints["max_length"] = field_schema["maxLength"]
+        if "pattern" in field_schema:
+            constraints["pattern"] = field_schema["pattern"]
+        if "description" in field_schema:
+            constraints["description"] = field_schema["description"]
+
+        if "error_messages" in field_schema:
+            constraints["error_messages"] = field_schema["error_messages"]
+
+        fields[name] = (Optional[field.annotation], Field(default=None, **constraints))
+
+    return create_model(f"Optional{model.__name__}", **fields)

{abs_utils-0.2.0 → abs_utils-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "abs-utils"
-version = "0.2.0"
+version = "0.2.2"
 description = "AutoBridge Systems Utility Library"
 authors = [
     {name = "AutoBridgeSystems", email = "info@autobridgesystems.com"}