databricks-sdk 0.0.7__py3-none-any.whl → 0.1.1__py3-none-any.whl

databricks/sdk/service/ipaccesslists.py

@@ -1,340 +0,0 @@
-# Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
-
-import logging
-from dataclasses import dataclass
-from enum import Enum
-from typing import Dict, Iterator, List
-
-from ._internal import _enum, _from_dict, _repeated
-
-_LOG = logging.getLogger('databricks.sdk')
-
-# all definitions in this file are in alphabetical order
-
-
-@dataclass
-class CreateIpAccessList:
-    label: str
-    list_type: 'ListType'
-    ip_addresses: 'List[str]'
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.ip_addresses: body['ip_addresses'] = [v for v in self.ip_addresses]
-        if self.label: body['label'] = self.label
-        if self.list_type: body['list_type'] = self.list_type.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'CreateIpAccessList':
-        return cls(ip_addresses=d.get('ip_addresses', None),
-                   label=d.get('label', None),
-                   list_type=_enum(d, 'list_type', ListType))
-
-
-@dataclass
-class CreateIpAccessListResponse:
-    ip_access_list: 'IpAccessListInfo' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.ip_access_list: body['ip_access_list'] = self.ip_access_list.as_dict()
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'CreateIpAccessListResponse':
-        return cls(ip_access_list=_from_dict(d, 'ip_access_list', IpAccessListInfo))
-
-
-@dataclass
-class Delete:
-    """Delete access list"""
-
-    ip_access_list_id: str
-
-
-@dataclass
-class FetchIpAccessListResponse:
-    ip_access_list: 'IpAccessListInfo' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.ip_access_list: body['ip_access_list'] = self.ip_access_list.as_dict()
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'FetchIpAccessListResponse':
-        return cls(ip_access_list=_from_dict(d, 'ip_access_list', IpAccessListInfo))
-
-
-@dataclass
-class Get:
-    """Get access list"""
-
-    ip_access_list_id: str
-
-
-@dataclass
-class GetIpAccessListResponse:
-    ip_access_lists: 'List[IpAccessListInfo]' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.ip_access_lists: body['ip_access_lists'] = [v.as_dict() for v in self.ip_access_lists]
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'GetIpAccessListResponse':
-        return cls(ip_access_lists=_repeated(d, 'ip_access_lists', IpAccessListInfo))
-
-
-@dataclass
-class IpAccessListInfo:
-    address_count: int = None
-    created_at: int = None
-    created_by: int = None
-    enabled: bool = None
-    ip_addresses: 'List[str]' = None
-    label: str = None
-    list_id: str = None
-    list_type: 'ListType' = None
-    updated_at: int = None
-    updated_by: int = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.address_count: body['address_count'] = self.address_count
-        if self.created_at: body['created_at'] = self.created_at
-        if self.created_by: body['created_by'] = self.created_by
-        if self.enabled: body['enabled'] = self.enabled
-        if self.ip_addresses: body['ip_addresses'] = [v for v in self.ip_addresses]
-        if self.label: body['label'] = self.label
-        if self.list_id: body['list_id'] = self.list_id
-        if self.list_type: body['list_type'] = self.list_type.value
-        if self.updated_at: body['updated_at'] = self.updated_at
-        if self.updated_by: body['updated_by'] = self.updated_by
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'IpAccessListInfo':
-        return cls(address_count=d.get('address_count', None),
-                   created_at=d.get('created_at', None),
-                   created_by=d.get('created_by', None),
-                   enabled=d.get('enabled', None),
-                   ip_addresses=d.get('ip_addresses', None),
-                   label=d.get('label', None),
-                   list_id=d.get('list_id', None),
-                   list_type=_enum(d, 'list_type', ListType),
-                   updated_at=d.get('updated_at', None),
-                   updated_by=d.get('updated_by', None))
-
-
-class ListType(Enum):
-    """This describes an enum"""
-
-    ALLOW = 'ALLOW'
-    BLOCK = 'BLOCK'
-
-
-@dataclass
-class ReplaceIpAccessList:
-    label: str
-    list_type: 'ListType'
-    ip_addresses: 'List[str]'
-    enabled: bool
-    ip_access_list_id: str
-    list_id: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.enabled: body['enabled'] = self.enabled
-        if self.ip_access_list_id: body['ip_access_list_id'] = self.ip_access_list_id
-        if self.ip_addresses: body['ip_addresses'] = [v for v in self.ip_addresses]
-        if self.label: body['label'] = self.label
-        if self.list_id: body['list_id'] = self.list_id
-        if self.list_type: body['list_type'] = self.list_type.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'ReplaceIpAccessList':
-        return cls(enabled=d.get('enabled', None),
-                   ip_access_list_id=d.get('ip_access_list_id', None),
-                   ip_addresses=d.get('ip_addresses', None),
-                   label=d.get('label', None),
-                   list_id=d.get('list_id', None),
-                   list_type=_enum(d, 'list_type', ListType))
-
-
-@dataclass
-class UpdateIpAccessList:
-    label: str
-    list_type: 'ListType'
-    ip_addresses: 'List[str]'
-    enabled: bool
-    ip_access_list_id: str
-    list_id: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.enabled: body['enabled'] = self.enabled
-        if self.ip_access_list_id: body['ip_access_list_id'] = self.ip_access_list_id
-        if self.ip_addresses: body['ip_addresses'] = [v for v in self.ip_addresses]
-        if self.label: body['label'] = self.label
-        if self.list_id: body['list_id'] = self.list_id
-        if self.list_type: body['list_type'] = self.list_type.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'UpdateIpAccessList':
-        return cls(enabled=d.get('enabled', None),
-                   ip_access_list_id=d.get('ip_access_list_id', None),
-                   ip_addresses=d.get('ip_addresses', None),
-                   label=d.get('label', None),
-                   list_id=d.get('list_id', None),
-                   list_type=_enum(d, 'list_type', ListType))
-
-
-class IpAccessListsAPI:
-    """IP Access List enables admins to configure IP access lists.
-    
-    IP access lists affect web application access and REST API access to this workspace only. If the feature
-    is disabled for a workspace, all access is allowed for this workspace. There is support for allow lists
-    (inclusion) and block lists (exclusion).
-    
-    When a connection is attempted: 1. **First, all block lists are checked.** If the connection IP address
-    matches any block list, the connection is rejected. 2. **If the connection was not rejected by block
-    lists**, the IP address is compared with the allow lists.
-    
-    If there is at least one allow list for the workspace, the connection is allowed only if the IP address
-    matches an allow list. If there are no allow lists for the workspace, all IP addresses are allowed.
-    
-    For all allow lists and block lists combined, the workspace supports a maximum of 1000 IP/CIDR values,
-    where one CIDR counts as a single value.
-    
-    After changes to the IP access list feature, it can take a few minutes for changes to take effect."""
-
-    def __init__(self, api_client):
-        self._api = api_client
-
-    def create(self, label: str, list_type: ListType, ip_addresses: List[str],
-               **kwargs) -> CreateIpAccessListResponse:
-        """Create access list.
-        
-        Creates an IP access list for this workspace.
-        
-        A list can be an allow list or a block list. See the top of this file for a description of how the
-        server treats allow lists and block lists at runtime.
-        
-        When creating or updating an IP access list:
-        
-        * For all allow lists and block lists combined, the API supports a maximum of 1000 IP/CIDR values,
-        where one CIDR counts as a single value. Attempts to exceed that number return error 400 with
-        `error_code` value `QUOTA_EXCEEDED`. * If the new list would block the calling user's current IP,
-        error 400 is returned with `error_code` value `INVALID_STATE`.
-        
-        It can take a few minutes for the changes to take effect. **Note**: Your new IP access list has no
-        effect until you enable the feature. See :method:workspaceconf/setStatus"""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = CreateIpAccessList(ip_addresses=ip_addresses, label=label, list_type=list_type)
-        body = request.as_dict()
-
-        json = self._api.do('POST', '/api/2.0/ip-access-lists', body=body)
-        return CreateIpAccessListResponse.from_dict(json)
-
-    def delete(self, ip_access_list_id: str, **kwargs):
-        """Delete access list.
-        
-        Deletes an IP access list, specified by its list ID."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = Delete(ip_access_list_id=ip_access_list_id)
-
-        self._api.do('DELETE', f'/api/2.0/ip-access-lists/{request.ip_access_list_id}')
-
-    def get(self, ip_access_list_id: str, **kwargs) -> FetchIpAccessListResponse:
-        """Get access list.
-        
-        Gets an IP access list, specified by its list ID."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = Get(ip_access_list_id=ip_access_list_id)
-
-        json = self._api.do('GET', f'/api/2.0/ip-access-lists/{request.ip_access_list_id}')
-        return FetchIpAccessListResponse.from_dict(json)
-
-    def list(self) -> Iterator[IpAccessListInfo]:
-        """Get access lists.
-        
-        Gets all IP access lists for the specified workspace."""
-
-        json = self._api.do('GET', '/api/2.0/ip-access-lists')
-        return [IpAccessListInfo.from_dict(v) for v in json.get('ip_access_lists', [])]
-
-    def replace(self,
-                label: str,
-                list_type: ListType,
-                ip_addresses: List[str],
-                enabled: bool,
-                ip_access_list_id: str,
-                *,
-                list_id: str = None,
-                **kwargs):
-        """Replace access list.
-        
-        Replaces an IP access list, specified by its ID.
-        
-        A list can include allow lists and block lists. See the top of this file for a description of how the
-        server treats allow lists and block lists at run time. When replacing an IP access list: * For all
-        allow lists and block lists combined, the API supports a maximum of 1000 IP/CIDR values, where one
-        CIDR counts as a single value. Attempts to exceed that number return error 400 with `error_code` value
-        `QUOTA_EXCEEDED`. * If the resulting list would block the calling user's current IP, error 400 is
-        returned with `error_code` value `INVALID_STATE`. It can take a few minutes for the changes to take
-        effect. Note that your resulting IP access list has no effect until you enable the feature. See
-        :method:workspaceconf/setStatus."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = ReplaceIpAccessList(enabled=enabled,
-                                          ip_access_list_id=ip_access_list_id,
-                                          ip_addresses=ip_addresses,
-                                          label=label,
-                                          list_id=list_id,
-                                          list_type=list_type)
-        body = request.as_dict()
-        self._api.do('PUT', f'/api/2.0/ip-access-lists/{request.ip_access_list_id}', body=body)
-
-    def update(self,
-               label: str,
-               list_type: ListType,
-               ip_addresses: List[str],
-               enabled: bool,
-               ip_access_list_id: str,
-               *,
-               list_id: str = None,
-               **kwargs):
-        """Update access list.
-        
-        Updates an existing IP access list, specified by its ID.
-        
-        A list can include allow lists and block lists. See the top of this file for a description of how the
-        server treats allow lists and block lists at run time.
-        
-        When updating an IP access list:
-        
-        * For all allow lists and block lists combined, the API supports a maximum of 1000 IP/CIDR values,
-        where one CIDR counts as a single value. Attempts to exceed that number return error 400 with
-        `error_code` value `QUOTA_EXCEEDED`. * If the updated list would block the calling user's current IP,
-        error 400 is returned with `error_code` value `INVALID_STATE`.
-        
-        It can take a few minutes for the changes to take effect. Note that your resulting IP access list has
-        no effect until you enable the feature. See :method:workspaceconf/setStatus."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = UpdateIpAccessList(enabled=enabled,
-                                         ip_access_list_id=ip_access_list_id,
-                                         ip_addresses=ip_addresses,
-                                         label=label,
-                                         list_id=list_id,
-                                         list_type=list_type)
-        body = request.as_dict()
-        self._api.do('PATCH', f'/api/2.0/ip-access-lists/{request.ip_access_list_id}', body=body)

