|
16 | 16 | from typing import Any, Optional
|
17 | 17 |
|
18 | 18 | from azure.core.credentials import TokenCredential
|
| 19 | + from azure.core.pipeline.transport import HttpRequest, HttpResponse |
19 | 20 |
|
20 | 21 | from ._configuration import CosmosDBManagementClientConfiguration
|
21 | 22 | from .operations import DatabaseAccountsOperations
|
|
36 | 37 | from .operations import TableResourcesOperations
|
37 | 38 | from .operations import CassandraResourcesOperations
|
38 | 39 | from .operations import GremlinResourcesOperations
|
| 40 | +from .operations import RestorableDatabaseAccountsOperations |
| 41 | +from .operations import CosmosDBManagementClientOperationsMixin |
39 | 42 | from .operations import NotebookWorkspacesOperations
|
| 43 | +from .operations import RestorableSqlDatabasesOperations |
| 44 | +from .operations import RestorableSqlContainersOperations |
| 45 | +from .operations import RestorableSqlResourcesOperations |
| 46 | +from .operations import RestorableMongodbDatabasesOperations |
| 47 | +from .operations import RestorableMongodbCollectionsOperations |
| 48 | +from .operations import RestorableMongodbResourcesOperations |
| 49 | +from .operations import CassandraClustersOperations |
| 50 | +from .operations import CassandraDataCentersOperations |
40 | 51 | from .operations import PrivateLinkResourcesOperations
|
41 | 52 | from .operations import PrivateEndpointConnectionsOperations
|
| 53 | +from .operations import ServiceOperations |
42 | 54 | from . import models
|
43 | 55 |
|
44 | 56 |
|
45 |
| -class CosmosDBManagementClient(object): |
| 57 | +class CosmosDBManagementClient(CosmosDBManagementClientOperationsMixin): |
46 | 58 | """Azure Cosmos DB Database Service Resource Provider REST API.
|
47 | 59 |
|
48 | 60 | :ivar database_accounts: DatabaseAccountsOperations operations
|
@@ -81,12 +93,32 @@ class CosmosDBManagementClient(object):
|
81 | 93 | :vartype cassandra_resources: azure.mgmt.cosmosdb.operations.CassandraResourcesOperations
|
82 | 94 | :ivar gremlin_resources: GremlinResourcesOperations operations
|
83 | 95 | :vartype gremlin_resources: azure.mgmt.cosmosdb.operations.GremlinResourcesOperations
|
| 96 | + :ivar restorable_database_accounts: RestorableDatabaseAccountsOperations operations |
| 97 | + :vartype restorable_database_accounts: azure.mgmt.cosmosdb.operations.RestorableDatabaseAccountsOperations |
84 | 98 | :ivar notebook_workspaces: NotebookWorkspacesOperations operations
|
85 | 99 | :vartype notebook_workspaces: azure.mgmt.cosmosdb.operations.NotebookWorkspacesOperations
|
| 100 | + :ivar restorable_sql_databases: RestorableSqlDatabasesOperations operations |
| 101 | + :vartype restorable_sql_databases: azure.mgmt.cosmosdb.operations.RestorableSqlDatabasesOperations |
| 102 | + :ivar restorable_sql_containers: RestorableSqlContainersOperations operations |
| 103 | + :vartype restorable_sql_containers: azure.mgmt.cosmosdb.operations.RestorableSqlContainersOperations |
| 104 | + :ivar restorable_sql_resources: RestorableSqlResourcesOperations operations |
| 105 | + :vartype restorable_sql_resources: azure.mgmt.cosmosdb.operations.RestorableSqlResourcesOperations |
| 106 | + :ivar restorable_mongodb_databases: RestorableMongodbDatabasesOperations operations |
| 107 | + :vartype restorable_mongodb_databases: azure.mgmt.cosmosdb.operations.RestorableMongodbDatabasesOperations |
| 108 | + :ivar restorable_mongodb_collections: RestorableMongodbCollectionsOperations operations |
| 109 | + :vartype restorable_mongodb_collections: azure.mgmt.cosmosdb.operations.RestorableMongodbCollectionsOperations |
| 110 | + :ivar restorable_mongodb_resources: RestorableMongodbResourcesOperations operations |
| 111 | + :vartype restorable_mongodb_resources: azure.mgmt.cosmosdb.operations.RestorableMongodbResourcesOperations |
| 112 | + :ivar cassandra_clusters: CassandraClustersOperations operations |
| 113 | + :vartype cassandra_clusters: azure.mgmt.cosmosdb.operations.CassandraClustersOperations |
| 114 | + :ivar cassandra_data_centers: CassandraDataCentersOperations operations |
| 115 | + :vartype cassandra_data_centers: azure.mgmt.cosmosdb.operations.CassandraDataCentersOperations |
86 | 116 | :ivar private_link_resources: PrivateLinkResourcesOperations operations
|
87 | 117 | :vartype private_link_resources: azure.mgmt.cosmosdb.operations.PrivateLinkResourcesOperations
|
88 | 118 | :ivar private_endpoint_connections: PrivateEndpointConnectionsOperations operations
|
89 | 119 | :vartype private_endpoint_connections: azure.mgmt.cosmosdb.operations.PrivateEndpointConnectionsOperations
|
| 120 | + :ivar service: ServiceOperations operations |
| 121 | + :vartype service: azure.mgmt.cosmosdb.operations.ServiceOperations |
90 | 122 | :param credential: Credential needed for the client to connect to Azure.
|
91 | 123 | :type credential: ~azure.core.credentials.TokenCredential
|
92 | 124 | :param subscription_id: The ID of the target subscription.
|
@@ -149,12 +181,50 @@ def __init__(
|
149 | 181 | self._client, self._config, self._serialize, self._deserialize)
|
150 | 182 | self.gremlin_resources = GremlinResourcesOperations(
|
151 | 183 | self._client, self._config, self._serialize, self._deserialize)
|
| 184 | + self.restorable_database_accounts = RestorableDatabaseAccountsOperations( |
| 185 | + self._client, self._config, self._serialize, self._deserialize) |
152 | 186 | self.notebook_workspaces = NotebookWorkspacesOperations(
|
153 | 187 | self._client, self._config, self._serialize, self._deserialize)
|
| 188 | + self.restorable_sql_databases = RestorableSqlDatabasesOperations( |
| 189 | + self._client, self._config, self._serialize, self._deserialize) |
| 190 | + self.restorable_sql_containers = RestorableSqlContainersOperations( |
| 191 | + self._client, self._config, self._serialize, self._deserialize) |
| 192 | + self.restorable_sql_resources = RestorableSqlResourcesOperations( |
| 193 | + self._client, self._config, self._serialize, self._deserialize) |
| 194 | + self.restorable_mongodb_databases = RestorableMongodbDatabasesOperations( |
| 195 | + self._client, self._config, self._serialize, self._deserialize) |
| 196 | + self.restorable_mongodb_collections = RestorableMongodbCollectionsOperations( |
| 197 | + self._client, self._config, self._serialize, self._deserialize) |
| 198 | + self.restorable_mongodb_resources = RestorableMongodbResourcesOperations( |
| 199 | + self._client, self._config, self._serialize, self._deserialize) |
| 200 | + self.cassandra_clusters = CassandraClustersOperations( |
| 201 | + self._client, self._config, self._serialize, self._deserialize) |
| 202 | + self.cassandra_data_centers = CassandraDataCentersOperations( |
| 203 | + self._client, self._config, self._serialize, self._deserialize) |
154 | 204 | self.private_link_resources = PrivateLinkResourcesOperations(
|
155 | 205 | self._client, self._config, self._serialize, self._deserialize)
|
156 | 206 | self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
|
157 | 207 | self._client, self._config, self._serialize, self._deserialize)
|
| 208 | + self.service = ServiceOperations( |
| 209 | + self._client, self._config, self._serialize, self._deserialize) |
| 210 | + |
| 211 | + def _send_request(self, http_request, **kwargs): |
| 212 | + # type: (HttpRequest, Any) -> HttpResponse |
| 213 | + """Runs the network request through the client's chained policies. |
| 214 | +
|
| 215 | + :param http_request: The network request you want to make. Required. |
| 216 | + :type http_request: ~azure.core.pipeline.transport.HttpRequest |
| 217 | + :keyword bool stream: Whether the response payload will be streamed. Defaults to True. |
| 218 | + :return: The response of your network call. Does not do error handling on your response. |
| 219 | + :rtype: ~azure.core.pipeline.transport.HttpResponse |
| 220 | + """ |
| 221 | + path_format_arguments = { |
| 222 | + 'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str', min_length=1), |
| 223 | + } |
| 224 | + http_request.url = self._client.format_url(http_request.url, **path_format_arguments) |
| 225 | + stream = kwargs.pop("stream", True) |
| 226 | + pipeline_response = self._client._pipeline.run(http_request, stream=stream, **kwargs) |
| 227 | + return pipeline_response.http_response |
158 | 228 |
|
159 | 229 | def close(self):
|
160 | 230 | # type: () -> None
|
|
0 commit comments