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 .partition_safety_check import PartitionSafetyCheck
  
  
  class WaitForPrimarySwapSafetyCheck(PartitionSafetyCheck):
      """Safety check that waits for the primary replica to be moved out of the node
      before starting an upgrade to ensure the availability of the primary
      replica for the partition.
  
      All required parameters must be populated in order to send to Azure.
  
      :param kind: Required. Constant filled by server.
      :type kind: str
      :param partition_id: Id of the partition which is undergoing the safety
       check.
      :type partition_id: str
      """
  
      _validation = {
          'kind': {'required': True},
      }
  
      _attribute_map = {
          'kind': {'key': 'Kind', 'type': 'str'},
          'partition_id': {'key': 'PartitionId', 'type': 'str'},
      }
  
      def __init__(self, **kwargs):
          super(WaitForPrimarySwapSafetyCheck, self).__init__(**kwargs)
          self.kind = 'WaitForPrimarySwap'
  