text: # coding=utf-8
  # --------------------------------------------------------------------------
  # Copyright (c) Microsoft Corporation. All rights reserved.
  # Licensed under the MIT License. See License.txt in the project root for license information.
  # Code generated by Microsoft (R) AutoRest Code Generator.
  # Changes may cause incorrect behavior and will be lost if the code is regenerated.
  # --------------------------------------------------------------------------
  from typing import TYPE_CHECKING
  import warnings
  
  from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
  from azure.core.pipeline import PipelineResponse
  from azure.core.pipeline.transport import HttpRequest, HttpResponse
  from azure.core.polling import LROPoller, NoPolling, PollingMethod
  from azure.mgmt.core.exceptions import ARMErrorFormat
  from azure.mgmt.core.polling.arm_polling import ARMPolling
  
  from .. import models as _models
  
  if TYPE_CHECKING:
      # pylint: disable=unused-import,ungrouped-imports
      from typing import Any, Callable, Dict, Generic, Optional, TypeVar, Union
  
      T = TypeVar('T')
      ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]]
  
  class RecoveryServicesBackupClientOperationsMixin(object):
  
      def get_operation_status(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          operation_id,  # type: str
          **kwargs  # type: Any
      ):
          # type: (...) -> "_models.OperationStatus"
          """Fetches operation status for data move operation on vault.
  
          :param vault_name: The name of the recovery services vault.
          :type vault_name: str
          :param resource_group_name: The name of the resource group where the recovery services vault is
           present.
          :type resource_group_name: str
          :param operation_id:
          :type operation_id: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: OperationStatus, or the result of cls(response)
          :rtype: ~azure.mgmt.recoveryservicesbackup.models.OperationStatus
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatus"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          api_version = "2021-03-01"
          accept = "application/json"
  
          # Construct URL
          url = self.get_operation_status.metadata['url']  # type: ignore
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
              'operationId': self._serialize.url("operation_id", operation_id, 'str'),
          }
          url = self._client.format_url(url, **path_format_arguments)
  
          # Construct parameters
          query_parameters = {}  # type: Dict[str, Any]
          query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')
  
          # Construct headers
          header_parameters = {}  # type: Dict[str, Any]
          header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')
  
          request = self._client.get(url, query_parameters, header_parameters)
          pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs)
          response = pipeline_response.http_response
  
          if response.status_code not in [200]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              error = self._deserialize.failsafe_deserialize(_models.NewErrorResponse, response)
              raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
  
          deserialized = self._deserialize('OperationStatus', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
      get_operation_status.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupstorageconfig/vaultstorageconfig/operationStatus/{operationId}'}  # type: ignore
  
      def _bms_prepare_data_move_initial(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          parameters,  # type: "_models.PrepareDataMoveRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> None
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          api_version = "2021-03-01"
          content_type = kwargs.pop("content_type", "application/json")
          accept = "application/json"
  
          # Construct URL
          url = self._bms_prepare_data_move_initial.metadata['url']  # type: ignore
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
          }
          url = self._client.format_url(url, **path_format_arguments)
  
          # Construct parameters
          query_parameters = {}  # type: Dict[str, Any]
          query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')
  
          # Construct headers
          header_parameters = {}  # type: Dict[str, Any]
          header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str')
          header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')
  
          body_content_kwargs = {}  # type: Dict[str, Any]
          body_content = self._serialize.body(parameters, 'PrepareDataMoveRequest')
          body_content_kwargs['content'] = body_content
          request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs)
          pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs)
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              error = self._deserialize.failsafe_deserialize(_models.NewErrorResponse, response)
              raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _bms_prepare_data_move_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupstorageconfig/vaultstorageconfig/prepareDataMove'}  # type: ignore
  
      def begin_bms_prepare_data_move(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          parameters,  # type: "_models.PrepareDataMoveRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> LROPoller[None]
          """Prepares source vault for Data Move operation.
  
          :param vault_name: The name of the recovery services vault.
          :type vault_name: str
          :param resource_group_name: The name of the resource group where the recovery services vault is
           present.
          :type resource_group_name: str
          :param parameters: Prepare data move request.
          :type parameters: ~azure.mgmt.recoveryservicesbackup.models.PrepareDataMoveRequest
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: Pass in True if you'd like the ARMPolling polling method,
           False for no polling, or your own initialized polling object for a personal polling strategy.
          :paramtype polling: bool or ~azure.core.polling.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
          :return: An instance of LROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.LROPoller[None]
          :raises ~azure.core.exceptions.HttpResponseError:
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = self._bms_prepare_data_move_initial(
                  vault_name=vault_name,
                  resource_group_name=resource_group_name,
                  parameters=parameters,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
  
          kwargs.pop('error_map', None)
          kwargs.pop('content_type', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
          }
  
          if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments,  **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(self._client, raw_result, get_long_running_output, polling_method)
      begin_bms_prepare_data_move.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupstorageconfig/vaultstorageconfig/prepareDataMove'}  # type: ignore
  
      def _bms_trigger_data_move_initial(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          parameters,  # type: "_models.TriggerDataMoveRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> None
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          api_version = "2021-03-01"
          content_type = kwargs.pop("content_type", "application/json")
          accept = "application/json"
  
          # Construct URL
          url = self._bms_trigger_data_move_initial.metadata['url']  # type: ignore
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
          }
          url = self._client.format_url(url, **path_format_arguments)
  
          # Construct parameters
          query_parameters = {}  # type: Dict[str, Any]
          query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')
  
          # Construct headers
          header_parameters = {}  # type: Dict[str, Any]
          header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str')
          header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')
  
          body_content_kwargs = {}  # type: Dict[str, Any]
          body_content = self._serialize.body(parameters, 'TriggerDataMoveRequest')
          body_content_kwargs['content'] = body_content
          request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs)
          pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs)
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              error = self._deserialize.failsafe_deserialize(_models.NewErrorResponse, response)
              raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _bms_trigger_data_move_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupstorageconfig/vaultstorageconfig/triggerDataMove'}  # type: ignore
  
      def begin_bms_trigger_data_move(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          parameters,  # type: "_models.TriggerDataMoveRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> LROPoller[None]
          """Triggers Data Move Operation on target vault.
  
          :param vault_name: The name of the recovery services vault.
          :type vault_name: str
          :param resource_group_name: The name of the resource group where the recovery services vault is
           present.
          :type resource_group_name: str
          :param parameters: Trigger data move request.
          :type parameters: ~azure.mgmt.recoveryservicesbackup.models.TriggerDataMoveRequest
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: Pass in True if you'd like the ARMPolling polling method,
           False for no polling, or your own initialized polling object for a personal polling strategy.
          :paramtype polling: bool or ~azure.core.polling.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
          :return: An instance of LROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.LROPoller[None]
          :raises ~azure.core.exceptions.HttpResponseError:
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = self._bms_trigger_data_move_initial(
                  vault_name=vault_name,
                  resource_group_name=resource_group_name,
                  parameters=parameters,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
  
          kwargs.pop('error_map', None)
          kwargs.pop('content_type', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
          }
  
          if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments,  **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(self._client, raw_result, get_long_running_output, polling_method)
      begin_bms_trigger_data_move.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupstorageconfig/vaultstorageconfig/triggerDataMove'}  # type: ignore
  
      def _move_recovery_point_initial(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          fabric_name,  # type: str
          container_name,  # type: str
          protected_item_name,  # type: str
          recovery_point_id,  # type: str
          parameters,  # type: "_models.MoveRPAcrossTiersRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> None
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          api_version = "2021-03-01"
          content_type = kwargs.pop("content_type", "application/json")
          accept = "application/json"
  
          # Construct URL
          url = self._move_recovery_point_initial.metadata['url']  # type: ignore
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
              'fabricName': self._serialize.url("fabric_name", fabric_name, 'str'),
              'containerName': self._serialize.url("container_name", container_name, 'str'),
              'protectedItemName': self._serialize.url("protected_item_name", protected_item_name, 'str'),
              'recoveryPointId': self._serialize.url("recovery_point_id", recovery_point_id, 'str'),
          }
          url = self._client.format_url(url, **path_format_arguments)
  
          # Construct parameters
          query_parameters = {}  # type: Dict[str, Any]
          query_parameters['api-version'] = self._serialize.query("api_version", api_version, 'str')
  
          # Construct headers
          header_parameters = {}  # type: Dict[str, Any]
          header_parameters['Content-Type'] = self._serialize.header("content_type", content_type, 'str')
          header_parameters['Accept'] = self._serialize.header("accept", accept, 'str')
  
          body_content_kwargs = {}  # type: Dict[str, Any]
          body_content = self._serialize.body(parameters, 'MoveRPAcrossTiersRequest')
          body_content_kwargs['content'] = body_content
          request = self._client.post(url, query_parameters, header_parameters, **body_content_kwargs)
          pipeline_response = self._client._pipeline.run(request, stream=False, **kwargs)
          response = pipeline_response.http_response
  
          if response.status_code not in [202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _move_recovery_point_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupFabrics/{fabricName}/protectionContainers/{containerName}/protectedItems/{protectedItemName}/recoveryPoints/{recoveryPointId}/move'}  # type: ignore
  
      def begin_move_recovery_point(
          self,
          vault_name,  # type: str
          resource_group_name,  # type: str
          fabric_name,  # type: str
          container_name,  # type: str
          protected_item_name,  # type: str
          recovery_point_id,  # type: str
          parameters,  # type: "_models.MoveRPAcrossTiersRequest"
          **kwargs  # type: Any
      ):
          # type: (...) -> LROPoller[None]
          """Move recovery point from one datastore to another store.
  
          Move recovery point from one datastore to another store.
  
          :param vault_name: The name of the recovery services vault.
          :type vault_name: str
          :param resource_group_name: The name of the resource group where the recovery services vault is
           present.
          :type resource_group_name: str
          :param fabric_name:
          :type fabric_name: str
          :param container_name:
          :type container_name: str
          :param protected_item_name:
          :type protected_item_name: str
          :param recovery_point_id:
          :type recovery_point_id: str
          :param parameters: Move Resource Across Tiers Request.
          :type parameters: ~azure.mgmt.recoveryservicesbackup.models.MoveRPAcrossTiersRequest
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: Pass in True if you'd like the ARMPolling polling method,
           False for no polling, or your own initialized polling object for a personal polling strategy.
          :paramtype polling: bool or ~azure.core.polling.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no Retry-After header is present.
          :return: An instance of LROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.LROPoller[None]
          :raises ~azure.core.exceptions.HttpResponseError:
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = self._move_recovery_point_initial(
                  vault_name=vault_name,
                  resource_group_name=resource_group_name,
                  fabric_name=fabric_name,
                  container_name=container_name,
                  protected_item_name=protected_item_name,
                  recovery_point_id=recovery_point_id,
                  parameters=parameters,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
  
          kwargs.pop('error_map', None)
          kwargs.pop('content_type', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
          path_format_arguments = {
              'vaultName': self._serialize.url("vault_name", vault_name, 'str'),
              'resourceGroupName': self._serialize.url("resource_group_name", resource_group_name, 'str'),
              'subscriptionId': self._serialize.url("self._config.subscription_id", self._config.subscription_id, 'str'),
              'fabricName': self._serialize.url("fabric_name", fabric_name, 'str'),
              'containerName': self._serialize.url("container_name", container_name, 'str'),
              'protectedItemName': self._serialize.url("protected_item_name", protected_item_name, 'str'),
              'recoveryPointId': self._serialize.url("recovery_point_id", recovery_point_id, 'str'),
          }
  
          if polling is True: polling_method = ARMPolling(lro_delay, path_format_arguments=path_format_arguments,  **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(self._client, raw_result, get_long_running_output, polling_method)
      begin_move_recovery_point.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.RecoveryServices/vaults/{vaultName}/backupFabrics/{fabricName}/protectionContainers/{containerName}/protectedItems/{protectedItemName}/recoveryPoints/{recoveryPointId}/move'}  # type: ignore
  