diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/_my_sql_management_client.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/_my_sql_management_client.py index 59a458d85378..5a1ec3d6105b 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/_my_sql_management_client.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/_my_sql_management_client.py @@ -24,8 +24,8 @@ from .operations import ServerAdministratorsOperations from .operations import LocationBasedPerformanceTierOperations from .operations import CheckNameAvailabilityOperations -from .operations import ServerSecurityAlertPoliciesOperations from .operations import Operations +from .operations import ServerSecurityAlertPoliciesOperations from .operations import QueryTextsOperations from .operations import TopQueryStatisticsOperations from .operations import WaitStatisticsOperations @@ -65,10 +65,10 @@ class MySQLManagementClient(MySQLManagementClientOperationsMixin, SDKClient): :vartype location_based_performance_tier: azure.mgmt.rdbms.mysql.operations.LocationBasedPerformanceTierOperations :ivar check_name_availability: CheckNameAvailability operations :vartype check_name_availability: azure.mgmt.rdbms.mysql.operations.CheckNameAvailabilityOperations - :ivar server_security_alert_policies: ServerSecurityAlertPolicies operations - :vartype server_security_alert_policies: azure.mgmt.rdbms.mysql.operations.ServerSecurityAlertPoliciesOperations :ivar operations: Operations operations :vartype operations: azure.mgmt.rdbms.mysql.operations.Operations + :ivar server_security_alert_policies: ServerSecurityAlertPolicies operations + :vartype server_security_alert_policies: azure.mgmt.rdbms.mysql.operations.ServerSecurityAlertPoliciesOperations :ivar query_texts: QueryTexts operations :vartype query_texts: azure.mgmt.rdbms.mysql.operations.QueryTextsOperations :ivar top_query_statistics: TopQueryStatistics operations @@ -128,10 +128,10 @@ def __init__( self._client, self.config, self._serialize, self._deserialize) self.check_name_availability = CheckNameAvailabilityOperations( self._client, self.config, self._serialize, self._deserialize) - self.server_security_alert_policies = ServerSecurityAlertPoliciesOperations( - self._client, self.config, self._serialize, self._deserialize) self.operations = Operations( self._client, self.config, self._serialize, self._deserialize) + self.server_security_alert_policies = ServerSecurityAlertPoliciesOperations( + self._client, self.config, self._serialize, self._deserialize) self.query_texts = QueryTextsOperations( self._client, self.config, self._serialize, self._deserialize) self.top_query_statistics = TopQueryStatisticsOperations( diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/__init__.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/__init__.py index 8d9d59cf49ea..03a54f14bfbd 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/__init__.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/__init__.py @@ -52,6 +52,7 @@ from ._models_py3 import ServerPropertiesForRestore from ._models_py3 import ServerSecurityAlertPolicy from ._models_py3 import ServerUpdateParameters + from ._models_py3 import ServerUpgradeParameters from ._models_py3 import Sku from ._models_py3 import StorageProfile from ._models_py3 import TagsObject @@ -103,6 +104,7 @@ from ._models import ServerPropertiesForRestore from ._models import ServerSecurityAlertPolicy from ._models import ServerUpdateParameters + from ._models import ServerUpgradeParameters from ._models import Sku from ._models import StorageProfile from ._models import TagsObject @@ -189,6 +191,7 @@ 'ServerPropertiesForRestore', 'ServerSecurityAlertPolicy', 'ServerUpdateParameters', + 'ServerUpgradeParameters', 'Sku', 'StorageProfile', 'TagsObject', diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models.py index 35c28b535292..2001260e3c17 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models.py @@ -158,7 +158,7 @@ def __init__(self, **kwargs): class CloudError(Model): """An error response from the Batch service. - :param error: + :param error: Error message :type error: ~azure.mgmt.rdbms.mysql.models.ErrorResponse """ @@ -1979,6 +1979,22 @@ def __init__(self, **kwargs): self.tags = kwargs.get('tags', None) +class ServerUpgradeParameters(Model): + """ServerUpgradeParameters. + + :param target_server_version: Represents an server storage profile. + :type target_server_version: str + """ + + _attribute_map = { + 'target_server_version': {'key': 'properties.targetServerVersion', 'type': 'str'}, + } + + def __init__(self, **kwargs): + super(ServerUpgradeParameters, self).__init__(**kwargs) + self.target_server_version = kwargs.get('target_server_version', None) + + class Sku(Model): """Billing information related properties of a server. diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models_py3.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models_py3.py index 17d3aa9f63f7..e14d12fae6ee 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models_py3.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_models_py3.py @@ -158,7 +158,7 @@ def __init__(self, **kwargs) -> None: class CloudError(Model): """An error response from the Batch service. - :param error: + :param error: Error message :type error: ~azure.mgmt.rdbms.mysql.models.ErrorResponse """ @@ -1979,6 +1979,22 @@ def __init__(self, *, identity=None, sku=None, storage_profile=None, administrat self.tags = tags +class ServerUpgradeParameters(Model): + """ServerUpgradeParameters. + + :param target_server_version: Represents an server storage profile. + :type target_server_version: str + """ + + _attribute_map = { + 'target_server_version': {'key': 'properties.targetServerVersion', 'type': 'str'}, + } + + def __init__(self, *, target_server_version: str=None, **kwargs) -> None: + super(ServerUpgradeParameters, self).__init__(**kwargs) + self.target_server_version = target_server_version + + class Sku(Model): """Billing information related properties of a server. diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_my_sql_management_client_enums.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_my_sql_management_client_enums.py index 6f0b1e2c7d42..926d8bd46423 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_my_sql_management_client_enums.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/models/_my_sql_management_client_enums.py @@ -35,8 +35,8 @@ class MinimalTlsVersionEnum(str, Enum): class InfrastructureEncryption(str, Enum): - enabled = "Enabled" - disabled = "Disabled" + enabled = "Enabled" #: Default value for single layer of encryption for data at rest. + disabled = "Disabled" #: Additional (2nd) layer of encryption for data at rest class PublicNetworkAccessEnum(str, Enum): diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/__init__.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/__init__.py index df66b2e41971..10badf36caa8 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/__init__.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/__init__.py @@ -19,8 +19,8 @@ from ._server_administrators_operations import ServerAdministratorsOperations from ._location_based_performance_tier_operations import LocationBasedPerformanceTierOperations from ._check_name_availability_operations import CheckNameAvailabilityOperations -from ._server_security_alert_policies_operations import ServerSecurityAlertPoliciesOperations from ._operations import Operations +from ._server_security_alert_policies_operations import ServerSecurityAlertPoliciesOperations from ._query_texts_operations import QueryTextsOperations from ._top_query_statistics_operations import TopQueryStatisticsOperations from ._wait_statistics_operations import WaitStatisticsOperations @@ -44,8 +44,8 @@ 'ServerAdministratorsOperations', 'LocationBasedPerformanceTierOperations', 'CheckNameAvailabilityOperations', - 'ServerSecurityAlertPoliciesOperations', 'Operations', + 'ServerSecurityAlertPoliciesOperations', 'QueryTextsOperations', 'TopQueryStatisticsOperations', 'WaitStatisticsOperations', diff --git a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/_servers_operations.py b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/_servers_operations.py index bc547cbb22fd..cdef01c38027 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/_servers_operations.py +++ b/sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql/operations/_servers_operations.py @@ -27,7 +27,6 @@ class ServersOperations(object): :param config: Configuration of service client. :param serializer: An object model serializer. :param deserializer: An object model deserializer. - :ivar api_version: The API version to use for this operation. Constant value: "2017-12-01". """ models = models @@ -37,13 +36,14 @@ def __init__(self, client, config, serializer, deserializer): self._client = client self._serialize = serializer self._deserialize = deserializer - self.api_version = "2017-12-01" self.config = config def _create_initial( self, resource_group_name, server_name, parameters, custom_headers=None, raw=False, **operation_config): + api_version = "2017-12-01" + # Construct URL url = self.create.metadata['url'] path_format_arguments = { @@ -55,7 +55,7 @@ def _create_initial( # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) # Construct headers header_parameters = {} @@ -149,6 +149,8 @@ def get_long_running_output(response): def _update_initial( self, resource_group_name, server_name, parameters, custom_headers=None, raw=False, **operation_config): + api_version = "2017-12-01" + # Construct URL url = self.update.metadata['url'] path_format_arguments = { @@ -160,7 +162,7 @@ def _update_initial( # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) # Construct headers header_parameters = {} @@ -252,6 +254,8 @@ def get_long_running_output(response): def _delete_initial( self, resource_group_name, server_name, custom_headers=None, raw=False, **operation_config): + api_version = "2017-12-01" + # Construct URL url = self.delete.metadata['url'] path_format_arguments = { @@ -263,7 +267,7 @@ def _delete_initial( # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) # Construct headers header_parameters = {} @@ -348,6 +352,8 @@ def get( ~msrest.pipeline.ClientRawResponse :raises: :class:`CloudError` """ + api_version = "2017-12-01" + # Construct URL url = self.get.metadata['url'] path_format_arguments = { @@ -359,7 +365,7 @@ def get( # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) # Construct headers header_parameters = {} @@ -408,6 +414,8 @@ def list_by_resource_group( ~azure.mgmt.rdbms.mysql.models.ServerPaged[~azure.mgmt.rdbms.mysql.models.Server] :raises: :class:`CloudError` """ + api_version = "2017-12-01" + def prepare_request(next_link=None): if not next_link: # Construct URL @@ -420,7 +428,7 @@ def prepare_request(next_link=None): # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) else: url = next_link @@ -475,6 +483,8 @@ def list( ~azure.mgmt.rdbms.mysql.models.ServerPaged[~azure.mgmt.rdbms.mysql.models.Server] :raises: :class:`CloudError` """ + api_version = "2017-12-01" + def prepare_request(next_link=None): if not next_link: # Construct URL @@ -486,7 +496,7 @@ def prepare_request(next_link=None): # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) else: url = next_link @@ -530,6 +540,8 @@ def internal_paging(next_link=None): def _restart_initial( self, resource_group_name, server_name, custom_headers=None, raw=False, **operation_config): + api_version = "2017-12-01" + # Construct URL url = self.restart.metadata['url'] path_format_arguments = { @@ -541,7 +553,7 @@ def _restart_initial( # Construct parameters query_parameters = {} - query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str', min_length=1) + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) # Construct headers header_parameters = {} @@ -606,3 +618,258 @@ def get_long_running_output(response): else: polling_method = polling return LROPoller(self._client, raw_result, get_long_running_output, polling_method) restart.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/restart'} + + + def _start_initial( + self, resource_group_name, server_name, custom_headers=None, raw=False, **operation_config): + api_version = "2020-01-01" + + # Construct URL + url = self.start.metadata['url'] + path_format_arguments = { + 'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str', min_length=1), + 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + 'serverName': self._serialize.url("server_name", server_name, 'str') + } + url = self._client.format_url(url, **path_format_arguments) + + # Construct parameters + query_parameters = {} + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) + + # Construct headers + header_parameters = {} + if self.config.generate_client_request_id: + header_parameters['x-ms-client-request-id'] = str(uuid.uuid1()) + if custom_headers: + header_parameters.update(custom_headers) + if self.config.accept_language is not None: + header_parameters['accept-language'] = self._serialize.header("self.config.accept_language", self.config.accept_language, 'str') + + # Construct and send request + request = self._client.post(url, query_parameters, header_parameters) + response = self._client.send(request, stream=False, **operation_config) + + if response.status_code not in [200, 202]: + exp = CloudError(response) + exp.request_id = response.headers.get('x-ms-request-id') + raise exp + + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + def start( + self, resource_group_name, server_name, custom_headers=None, raw=False, polling=True, **operation_config): + """Starts a stopped server. + + :param resource_group_name: The name of the resource group. The name + is case insensitive. + :type resource_group_name: str + :param server_name: The name of the server. + :type server_name: str + :param dict custom_headers: headers that will be added to the request + :param bool raw: The poller return type is ClientRawResponse, the + direct response alongside the deserialized response + :param polling: True for ARMPolling, False for no polling, or a + polling object for personal polling strategy + :return: An instance of LROPoller that returns None or + ClientRawResponse if raw==True + :rtype: ~msrestazure.azure_operation.AzureOperationPoller[None] or + ~msrestazure.azure_operation.AzureOperationPoller[~msrest.pipeline.ClientRawResponse[None]] + :raises: :class:`CloudError` + """ + raw_result = self._start_initial( + resource_group_name=resource_group_name, + server_name=server_name, + custom_headers=custom_headers, + raw=True, + **operation_config + ) + + def get_long_running_output(response): + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + lro_delay = operation_config.get( + 'long_running_operation_timeout', + self.config.long_running_operation_timeout) + if polling is True: polling_method = ARMPolling(lro_delay, **operation_config) + elif polling is False: polling_method = NoPolling() + else: polling_method = polling + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + start.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/start'} + + + def _stop_initial( + self, resource_group_name, server_name, custom_headers=None, raw=False, **operation_config): + api_version = "2020-01-01" + + # Construct URL + url = self.stop.metadata['url'] + path_format_arguments = { + 'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str', min_length=1), + 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + 'serverName': self._serialize.url("server_name", server_name, 'str') + } + url = self._client.format_url(url, **path_format_arguments) + + # Construct parameters + query_parameters = {} + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) + + # Construct headers + header_parameters = {} + if self.config.generate_client_request_id: + header_parameters['x-ms-client-request-id'] = str(uuid.uuid1()) + if custom_headers: + header_parameters.update(custom_headers) + if self.config.accept_language is not None: + header_parameters['accept-language'] = self._serialize.header("self.config.accept_language", self.config.accept_language, 'str') + + # Construct and send request + request = self._client.post(url, query_parameters, header_parameters) + response = self._client.send(request, stream=False, **operation_config) + + if response.status_code not in [200, 202]: + exp = CloudError(response) + exp.request_id = response.headers.get('x-ms-request-id') + raise exp + + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + def stop( + self, resource_group_name, server_name, custom_headers=None, raw=False, polling=True, **operation_config): + """Stops a running server. + + :param resource_group_name: The name of the resource group. The name + is case insensitive. + :type resource_group_name: str + :param server_name: The name of the server. + :type server_name: str + :param dict custom_headers: headers that will be added to the request + :param bool raw: The poller return type is ClientRawResponse, the + direct response alongside the deserialized response + :param polling: True for ARMPolling, False for no polling, or a + polling object for personal polling strategy + :return: An instance of LROPoller that returns None or + ClientRawResponse if raw==True + :rtype: ~msrestazure.azure_operation.AzureOperationPoller[None] or + ~msrestazure.azure_operation.AzureOperationPoller[~msrest.pipeline.ClientRawResponse[None]] + :raises: :class:`CloudError` + """ + raw_result = self._stop_initial( + resource_group_name=resource_group_name, + server_name=server_name, + custom_headers=custom_headers, + raw=True, + **operation_config + ) + + def get_long_running_output(response): + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + lro_delay = operation_config.get( + 'long_running_operation_timeout', + self.config.long_running_operation_timeout) + if polling is True: polling_method = ARMPolling(lro_delay, **operation_config) + elif polling is False: polling_method = NoPolling() + else: polling_method = polling + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + stop.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/stop'} + + + def _upgrade_initial( + self, resource_group_name, server_name, target_server_version=None, custom_headers=None, raw=False, **operation_config): + parameters = models.ServerUpgradeParameters(target_server_version=target_server_version) + + api_version = "2020-01-01" + + # Construct URL + url = self.upgrade.metadata['url'] + path_format_arguments = { + 'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str', min_length=1), + 'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1, pattern=r'^[-\w\._\(\)]+$'), + 'serverName': self._serialize.url("server_name", server_name, 'str') + } + url = self._client.format_url(url, **path_format_arguments) + + # Construct parameters + query_parameters = {} + query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str', min_length=1) + + # Construct headers + header_parameters = {} + header_parameters['Content-Type'] = 'application/json; charset=utf-8' + if self.config.generate_client_request_id: + header_parameters['x-ms-client-request-id'] = str(uuid.uuid1()) + if custom_headers: + header_parameters.update(custom_headers) + if self.config.accept_language is not None: + header_parameters['accept-language'] = self._serialize.header("self.config.accept_language", self.config.accept_language, 'str') + + # Construct body + body_content = self._serialize.body(parameters, 'ServerUpgradeParameters') + + # Construct and send request + request = self._client.post(url, query_parameters, header_parameters, body_content) + response = self._client.send(request, stream=False, **operation_config) + + if response.status_code not in [200, 202]: + exp = CloudError(response) + exp.request_id = response.headers.get('x-ms-request-id') + raise exp + + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + def upgrade( + self, resource_group_name, server_name, target_server_version=None, custom_headers=None, raw=False, polling=True, **operation_config): + """Upgrade server version. + + :param resource_group_name: The name of the resource group. The name + is case insensitive. + :type resource_group_name: str + :param server_name: The name of the server. + :type server_name: str + :param target_server_version: Represents an server storage profile. + :type target_server_version: str + :param dict custom_headers: headers that will be added to the request + :param bool raw: The poller return type is ClientRawResponse, the + direct response alongside the deserialized response + :param polling: True for ARMPolling, False for no polling, or a + polling object for personal polling strategy + :return: An instance of LROPoller that returns None or + ClientRawResponse if raw==True + :rtype: ~msrestazure.azure_operation.AzureOperationPoller[None] or + ~msrestazure.azure_operation.AzureOperationPoller[~msrest.pipeline.ClientRawResponse[None]] + :raises: :class:`CloudError` + """ + raw_result = self._upgrade_initial( + resource_group_name=resource_group_name, + server_name=server_name, + target_server_version=target_server_version, + custom_headers=custom_headers, + raw=True, + **operation_config + ) + + def get_long_running_output(response): + if raw: + client_raw_response = ClientRawResponse(None, response) + return client_raw_response + + lro_delay = operation_config.get( + 'long_running_operation_timeout', + self.config.long_running_operation_timeout) + if polling is True: polling_method = ARMPolling(lro_delay, **operation_config) + elif polling is False: polling_method = NoPolling() + else: polling_method = polling + return LROPoller(self._client, raw_result, get_long_running_output, polling_method) + upgrade.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/upgrade'} diff --git a/sdk/rdbms/azure-mgmt-rdbms/setup.py b/sdk/rdbms/azure-mgmt-rdbms/setup.py index ccc2d30de0d8..60fdabe1cfc7 100644 --- a/sdk/rdbms/azure-mgmt-rdbms/setup.py +++ b/sdk/rdbms/azure-mgmt-rdbms/setup.py @@ -36,7 +36,9 @@ pass # Version extraction inspired from 'requests' -with open(os.path.join(package_folder_path, 'version.py'), 'r') as fd: +with open(os.path.join(package_folder_path, 'version.py') + if os.path.exists(os.path.join(package_folder_path, 'version.py')) + else os.path.join(package_folder_path, '_version.py'), 'r') as fd: version = re.search(r'^VERSION\s*=\s*[\'"]([^\'"]*)[\'"]', fd.read(), re.MULTILINE).group(1)