acryl-datahub 0.15.0rc25__py3-none-any.whl → 0.15.0.1__py3-none-any.whl

datahub/specific/custom_properties.py

@@ -1,37 +0,0 @@
-from typing import Generic, TypeVar
-
-from datahub.emitter.mcp_patch_builder import MetadataPatchProposal
-
-_Parent = TypeVar("_Parent", bound=MetadataPatchProposal)
-
-
-class CustomPropertiesPatchHelper(Generic[_Parent]):
-    def __init__(
-        self,
-        parent: _Parent,
-        aspect_name: str,
-    ) -> None:
-        self.aspect_name = aspect_name
-        self._parent = parent
-        self.aspect_field = "customProperties"
-
-    def parent(self) -> _Parent:
-        return self._parent
-
-    def add_property(self, key: str, value: str) -> "CustomPropertiesPatchHelper":
-        self._parent._add_patch(
-            self.aspect_name,
-            "add",
-            path=f"/{self.aspect_field}/{key}",
-            value=value,
-        )
-        return self
-
-    def remove_property(self, key: str) -> "CustomPropertiesPatchHelper":
-        self._parent._add_patch(
-            self.aspect_name,
-            "remove",
-            path=f"/{self.aspect_field}/{key}",
-            value={},
-        )
-        return self

datahub/specific/ownership.py

@@ -1,48 +0,0 @@
-from typing import Generic, List, Optional, TypeVar
-
-from datahub.emitter.mcp_patch_builder import MetadataPatchProposal
-from datahub.metadata.schema_classes import (
-    OwnerClass,
-    OwnershipClass,
-    OwnershipTypeClass,
-)
-
-_Parent = TypeVar("_Parent", bound=MetadataPatchProposal)
-
-
-class OwnershipPatchHelper(Generic[_Parent]):
-    def __init__(self, parent: _Parent) -> None:
-        self._parent = parent
-        self.aspect_field = OwnershipClass.ASPECT_NAME
-
-    def parent(self) -> _Parent:
-        return self._parent
-
-    def add_owner(self, owner: OwnerClass) -> "OwnershipPatchHelper":
-        self._parent._add_patch(
-            OwnershipClass.ASPECT_NAME,
-            "add",
-            path=f"/owners/{owner.owner}/{owner.type}",
-            value=owner,
-        )
-        return self
-
-    def remove_owner(
-        self, owner: str, owner_type: Optional[OwnershipTypeClass] = None
-    ) -> "OwnershipPatchHelper":
-        """
-        param: owner_type is optional
-        """
-        self._parent._add_patch(
-            OwnershipClass.ASPECT_NAME,
-            "remove",
-            path=f"/owners/{owner}" + (f"/{owner_type}" if owner_type else ""),
-            value=owner,
-        )
-        return self
-
-    def set_owners(self, owners: List[OwnerClass]) -> "OwnershipPatchHelper":
-        self._parent._add_patch(
-            OwnershipClass.ASPECT_NAME, "add", path="/owners", value=owners
-        )
-        return self

datahub/specific/structured_properties.py

@@ -1,53 +0,0 @@
-from typing import Generic, List, TypeVar, Union
-
-from datahub.emitter.mcp_patch_builder import MetadataPatchProposal
-from datahub.metadata.schema_classes import StructuredPropertyValueAssignmentClass
-from datahub.utilities.urns.structured_properties_urn import (
-    make_structured_property_urn,
-)
-
-_Parent = TypeVar("_Parent", bound=MetadataPatchProposal)
-
-
-class StructuredPropertiesPatchHelper(Generic[_Parent]):
-    def __init__(
-        self,
-        parent: _Parent,
-        aspect_name: str = "structuredProperties",
-    ) -> None:
-        self.aspect_name = aspect_name
-        self._parent = parent
-        self.aspect_field = "properties"
-
-    def parent(self) -> _Parent:
-        return self._parent
-
-    def set_property(
-        self, key: str, value: Union[str, float, List[Union[str, float]]]
-    ) -> "StructuredPropertiesPatchHelper":
-        self.remove_property(key)
-        self.add_property(key, value)
-        return self
-
-    def remove_property(self, key: str) -> "StructuredPropertiesPatchHelper":
-        self._parent._add_patch(
-            self.aspect_name,
-            "remove",
-            path=(self.aspect_field, make_structured_property_urn(key)),
-            value={},
-        )
-        return self
-
-    def add_property(
-        self, key: str, value: Union[str, float, List[Union[str, float]]]
-    ) -> "StructuredPropertiesPatchHelper":
-        self._parent._add_patch(
-            self.aspect_name,
-            "add",
-            path=(self.aspect_field, make_structured_property_urn(key)),
-            value=StructuredPropertyValueAssignmentClass(
-                propertyUrn=make_structured_property_urn(key),
-                values=value if isinstance(value, list) else [value],
-            ),
-        )
-        return self