Source code for polywrap_client_config_builder.configures.wrapper_configure

"""This module contains the wrapper configure class for the client config builder."""
from typing import Dict, List, Union, cast

from polywrap_core import Uri, Wrapper

from ..types import BuilderConfig, ClientConfigBuilder


[docs]class WrapperConfigure:
    """Allows configuring the wrappers."""

    config: BuilderConfig

[docs]    def get_wrapper(self, uri: Uri) -> Union[Wrapper, None]:
        """Return the set wrapper for the given uri."""
        return self.config.wrappers.get(uri)

[docs]    def get_wrappers(self) -> Dict[Uri, Wrapper]:
        """Return the wrappers from the builder's config."""
        return self.config.wrappers

[docs]    def set_wrapper(self, uri: Uri, wrapper: Wrapper) -> ClientConfigBuilder:
        """Set the wrapper by uri in the builder's config, overiding any existing values."""
        self.config.wrappers[uri] = wrapper
        return cast(ClientConfigBuilder, self)

[docs]    def set_wrappers(self, uri_wrappers: Dict[Uri, Wrapper]) -> ClientConfigBuilder:
        """Set the wrappers in the builder's config, overiding any existing values."""
        self.config.wrappers.update(uri_wrappers)
        return cast(ClientConfigBuilder, self)

[docs]    def remove_wrapper(self, uri: Uri) -> ClientConfigBuilder:
        """Remove the wrapper for the given uri."""
        self.config.wrappers.pop(uri, None)
        return cast(ClientConfigBuilder, self)

[docs]    def remove_wrappers(self, uris: List[Uri]) -> ClientConfigBuilder:
        """Remove the wrappers for the given uris."""
        for uri in uris:
            self.remove_wrapper(uri)
        return cast(ClientConfigBuilder, self)