databricks-sdk 0.28.0__py3-none-any.whl → 0.30.0__py3-none-any.whl

databricks/sdk/service/marketplace.py

@@ -56,7 +56,38 @@ class AssetType(Enum):
     ASSET_TYPE_MEDIA = 'ASSET_TYPE_MEDIA'
     ASSET_TYPE_MODEL = 'ASSET_TYPE_MODEL'
     ASSET_TYPE_NOTEBOOK = 'ASSET_TYPE_NOTEBOOK'
-    ASSET_TYPE_UNSPECIFIED = 'ASSET_TYPE_UNSPECIFIED'
+
+
+@dataclass
+class BatchGetListingsResponse:
+    listings: Optional[List[Listing]] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the BatchGetListingsResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.listings: body['listings'] = [v.as_dict() for v in self.listings]
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> BatchGetListingsResponse:
+        """Deserializes the BatchGetListingsResponse from a dictionary."""
+        return cls(listings=_repeated_dict(d, 'listings', Listing))
+
+
+@dataclass
+class BatchGetProvidersResponse:
+    providers: Optional[List[ProviderInfo]] = None
+
+    def as_dict(self) -> dict:
+        """Serializes the BatchGetProvidersResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.providers: body['providers'] = [v.as_dict() for v in self.providers]
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> BatchGetProvidersResponse:
+        """Deserializes the BatchGetProvidersResponse from a dictionary."""
+        return cls(providers=_repeated_dict(d, 'providers', ProviderInfo))
 
 
 class Category(Enum):
@@ -772,11 +803,6 @@ class FileStatus(Enum):
     FILE_STATUS_STAGING = 'FILE_STATUS_STAGING'
 
 
-class FilterType(Enum):
-
-    METASTORE = 'METASTORE'
-
-
 class FulfillmentType(Enum):
 
     INSTALL = 'INSTALL'
@@ -1423,23 +1449,18 @@ class ListingFulfillment:
 
 @dataclass
 class ListingSetting:
-    filters: Optional[List[VisibilityFilter]] = None
-    """filters are joined with `or` conjunction."""
-
     visibility: Optional[Visibility] = None
 
     def as_dict(self) -> dict:
         """Serializes the ListingSetting into a dictionary suitable for use as a JSON request body."""
         body = {}
-        if self.filters: body['filters'] = [v.as_dict() for v in self.filters]
         if self.visibility is not None: body['visibility'] = self.visibility.value
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> ListingSetting:
         """Deserializes the ListingSetting from a dictionary."""
-        return cls(filters=_repeated_dict(d, 'filters', VisibilityFilter),
-                   visibility=_enum(d, 'visibility', Visibility))
+        return cls(visibility=_enum(d, 'visibility', Visibility))
 
 
 class ListingShareType(Enum):
@@ -1479,8 +1500,6 @@ class ListingSummary:
     """if a git repo is being created, a listing will be initialized with this field as opposed to a
     share"""
 
-    metastore_id: Optional[str] = None
-
     provider_id: Optional[str] = None
 
     provider_region: Optional[RegionInfo] = None
@@ -1514,7 +1533,6 @@ class ListingSummary:
         if self.exchange_ids: body['exchange_ids'] = [v for v in self.exchange_ids]
         if self.git_repo: body['git_repo'] = self.git_repo.as_dict()
         if self.listing_type is not None: body['listingType'] = self.listing_type.value
-        if self.metastore_id is not None: body['metastore_id'] = self.metastore_id
         if self.name is not None: body['name'] = self.name
         if self.provider_id is not None: body['provider_id'] = self.provider_id
         if self.provider_region: body['provider_region'] = self.provider_region.as_dict()
@@ -1539,7 +1557,6 @@ class ListingSummary:
                    exchange_ids=d.get('exchange_ids', None),
                    git_repo=_from_dict(d, 'git_repo', RepoInfo),
                    listing_type=_enum(d, 'listingType', ListingType),
-                   metastore_id=d.get('metastore_id', None),
                    name=d.get('name', None),
                    provider_id=d.get('provider_id', None),
                    provider_region=_from_dict(d, 'provider_region', RegionInfo),
@@ -1579,7 +1596,6 @@ class ListingTagType(Enum):
 
     LISTING_TAG_TYPE_LANGUAGE = 'LISTING_TAG_TYPE_LANGUAGE'
     LISTING_TAG_TYPE_TASK = 'LISTING_TAG_TYPE_TASK'
-    LISTING_TAG_TYPE_UNSPECIFIED = 'LISTING_TAG_TYPE_UNSPECIFIED'
 
 
 class ListingType(Enum):
@@ -1900,14 +1916,6 @@ class SharedDataObject:
         return cls(data_object_type=d.get('data_object_type', None), name=d.get('name', None))
 
 
-class SortBy(Enum):
-
-    SORT_BY_DATE = 'SORT_BY_DATE'
-    SORT_BY_RELEVANCE = 'SORT_BY_RELEVANCE'
-    SORT_BY_TITLE = 'SORT_BY_TITLE'
-    SORT_BY_UNSPECIFIED = 'SORT_BY_UNSPECIFIED'
-
-
 @dataclass
 class TokenDetail:
     bearer_token: Optional[str] = None
@@ -2273,25 +2281,6 @@ class Visibility(Enum):
     PUBLIC = 'PUBLIC'
 
 
-@dataclass
-class VisibilityFilter:
-    filter_type: Optional[FilterType] = None
-
-    filter_value: Optional[str] = None
-
-    def as_dict(self) -> dict:
-        """Serializes the VisibilityFilter into a dictionary suitable for use as a JSON request body."""
-        body = {}
-        if self.filter_type is not None: body['filterType'] = self.filter_type.value
-        if self.filter_value is not None: body['filterValue'] = self.filter_value
-        return body
-
-    @classmethod
-    def from_dict(cls, d: Dict[str, any]) -> VisibilityFilter:
-        """Deserializes the VisibilityFilter from a dictionary."""
-        return cls(filter_type=_enum(d, 'filterType', FilterType), filter_value=d.get('filterValue', None))
-
-
 class ConsumerFulfillmentsAPI:
     """Fulfillments are entities that allow consumers to preview installations."""
 
@@ -2532,6 +2521,26 @@ class ConsumerListingsAPI:
     def __init__(self, api_client):
         self._api = api_client
 
+    def batch_get(self, *, ids: Optional[List[str]] = None) -> BatchGetListingsResponse:
+        """Get one batch of listings. One may specify up to 50 IDs per request.
+        
+        Batch get a published listing in the Databricks Marketplace that the consumer has access to.
+        
+        :param ids: List[str] (optional)
+        
+        :returns: :class:`BatchGetListingsResponse`
+        """
+
+        query = {}
+        if ids is not None: query['ids'] = [v for v in ids]
+        headers = {'Accept': 'application/json', }
+
+        res = self._api.do('GET',
+                           '/api/2.1/marketplace-consumer/listings:batchGet',
+                           query=query,
+                           headers=headers)
+        return BatchGetListingsResponse.from_dict(res)
+
     def get(self, id: str) -> GetListingResponse:
         """Get listing.
         
