openshift-python-wrapper 4.14.50__tar.gz → 4.14.51__tar.gz

{openshift_python_wrapper-4.14.50 → openshift_python_wrapper-4.14.51}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openshift-python-wrapper
-Version: 4.14.50
+Version: 4.14.51
 Summary: Wrapper around https://github.com/openshift/openshift-restclient-python
 Home-page: https://github.com/RedHatQE/openshift-python-wrapper
 License: Apache-2.0

openshift_python_wrapper-4.14.51/ocp_resources/data_source.py

@@ -0,0 +1,68 @@
+from warnings import warn
+from kubernetes.dynamic.exceptions import ResourceNotFoundError
+from ocp_resources.persistent_volume_claim import PersistentVolumeClaim
+from ocp_resources.resource import NamespacedResource
+from ocp_resources.volume_snapshot import VolumeSnapshot
+
+
+class DataSource(NamespacedResource):
+    """
+    DataSource object.
+
+    https://kubevirt.io/cdi-api-reference/main/definitions.html#_v1beta1_datasource
+    """
+
+    api_group = NamespacedResource.ApiGroup.CDI_KUBEVIRT_IO
+
+    def __init__(self, source=None, **kwargs):
+        """
+        Args:
+            source (dict): The source of the data.
+        """
+        super().__init__(**kwargs)
+        self._source = source
+
+    def to_dict(self):
+        super().to_dict()
+        if not self.yaml_file:
+            self.res.update(
+                {
+                    "spec": {
+                        "source": self.source,
+                    },
+                }
+            )
+            if not self._source:
+                raise ValueError("Passing yaml_file or parameter 'source' is required")
+
+            self.res.update({"spec": {"source": self._source}})
+
+    @property
+    def pvc(self):
+        warn("pvc will be deprecated in v4.16, Use source instead", DeprecationWarning, stacklevel=2)
+        data_source_pvc = self.instance.spec.source.pvc
+        pvc_name = data_source_pvc.name
+        pvc_namespace = data_source_pvc.namespace
+        try:
+            return PersistentVolumeClaim(
+                client=self.client,
+                name=pvc_name,
+                namespace=pvc_namespace,
+            )
+        except ResourceNotFoundError:
+            self.logger.warning(
+                f"dataSource {self.name} is pointing to a non-existing PVC, name:"
+                f" {pvc_name}, namespace: {pvc_namespace}"
+            )
+
+    @property
+    def source(self):
+        _instance_source = self.instance.spec.source
+        _source = [*_instance_source][0][0]
+        _source_mapping = {"pvc": PersistentVolumeClaim, "snapshot": VolumeSnapshot}
+
+        return _source_mapping[_source](
+            client=self.client,
+            name=_instance_source[_source].name,
+            namespace=_instance_source[_source].namespace,
+        )

{openshift_python_wrapper-4.14.50 → openshift_python_wrapper-4.14.51}/pyproject.toml

@@ -9,7 +9,7 @@ exclude = [".git", ".venv", ".mypy_cache", ".tox", "__pycache__"]
 
 [tool.poetry]
 name = "openshift-python-wrapper"
-version = "4.14.50"
+version = "4.14.51"
 description = "Wrapper around https://github.com/openshift/openshift-restclient-python"
 authors = ["Meni Yakove <myakove@gmail.com>", "Ruth Netser <rnetser@gmail.com>"]
 readme = "README.md"

openshift_python_wrapper-4.14.50/ocp_resources/data_source.py

@@ -1,59 +0,0 @@
-from openshift.dynamic.exceptions import ResourceNotFoundError
-
-from ocp_resources.constants import TIMEOUT_4MINUTES
-from ocp_resources.persistent_volume_claim import PersistentVolumeClaim
-from ocp_resources.resource import NamespacedResource
-
-
-class DataSource(NamespacedResource):
-    api_group = NamespacedResource.ApiGroup.CDI_KUBEVIRT_IO
-
-    def __init__(
-        self,
-        name=None,
-        namespace=None,
-        client=None,
-        source=None,
-        teardown=True,
-        yaml_file=None,
-        delete_timeout=TIMEOUT_4MINUTES,
-        **kwargs,
-    ):
-        super().__init__(
-            name=name,
-            namespace=namespace,
-            client=client,
-            teardown=teardown,
-            yaml_file=yaml_file,
-            delete_timeout=delete_timeout,
-            **kwargs,
-        )
-        self.source = source
-
-    def to_dict(self):
-        super().to_dict()
-        if not self.yaml_file:
-            self.res.update(
-                {
-                    "spec": {
-                        "source": self.source,
-                    },
-                }
-            )
-
-    @property
-    def pvc(self):
-        data_source_pvc = self.instance.spec.source.pvc
-        pvc_name = data_source_pvc.name
-        pvc_namespace = data_source_pvc.namespace
-        try:
-            return PersistentVolumeClaim(
-                client=self.client,
-                name=pvc_name,
-                namespace=pvc_namespace,
-            )
-        except ResourceNotFoundError:
-            self.logger.warning(
-                f"dataSource {self.name} is pointing to a non-existing PVC, name:"
-                f" {pvc_name}, namespace: {pvc_namespace}"
-            )