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

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

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

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.1 → abs_utils-0.2.2}/pyproject.toml

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