@@ -2551,14 +2560,12 @@ class ConsumerListingsAPI:
              *,
              assets: Optional[List[AssetType]] = None,
              categories: Optional[List[Category]] = None,
-             is_ascending: Optional[bool] = None,
              is_free: Optional[bool] = None,
              is_private_exchange: Optional[bool] = None,
              is_staff_pick: Optional[bool] = None,
              page_size: Optional[int] = None,
              page_token: Optional[str] = None,
              provider_ids: Optional[List[str]] = None,
-             sort_by: Optional[SortBy] = None,
              tags: Optional[List[ListingTag]] = None) -> Iterator[Listing]:
         """List listings.
         
@@ -2568,7 +2575,6 @@ class ConsumerListingsAPI:
           Matches any of the following asset types
         :param categories: List[:class:`Category`] (optional)
           Matches any of the following categories
-        :param is_ascending: bool (optional)
         :param is_free: bool (optional)
           Filters each listing based on if it is free.
         :param is_private_exchange: bool (optional)
@@ -2579,8 +2585,6 @@ class ConsumerListingsAPI:
         :param page_token: str (optional)
         :param provider_ids: List[str] (optional)
           Matches any of the following provider ids
-        :param sort_by: :class:`SortBy` (optional)
-          Criteria for sorting the resulting set of listings.
         :param tags: List[:class:`ListingTag`] (optional)
           Matches any of the following tags
         
@@ -2590,14 +2594,12 @@ class ConsumerListingsAPI:
         query = {}
         if assets is not None: query['assets'] = [v.value for v in assets]
         if categories is not None: query['categories'] = [v.value for v in categories]
-        if is_ascending is not None: query['is_ascending'] = is_ascending
         if is_free is not None: query['is_free'] = is_free
         if is_private_exchange is not None: query['is_private_exchange'] = is_private_exchange
         if is_staff_pick is not None: query['is_staff_pick'] = is_staff_pick
         if page_size is not None: query['page_size'] = page_size
         if page_token is not None: query['page_token'] = page_token
         if provider_ids is not None: query['provider_ids'] = [v for v in provider_ids]
-        if sort_by is not None: query['sort_by'] = sort_by.value
         if tags is not None: query['tags'] = [v.as_dict() for v in tags]
         headers = {'Accept': 'application/json', }
 
@@ -2615,13 +2617,11 @@ class ConsumerListingsAPI:
                *,
                assets: Optional[List[AssetType]] = None,
                categories: Optional[List[Category]] = None,
-               is_ascending: Optional[bool] = None,
                is_free: Optional[bool] = None,
                is_private_exchange: Optional[bool] = None,
                page_size: Optional[int] = None,
                page_token: Optional[str] = None,
-               provider_ids: Optional[List[str]] = None,
-               sort_by: Optional[SortBy] = None) -> Iterator[Listing]:
+               provider_ids: Optional[List[str]] = None) -> Iterator[Listing]:
         """Search listings.
         
         Search published listings in the Databricks Marketplace that the consumer has access to. This query