databricks/sdk/service/libraries.py

@@ -1,282 +0,0 @@
-# Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
-
-import logging
-from dataclasses import dataclass
-from enum import Enum
-from typing import Dict, List
-
-from ._internal import _enum, _from_dict, _repeated
-
-_LOG = logging.getLogger('databricks.sdk')
-
-# all definitions in this file are in alphabetical order
-
-
-@dataclass
-class ClusterLibraryStatuses:
-    cluster_id: str = None
-    library_statuses: 'List[LibraryFullStatus]' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.cluster_id: body['cluster_id'] = self.cluster_id
-        if self.library_statuses: body['library_statuses'] = [v.as_dict() for v in self.library_statuses]
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'ClusterLibraryStatuses':
-        return cls(cluster_id=d.get('cluster_id', None),
-                   library_statuses=_repeated(d, 'library_statuses', LibraryFullStatus))
-
-
-@dataclass
-class ClusterStatus:
-    """Get status"""
-
-    cluster_id: str
-
-
-@dataclass
-class InstallLibraries:
-    cluster_id: str
-    libraries: 'List[Library]'
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.cluster_id: body['cluster_id'] = self.cluster_id
-        if self.libraries: body['libraries'] = [v.as_dict() for v in self.libraries]
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'InstallLibraries':
-        return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated(d, 'libraries', Library))
-
-
-@dataclass
-class Library:
-    cran: 'RCranLibrary' = None
-    egg: str = None
-    jar: str = None
-    maven: 'MavenLibrary' = None
-    pypi: 'PythonPyPiLibrary' = None
-    whl: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.cran: body['cran'] = self.cran.as_dict()
-        if self.egg: body['egg'] = self.egg
-        if self.jar: body['jar'] = self.jar
-        if self.maven: body['maven'] = self.maven.as_dict()
-        if self.pypi: body['pypi'] = self.pypi.as_dict()
-        if self.whl: body['whl'] = self.whl
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'Library':
-        return cls(cran=_from_dict(d, 'cran', RCranLibrary),
-                   egg=d.get('egg', None),
-                   jar=d.get('jar', None),
-                   maven=_from_dict(d, 'maven', MavenLibrary),
-                   pypi=_from_dict(d, 'pypi', PythonPyPiLibrary),
-                   whl=d.get('whl', None))
-
-
-@dataclass
-class LibraryFullStatus:
-    is_library_for_all_clusters: bool = None
-    library: 'Library' = None
-    messages: 'List[str]' = None
-    status: 'LibraryFullStatusStatus' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.is_library_for_all_clusters:
-            body['is_library_for_all_clusters'] = self.is_library_for_all_clusters
-        if self.library: body['library'] = self.library.as_dict()
-        if self.messages: body['messages'] = [v for v in self.messages]
-        if self.status: body['status'] = self.status.value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'LibraryFullStatus':
-        return cls(is_library_for_all_clusters=d.get('is_library_for_all_clusters', None),
-                   library=_from_dict(d, 'library', Library),
-                   messages=d.get('messages', None),
-                   status=_enum(d, 'status', LibraryFullStatusStatus))
-
-
-class LibraryFullStatusStatus(Enum):
-    """Status of installing the library on the cluster."""
-
-    FAILED = 'FAILED'
-    INSTALLED = 'INSTALLED'
-    INSTALLING = 'INSTALLING'
-    PENDING = 'PENDING'
-    RESOLVING = 'RESOLVING'
-    SKIPPED = 'SKIPPED'
-    UNINSTALL_ON_RESTART = 'UNINSTALL_ON_RESTART'
-
-
-@dataclass
-class ListAllClusterLibraryStatusesResponse:
-    statuses: 'List[ClusterLibraryStatuses]' = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.statuses: body['statuses'] = [v.as_dict() for v in self.statuses]
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'ListAllClusterLibraryStatusesResponse':
-        return cls(statuses=_repeated(d, 'statuses', ClusterLibraryStatuses))
-
-
-@dataclass
-class MavenLibrary:
-    coordinates: str
-    exclusions: 'List[str]' = None
-    repo: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.coordinates: body['coordinates'] = self.coordinates
-        if self.exclusions: body['exclusions'] = [v for v in self.exclusions]
-        if self.repo: body['repo'] = self.repo
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'MavenLibrary':
-        return cls(coordinates=d.get('coordinates', None),
-                   exclusions=d.get('exclusions', None),
-                   repo=d.get('repo', None))
-
-
-@dataclass
-class PythonPyPiLibrary:
-    package: str
-    repo: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.package: body['package'] = self.package
-        if self.repo: body['repo'] = self.repo
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'PythonPyPiLibrary':
-        return cls(package=d.get('package', None), repo=d.get('repo', None))
-
-
-@dataclass
-class RCranLibrary:
-    package: str
-    repo: str = None
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.package: body['package'] = self.package
-        if self.repo: body['repo'] = self.repo
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'RCranLibrary':
-        return cls(package=d.get('package', None), repo=d.get('repo', None))
-
-
-@dataclass
-class UninstallLibraries:
-    cluster_id: str
-    libraries: 'List[Library]'
-
-    def as_dict(self) -> dict:
-        body = {}
-        if self.cluster_id: body['cluster_id'] = self.cluster_id
-        if self.libraries: body['libraries'] = [v.as_dict() for v in self.libraries]
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> 'UninstallLibraries':
-        return cls(cluster_id=d.get('cluster_id', None), libraries=_repeated(d, 'libraries', Library))
-
-
-class LibrariesAPI:
-    """The Libraries API allows you to install and uninstall libraries and get the status of libraries on a
-    cluster.
-    
-    To make third-party or custom code available to notebooks and jobs running on your clusters, you can
-    install a library. Libraries can be written in Python, Java, Scala, and R. You can upload Java, Scala, and
-    Python libraries and point to external packages in PyPI, Maven, and CRAN repositories.
-    
-    Cluster libraries can be used by all notebooks running on a cluster. You can install a cluster library
-    directly from a public repository such as PyPI or Maven, using a previously installed workspace library,
-    or using an init script.
-    
-    When you install a library on a cluster, a notebook already attached to that cluster will not immediately
-    see the new library. You must first detach and then reattach the notebook to the cluster.
-    
-    When you uninstall a library from a cluster, the library is removed only when you restart the cluster.
-    Until you restart the cluster, the status of the uninstalled library appears as Uninstall pending restart."""
-
-    def __init__(self, api_client):
-        self._api = api_client
-
-    def all_cluster_statuses(self) -> ListAllClusterLibraryStatusesResponse:
-        """Get all statuses.
-        
-        Get the status of all libraries on all clusters. A status will be available for all libraries
-        installed on this cluster via the API or the libraries UI as well as libraries set to be installed on
-        all clusters via the libraries UI."""
-
-        json = self._api.do('GET', '/api/2.0/libraries/all-cluster-statuses')
-        return ListAllClusterLibraryStatusesResponse.from_dict(json)
-
-    def cluster_status(self, cluster_id: str, **kwargs) -> ClusterLibraryStatuses:
-        """Get status.
-        
-        Get the status of libraries on a cluster. A status will be available for all libraries installed on
-        this cluster via the API or the libraries UI as well as libraries set to be installed on all clusters
-        via the libraries UI. The order of returned libraries will be as follows.
-        
-        1. Libraries set to be installed on this cluster will be returned first. Within this group, the final
-        order will be order in which the libraries were added to the cluster.
-        
-        2. Libraries set to be installed on all clusters are returned next. Within this group there is no
-        order guarantee.
-        
-        3. Libraries that were previously requested on this cluster or on all clusters, but now marked for
-        removal. Within this group there is no order guarantee."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = ClusterStatus(cluster_id=cluster_id)
-
-        query = {}
-        if cluster_id: query['cluster_id'] = request.cluster_id
-
-        json = self._api.do('GET', '/api/2.0/libraries/cluster-status', query=query)
-        return ClusterLibraryStatuses.from_dict(json)
-
-    def install(self, cluster_id: str, libraries: List[Library], **kwargs):
-        """Add a library.
-        
-        Add libraries to be installed on a cluster. The installation is asynchronous; it happens in the
-        background after the completion of this request.
-        
-        **Note**: The actual set of libraries to be installed on a cluster is the union of the libraries
-        specified via this method and the libraries set to be installed on all clusters via the libraries UI."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = InstallLibraries(cluster_id=cluster_id, libraries=libraries)
-        body = request.as_dict()
-        self._api.do('POST', '/api/2.0/libraries/install', body=body)
-
-    def uninstall(self, cluster_id: str, libraries: List[Library], **kwargs):
-        """Uninstall libraries.
-        
-        Set libraries to be uninstalled on a cluster. The libraries won't be uninstalled until the cluster is
-        restarted. Uninstalling libraries that are not installed on the cluster will have no impact but is not
-        an error."""
-        request = kwargs.get('request', None)
-        if not request: # request is not given through keyed args
-            request = UninstallLibraries(cluster_id=cluster_id, libraries=libraries)
-        body = request.as_dict()
-        self._api.do('POST', '/api/2.0/libraries/uninstall', body=body)