ruạṛ
a N��f� � @ s( d dl Z d dlmZ G dd� de�ZdS )� N)�WAIT_RESOURCE_NOT_FOUNDc @ s� e Zd ZdZdd� Zg i i fdd�Zg i i fdd�Zg i i fdd �Zg i i fd d�Zg i i fdd �Z g i i fdd�Z g i i fdd�Zg i i fdd�Zg i i fdd�Z g i i fdd�Zg i i fdd�Zg i i fdd�Zg i i fdd�Zg i i fdd�Zg i i fd d!�Zg i i fd"d#�Zg i i fd$d%�Zg i i fd&d'�Zg i i fd(d)�Zg i i fd*d+�Zg i i fd,d-�Zg i i fd.d/�Zg i i fd0d1�Zg i i fd2d3�Zg i i fd4d5�Zg i i fd6d7�Zd8S )9�$DataScienceClientCompositeOperationsa� This class provides a wrapper around :py:class:`~oci.data_science.DataScienceClient` 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 c K s || _ dS )z� Creates a new DataScienceClientCompositeOperations object :param DataScienceClient client: The service client which will be wrapped by this object N)�client)�selfr �kwargs� r ��/sparta/input/_build_configuration/image_build+validate/lib/bmcenv/lib64/python3.9/site-packages/oci/data_science/data_science_client_composite_operations.py�__init__ s z-DataScienceClientCompositeOperations.__init__c s� | j j|fi |��}|s|S dd� |D �� |jj}z4tj| j | j �|�fd� fdd�i|��}|}|W S ttfy� } z6|j s�d|_ |j d |_ tj j|g|d��W Y d }~n>d }~0 ty� } ztj j|g|d��W Y d }~n d }~0 0 d S ) a Calls :py:func:`~oci.data_science.DataScienceClient.activate_model` and waits for the :py:class:`~oci.data_science.models.Model` acted upon to enter the given state(s). :param str model_id: (required) The `OCID`__ of the model. __ https://docs.cloud.oracle.com/iaas/Content/General/Concepts/identifiers.htm :param list[str] wait_for_states: An array of states to wait on. These should be valid values for :py:attr:`~oci.data_science.models.Model.lifecycle_state` :param dict operation_kwargs: A dictionary of keyword arguments to pass to :py:func:`~oci.data_science.DataScienceClient.activate_model` :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 c S s g | ]}|� � �qS r ��lower��.0�wr r r � <listcomp>2 � zZDataScienceClientCompositeOperations.activate_model_and_wait_for_state.<locals>.<listcomp>�evaluate_responsec s t | jd�ot | jd��� � v S �N�lifecycle_state��getattr�datar ��r��lowered_wait_for_statesr r �<lambda>9 r zXDataScienceClientCompositeOperations.activate_model_and_wait_for_state.<locals>.<lambda>�� �a This composite operation is currently not supported in the SDK. Please use the operation from the service client and use waiters as an alternative. For more information on waiters, visit: "https://docs.oracle.com/en-us/iaas/tools/python/latest/api/waiters.html"��partial_results�causeN) r �activate_modelr �id�oci� wait_until� get_model� NameError� TypeError�args� exceptions�CompositeOperationError� Exception� r �model_id�wait_for_states�operation_kwargs� waiter_kwargs�operation_result� waiter_result�result_to_return�er r r �!activate_model_and_wait_for_state s. � ��&zFDataScienceClientCompositeOperations.activate_model_and_wait_for_statec s� | j j|fi |��}|s|S dd� |D �� |jd }z4tj| j | j �|�fd� fdd�i|��}|}|W S ty� } ztjj|g| d��W Y d} ~ n d} ~ 0 0 dS ) aC Calls :py:func:`~oci.data_science.DataScienceClient.activate_model_deployment` and waits for the :py:class:`~oci.data_science.models.WorkRequest` to enter the given state(s). :param str model_deployment_id: (required) The `OCID`__ of the model deployment. __ https://docs.cloud.oracle.com/iaas/Content/General/Concepts/identifiers.htm :param list[str] wait_for_states: An array of states to wait on. These should be valid values for :py:attr:`~oci.data_science.models.WorkRequest.status` :param dict operation_kwargs: A dictionary of keyword arguments to pass to :py:func:`~oci.data_science.DataScienceClient.activate_model_deployment` :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 c S s g | ]}|� � �qS r r r r r r r _ r zeDataScienceClientCompositeOperations.activate_model_deployment_and_wait_for_state.<locals>.<listcomp>�opc-work-request-idr c s t | jd�ot | jd��� � v S �N�statusr r r r r r f r zcDataScienceClientCompositeOperations.activate_model_deployment_and_wait_for_state.<locals>.<lambda>r N) r �activate_model_deployment�headersr$ r% �get_work_requestr, r* r+ � r �model_deployment_idr/ r0 r1 r2 �wait_for_resource_idr3 r4 r5 r r r �,activate_model_deployment_and_wait_for_stateG s$ � ��zQDataScienceClientCompositeOperations.activate_model_deployment_and_wait_for_statec s� | j j|fi |��}|s|S dd� |D �� |jd }z4tj| j | j �|�fd� fdd�i|��}|}|W S ty� } ztjj|g| d��W Y d} ~ n d} ~ 0 0 dS ) aC Calls :py:func:`~oci.data_science.DataScienceClient.activate_notebook_session` and waits for the :py:class:`~oci.data_science.models.WorkRequest` to enter the given state(s). :param str notebook_session_id: (required) The `OCID`__ of the notebook session. __ https://docs.cloud.oracle.com/iaas/Content/General/Concepts/identifiers.htm :param list[str] wait_for_states: An array of states to wait on. These should be valid values for :py:attr:`~oci.data_science.models.WorkRequest.status` :param dict operation_kwargs: A dictionary of keyword arguments to pass to :py:func:`~oci.data_science.DataScienceClient.activate_notebook_session` :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 c S s g | ]}|� � �qS r r r r r r r � r zeDataScienceClientCompositeOperations.activate_notebook_session_and_wait_for_state.<locals>.<listcomp>r7 r c s t | jd�ot | jd��� � v S r8 r r r r r r � r zcDataScienceClientCompositeOperations.activate_notebook_session_and_wait_for_state.<locals>.<lambda>r N) r �activate_notebook_sessionr; r$ r% r<