Generated from e94036625f69f01964f3b6821aa72c4cc578073e

Add Oracle to Azure Database For PostgreSQL scenarios and Service Task resource which supports the scenario.

sdk/datamigration/azure-mgmt-datamigration/azure/mgmt/datamigration/operations/service_tasks_operations.py

@@ -114,3 +114,372 @@ def internal_paging(next_link=None, raw=False):
 
         return deserialized
     list.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks'}
+
+    def create_or_update(
+            self, group_name, service_name, task_name, etag=None, properties=None, custom_headers=None, raw=False, **operation_config):
+        """Create or update service task.
+
+        The service tasks resource is a nested, proxy-only resource
+        representing work performed by a DMS instance. The PUT method creates a
+        new service task or updates an existing one, although since service
+        tasks have no mutable custom properties, there is little reason to
+        update an existing one.
+
+        :param group_name: Name of the resource group
+        :type group_name: str
+        :param service_name: Name of the service
+        :type service_name: str
+        :param task_name: Name of the Task
+        :type task_name: str
+        :param etag: HTTP strong entity tag value. This is ignored if
+         submitted.
+        :type etag: str
+        :param properties: Custom task properties
+        :type properties:
+         ~azure.mgmt.datamigration.models.ProjectTaskProperties
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: ProjectTask or ClientRawResponse if raw=true
+        :rtype: ~azure.mgmt.datamigration.models.ProjectTask or
+         ~msrest.pipeline.ClientRawResponse
+        :raises:
+         :class:`ApiErrorException<azure.mgmt.datamigration.models.ApiErrorException>`
+        """
+        parameters = models.ProjectTask(etag=etag, properties=properties)
+
+        # Construct URL
+        url = self.create_or_update.metadata['url']
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str'),
+            'groupName': self._serialize.url("group_name", group_name, 'str'),
+            'serviceName': self._serialize.url("service_name", service_name, 'str'),
+            'taskName': self._serialize.url("task_name", task_name, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        header_parameters['Accept'] = 'application/json'
+        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, 'ProjectTask')
+
+        # Construct and send request
+        request = self._client.put(url, query_parameters, header_parameters, body_content)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200, 201]:
+            raise models.ApiErrorException(self._deserialize, response)
+
+        deserialized = None
+
+        if response.status_code == 200:
+            deserialized = self._deserialize('ProjectTask', response)
+        if response.status_code == 201:
+            deserialized = self._deserialize('ProjectTask', response)
+
+        if raw:
+            client_raw_response = ClientRawResponse(deserialized, response)
+            return client_raw_response
+
+        return deserialized
+    create_or_update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks/{taskName}'}
+
+    def get(
+            self, group_name, service_name, task_name, expand=None, custom_headers=None, raw=False, **operation_config):
+        """Get service task information.
+
+        The service tasks resource is a nested, proxy-only resource
+        representing work performed by a DMS instance. The GET method retrieves
+        information about a service task.
+
+        :param group_name: Name of the resource group
+        :type group_name: str
+        :param service_name: Name of the service
+        :type service_name: str
+        :param task_name: Name of the Task
+        :type task_name: str
+        :param expand: Expand the response
+        :type expand: str
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: ProjectTask or ClientRawResponse if raw=true
+        :rtype: ~azure.mgmt.datamigration.models.ProjectTask or
+         ~msrest.pipeline.ClientRawResponse
+        :raises:
+         :class:`ApiErrorException<azure.mgmt.datamigration.models.ApiErrorException>`
+        """
+        # Construct URL
+        url = self.get.metadata['url']
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str'),
+            'groupName': self._serialize.url("group_name", group_name, 'str'),
+            'serviceName': self._serialize.url("service_name", service_name, 'str'),
+            'taskName': self._serialize.url("task_name", task_name, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        if expand is not None:
+            query_parameters['$expand'] = self._serialize.query("expand", expand, 'str')
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        header_parameters['Accept'] = 'application/json'
+        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.get(url, query_parameters, header_parameters)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200]:
+            raise models.ApiErrorException(self._deserialize, response)
+
+        deserialized = None
+
+        if response.status_code == 200:
+            deserialized = self._deserialize('ProjectTask', response)
+
+        if raw:
+            client_raw_response = ClientRawResponse(deserialized, response)
+            return client_raw_response
+
+        return deserialized
+    get.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks/{taskName}'}
+
+    def delete(
+            self, group_name, service_name, task_name, delete_running_tasks=None, custom_headers=None, raw=False, **operation_config):
+        """Delete service task.
+
+        The service tasks resource is a nested, proxy-only resource
+        representing work performed by a DMS instance. The DELETE method
+        deletes a service task, canceling it first if it's running.
+
+        :param group_name: Name of the resource group
+        :type group_name: str
+        :param service_name: Name of the service
+        :type service_name: str
+        :param task_name: Name of the Task
+        :type task_name: str
+        :param delete_running_tasks: Delete the resource even if it contains
+         running tasks
+        :type delete_running_tasks: bool
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: None or ClientRawResponse if raw=true
+        :rtype: None or ~msrest.pipeline.ClientRawResponse
+        :raises:
+         :class:`ApiErrorException<azure.mgmt.datamigration.models.ApiErrorException>`
+        """
+        # Construct URL
+        url = self.delete.metadata['url']
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str'),
+            'groupName': self._serialize.url("group_name", group_name, 'str'),
+            'serviceName': self._serialize.url("service_name", service_name, 'str'),
+            'taskName': self._serialize.url("task_name", task_name, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        if delete_running_tasks is not None:
+            query_parameters['deleteRunningTasks'] = self._serialize.query("delete_running_tasks", delete_running_tasks, 'bool')
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # 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.delete(url, query_parameters, header_parameters)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200, 204]:
+            raise models.ApiErrorException(self._deserialize, response)
+
+        if raw:
+            client_raw_response = ClientRawResponse(None, response)
+            return client_raw_response
+    delete.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks/{taskName}'}
+
+    def update(
+            self, group_name, service_name, task_name, etag=None, properties=None, custom_headers=None, raw=False, **operation_config):
+        """Create or update service task.
+
+        The service tasks resource is a nested, proxy-only resource
+        representing work performed by a DMS instance. The PATCH method updates
+        an existing service task, but since service tasks have no mutable
+        custom properties, there is little reason to do so.
+
+        :param group_name: Name of the resource group
+        :type group_name: str
+        :param service_name: Name of the service
+        :type service_name: str
+        :param task_name: Name of the Task
+        :type task_name: str
+        :param etag: HTTP strong entity tag value. This is ignored if
+         submitted.
+        :type etag: str
+        :param properties: Custom task properties
+        :type properties:
+         ~azure.mgmt.datamigration.models.ProjectTaskProperties
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: ProjectTask or ClientRawResponse if raw=true
+        :rtype: ~azure.mgmt.datamigration.models.ProjectTask or
+         ~msrest.pipeline.ClientRawResponse
+        :raises:
+         :class:`ApiErrorException<azure.mgmt.datamigration.models.ApiErrorException>`
+        """
+        parameters = models.ProjectTask(etag=etag, properties=properties)
+
+        # Construct URL
+        url = self.update.metadata['url']
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str'),
+            'groupName': self._serialize.url("group_name", group_name, 'str'),
+            'serviceName': self._serialize.url("service_name", service_name, 'str'),
+            'taskName': self._serialize.url("task_name", task_name, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        header_parameters['Accept'] = 'application/json'
+        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, 'ProjectTask')
+
+        # Construct and send request
+        request = self._client.patch(url, query_parameters, header_parameters, body_content)
+        response = self._client.send(request, stream=False, **operation_config)
+
+        if response.status_code not in [200]:
+            raise models.ApiErrorException(self._deserialize, response)
+
+        deserialized = None
+
+        if response.status_code == 200:
+            deserialized = self._deserialize('ProjectTask', response)
+
+        if raw:
+            client_raw_response = ClientRawResponse(deserialized, response)
+            return client_raw_response
+
+        return deserialized
+    update.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks/{taskName}'}
+
+    def cancel(
+            self, group_name, service_name, task_name, custom_headers=None, raw=False, **operation_config):
+        """Cancel a service task.
+
+        The service tasks resource is a nested, proxy-only resource
+        representing work performed by a DMS instance. This method cancels a
+        service task if it's currently queued or running.
+
+        :param group_name: Name of the resource group
+        :type group_name: str
+        :param service_name: Name of the service
+        :type service_name: str
+        :param task_name: Name of the Task
+        :type task_name: str
+        :param dict custom_headers: headers that will be added to the request
+        :param bool raw: returns the direct response alongside the
+         deserialized response
+        :param operation_config: :ref:`Operation configuration
+         overrides<msrest:optionsforoperations>`.
+        :return: ProjectTask or ClientRawResponse if raw=true
+        :rtype: ~azure.mgmt.datamigration.models.ProjectTask or
+         ~msrest.pipeline.ClientRawResponse
+        :raises:
+         :class:`ApiErrorException<azure.mgmt.datamigration.models.ApiErrorException>`
+        """
+        # Construct URL
+        url = self.cancel.metadata['url']
+        path_format_arguments = {
+            'subscriptionId': self._serialize.url("self.config.subscription_id", self.config.subscription_id, 'str'),
+            'groupName': self._serialize.url("group_name", group_name, 'str'),
+            'serviceName': self._serialize.url("service_name", service_name, 'str'),
+            'taskName': self._serialize.url("task_name", task_name, 'str')
+        }
+        url = self._client.format_url(url, **path_format_arguments)
+
+        # Construct parameters
+        query_parameters = {}
+        query_parameters['api-version'] = self._serialize.query("self.api_version", self.api_version, 'str')
+
+        # Construct headers
+        header_parameters = {}
+        header_parameters['Accept'] = 'application/json'
+        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]:
+            raise models.ApiErrorException(self._deserialize, response)
+
+        deserialized = None
+
+        if response.status_code == 200:
+            deserialized = self._deserialize('ProjectTask', response)
+
+        if raw:
+            client_raw_response = ClientRawResponse(deserialized, response)
+            return client_raw_response
+
+        return deserialized
+    cancel.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{groupName}/providers/Microsoft.DataMigration/services/{serviceName}/serviceTasks/{taskName}/cancel'}