text: # coding: utf-8
  # Copyright (c) 2016, 2021, Oracle and/or its affiliates.  All rights reserved.
  # This software is dual-licensed to you under the Universal Permissive License (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl or Apache License 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose either license.
  
  import oci  # noqa: F401
  from oci.util import WAIT_RESOURCE_NOT_FOUND  # noqa: F401
  
  
  class DataFlowClientCompositeOperations(object):
      """
      This class provides a wrapper around :py:class:`~oci.data_flow.DataFlowClient` and offers convenience methods
      for operations that would otherwise need to be chained together. For example, instead of performing an action
      on a resource (e.g. launching an instance, creating a load balancer) and then using a waiter to wait for the resource
      to enter a given state, you can call a single method in this class to accomplish the same functionality
      """
  
      def __init__(self, client, **kwargs):
          """
          Creates a new DataFlowClientCompositeOperations object
  
          :param DataFlowClient client:
              The service client which will be wrapped by this object
          """
          self.client = client
  
      def change_private_endpoint_compartment_and_wait_for_state(self, private_endpoint_id, change_private_endpoint_compartment_details, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.change_private_endpoint_compartment` and waits for the :py:class:`~oci.data_flow.models.WorkRequest`
          to enter the given state(s).
  
          :param str private_endpoint_id: (required)
              The unique ID for a private endpoint.
  
          :param oci.data_flow.models.ChangePrivateEndpointCompartmentDetails change_private_endpoint_compartment_details: (required)
              Details for changing a private endpoint's compartment.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.WorkRequest.status`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.change_private_endpoint_compartment`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.change_private_endpoint_compartment(private_endpoint_id, change_private_endpoint_compartment_details, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.headers['opc-work-request-id']
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_work_request(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'status') and getattr(r.data, 'status').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def create_application_and_wait_for_state(self, create_application_details, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.create_application` and waits for the :py:class:`~oci.data_flow.models.Application` acted upon
          to enter the given state(s).
  
          :param oci.data_flow.models.CreateApplicationDetails create_application_details: (required)
              Details to create an application.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Application.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.create_application`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.create_application(create_application_details, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.data.id
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_application(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def create_private_endpoint_and_wait_for_state(self, create_private_endpoint_details, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.create_private_endpoint` and waits for the :py:class:`~oci.data_flow.models.WorkRequest`
          to enter the given state(s).
  
          :param oci.data_flow.models.CreatePrivateEndpointDetails create_private_endpoint_details: (required)
              The parameters required to create a private endpoint.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.WorkRequest.status`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.create_private_endpoint`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.create_private_endpoint(create_private_endpoint_details, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.headers['opc-work-request-id']
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_work_request(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'status') and getattr(r.data, 'status').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def create_run_and_wait_for_state(self, create_run_details, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.create_run` and waits for the :py:class:`~oci.data_flow.models.Run` acted upon
          to enter the given state(s).
  
          :param oci.data_flow.models.CreateRunDetails create_run_details: (required)
              Details for creating a run of an application.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Run.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.create_run`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.create_run(create_run_details, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.data.id
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_run(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def delete_application_and_wait_for_state(self, application_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.delete_application` and waits for the :py:class:`~oci.data_flow.models.Application` acted upon
          to enter the given state(s).
  
          :param str application_id: (required)
              The unique ID for an application.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Application.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.delete_application`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          initial_get_result = self.client.get_application(application_id)
          operation_result = None
          try:
              operation_result = self.client.delete_application(application_id, **operation_kwargs)
          except oci.exceptions.ServiceError as e:
              if e.status == 404:
                  return WAIT_RESOURCE_NOT_FOUND
              else:
                  raise e
  
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  initial_get_result,
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  succeed_on_not_found=True,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def delete_private_endpoint_and_wait_for_state(self, private_endpoint_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.delete_private_endpoint` and waits for the :py:class:`~oci.data_flow.models.WorkRequest`
          to enter the given state(s).
  
          :param str private_endpoint_id: (required)
              The unique ID for a private endpoint.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.WorkRequest.status`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.delete_private_endpoint`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = None
          try:
              operation_result = self.client.delete_private_endpoint(private_endpoint_id, **operation_kwargs)
          except oci.exceptions.ServiceError as e:
              if e.status == 404:
                  return WAIT_RESOURCE_NOT_FOUND
              else:
                  raise e
  
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.headers['opc-work-request-id']
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_work_request(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'status') and getattr(r.data, 'status').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def delete_run_and_wait_for_state(self, run_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.delete_run` and waits for the :py:class:`~oci.data_flow.models.Run` acted upon
          to enter the given state(s).
  
          :param str run_id: (required)
              The unique ID for the run
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Run.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.delete_run`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          initial_get_result = self.client.get_run(run_id)
          operation_result = None
          try:
              operation_result = self.client.delete_run(run_id, **operation_kwargs)
          except oci.exceptions.ServiceError as e:
              if e.status == 404:
                  return WAIT_RESOURCE_NOT_FOUND
              else:
                  raise e
  
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  initial_get_result,
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  succeed_on_not_found=True,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def update_application_and_wait_for_state(self, update_application_details, application_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.update_application` and waits for the :py:class:`~oci.data_flow.models.Application` acted upon
          to enter the given state(s).
  
          :param oci.data_flow.models.UpdateApplicationDetails update_application_details: (required)
              Details for updating an application.
  
          :param str application_id: (required)
              The unique ID for an application.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Application.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.update_application`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.update_application(update_application_details, application_id, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.data.id
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_application(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def update_private_endpoint_and_wait_for_state(self, update_private_endpoint_details, private_endpoint_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.update_private_endpoint` and waits for the :py:class:`~oci.data_flow.models.WorkRequest`
          to enter the given state(s).
  
          :param oci.data_flow.models.UpdatePrivateEndpointDetails update_private_endpoint_details: (required)
              Details for updating a private endpoint.
  
          :param str private_endpoint_id: (required)
              The unique ID for a private endpoint.
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.WorkRequest.status`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.update_private_endpoint`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.update_private_endpoint(update_private_endpoint_details, private_endpoint_id, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.headers['opc-work-request-id']
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_work_request(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'status') and getattr(r.data, 'status').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  
      def update_run_and_wait_for_state(self, update_run_details, run_id, wait_for_states=[], operation_kwargs={}, waiter_kwargs={}):
          """
          Calls :py:func:`~oci.data_flow.DataFlowClient.update_run` and waits for the :py:class:`~oci.data_flow.models.Run` acted upon
          to enter the given state(s).
  
          :param oci.data_flow.models.UpdateRunDetails update_run_details: (required)
              Details for updating a run.
  
          :param str run_id: (required)
              The unique ID for the run
  
          :param list[str] wait_for_states:
              An array of states to wait on. These should be valid values for :py:attr:`~oci.data_flow.models.Run.lifecycle_state`
  
          :param dict operation_kwargs:
              A dictionary of keyword arguments to pass to :py:func:`~oci.data_flow.DataFlowClient.update_run`
  
          :param dict waiter_kwargs:
              A dictionary of keyword arguments to pass to the :py:func:`oci.wait_until` function. For example, you could pass ``max_interval_seconds`` or ``max_interval_seconds``
              as dictionary keys to modify how long the waiter function will wait between retries and the maximum amount of time it will wait
          """
          operation_result = self.client.update_run(update_run_details, run_id, **operation_kwargs)
          if not wait_for_states:
              return operation_result
  
          lowered_wait_for_states = [w.lower() for w in wait_for_states]
          wait_for_resource_id = operation_result.data.id
  
          try:
              waiter_result = oci.wait_until(
                  self.client,
                  self.client.get_run(wait_for_resource_id),
                  evaluate_response=lambda r: getattr(r.data, 'lifecycle_state') and getattr(r.data, 'lifecycle_state').lower() in lowered_wait_for_states,
                  **waiter_kwargs
              )
              result_to_return = waiter_result
  
              return result_to_return
          except Exception as e:
              raise oci.exceptions.CompositeOperationError(partial_results=[operation_result], cause=e)
  