@@ -2633,14 +2633,12 @@ class ConsumerListingsAPI:
           Matches any of the following asset types
         :param categories: List[:class:`Category`] (optional)
           Matches any of the following categories
-        :param is_ascending: bool (optional)
         :param is_free: bool (optional)
         :param is_private_exchange: bool (optional)
         :param page_size: int (optional)
         :param page_token: str (optional)
         :param provider_ids: List[str] (optional)
           Matches any of the following provider ids
-        :param sort_by: :class:`SortBy` (optional)
         
         :returns: Iterator over :class:`Listing`
         """
@@ -2648,14 +2646,12 @@ class ConsumerListingsAPI:
         query = {}
         if assets is not None: query['assets'] = [v.value for v in assets]
         if categories is not None: query['categories'] = [v.value for v in categories]
-        if is_ascending is not None: query['is_ascending'] = is_ascending
         if is_free is not None: query['is_free'] = is_free
         if is_private_exchange is not None: query['is_private_exchange'] = is_private_exchange
         if page_size is not None: query['page_size'] = page_size
         if page_token is not None: query['page_token'] = page_token
         if provider_ids is not None: query['provider_ids'] = [v for v in provider_ids]
         if query is not None: query['query'] = query
-        if sort_by is not None: query['sort_by'] = sort_by.value
         headers = {'Accept': 'application/json', }
 
         while True:
@@ -2780,6 +2776,26 @@ class ConsumerProvidersAPI:
     def __init__(self, api_client):
         self._api = api_client
 
+    def batch_get(self, *, ids: Optional[List[str]] = None) -> BatchGetProvidersResponse:
+        """Get one batch of providers. One may specify up to 50 IDs per request.
+        
+        Batch get a provider in the Databricks Marketplace with at least one visible listing.
+        
+        :param ids: List[str] (optional)
+        
+        :returns: :class:`BatchGetProvidersResponse`
+        """
+
+        query = {}
+        if ids is not None: query['ids'] = [v for v in ids]
+        headers = {'Accept': 'application/json', }
+
+        res = self._api.do('GET',
+                           '/api/2.1/marketplace-consumer/providers:batchGet',
+                           query=query,
+                           headers=headers)
+        return BatchGetProvidersResponse.from_dict(res)
+
     def get(self, id: str) -> GetProviderResponse:
         """Get a provider.