prefect-client 3.0.0rc3__py3-none-any.whl → 3.0.0rc4__py3-none-any.whl

prefect/blocks/kubernetes.py

@@ -1,115 +0,0 @@
-from pathlib import Path
-from typing import TYPE_CHECKING, Dict, Optional, Type
-
-import yaml
-from pydantic import Field, field_validator
-from typing_extensions import Self
-
-from prefect._internal.compatibility.deprecated import deprecated_class
-from prefect._internal.schemas.validators import validate_yaml
-from prefect.blocks.core import Block
-from prefect.utilities.collections import listrepr
-from prefect.utilities.importtools import lazy_import
-
-if TYPE_CHECKING:
-    import kubernetes
-    from kubernetes.client.api_client import ApiClient
-else:
-    kubernetes = lazy_import("kubernetes")
-
-
-@deprecated_class(
-    start_date="Mar 2024",
-    help="Use the KubernetesClusterConfig block from prefect-kubernetes instead.",
-)
-class KubernetesClusterConfig(Block):
-    """
-    Stores configuration for interaction with Kubernetes clusters.
-
-    See `from_file` for creation.
-
-    Attributes:
-        config: The entire loaded YAML contents of a kubectl config file
-        context_name: The name of the kubectl context to use
-
-    Example:
-        Load a saved Kubernetes cluster config:
-        ```python
-        from prefect.blocks.kubernetes import KubernetesClusterConfig
-
-        cluster_config_block = KubernetesClusterConfig.load("BLOCK_NAME")
-        ```
-    """
-
-    _block_type_name = "Kubernetes Cluster Config"
-    _logo_url = "https://cdn.sanity.io/images/3ugk85nk/production/2d0b896006ad463b49c28aaac14f31e00e32cfab-250x250.png"
-    _documentation_url = "https://docs.prefect.io/api-ref/prefect/blocks/kubernetes/#prefect.blocks.kubernetes.KubernetesClusterConfig"
-
-    config: Dict = Field(
-        default=..., description="The entire contents of a kubectl config file."
-    )
-    context_name: str = Field(
-        default=..., description="The name of the kubectl context to use."
-    )
-
-    @field_validator("config", mode="before")
-    @classmethod
-    def parse_yaml_config(cls, value):
-        return validate_yaml(value)
-
-    @classmethod
-    def from_file(
-        cls: Type[Self], path: Optional[Path] = None, context_name: Optional[str] = None
-    ) -> Self:
-        """
-        Create a cluster config from the a Kubernetes config file.
-
-        By default, the current context in the default Kubernetes config file will be
-        used.
-
-        An alternative file or context may be specified.
-
-        The entire config file will be loaded and stored.
-        """
-        kube_config = kubernetes.config.kube_config
-
-        path = Path(path or kube_config.KUBE_CONFIG_DEFAULT_LOCATION)
-        path = path.expanduser().resolve()
-
-        # Determine the context
-        existing_contexts, current_context = kube_config.list_kube_config_contexts(
-            config_file=str(path)
-        )
-        context_names = {ctx["name"] for ctx in existing_contexts}
-        if context_name:
-            if context_name not in context_names:
-                raise ValueError(
-                    f"Context {context_name!r} not found. "
-                    f"Specify one of: {listrepr(context_names, sep=', ')}."
-                )
-        else:
-            context_name = current_context["name"]
-
-        # Load the entire config file
-        config_file_contents = path.read_text()
-        config_dict = yaml.safe_load(config_file_contents)
-
-        return cls(config=config_dict, context_name=context_name)
-
-    def get_api_client(self) -> "ApiClient":
-        """
-        Returns a Kubernetes API client for this cluster config.
-        """
-        return kubernetes.config.kube_config.new_client_from_config_dict(
-            config_dict=self.config, context=self.context_name
-        )
-
-    def configure_client(self) -> None:
-        """
-        Activates this cluster configuration by loading the configuration into the
-        Kubernetes Python client. After calling this, Kubernetes API clients can use
-        this config's context.
-        """
-        kubernetes.config.kube_config.load_kube_config_from_dict(
-            config_dict=self.config, context=self.context_name
-        )