o
    ɶdmr                     @   s   d dl mZ d dlmZ d dlmZ d dlmZmZ d dlm	Z	 d dl
mZmZ d dlmZ d dlmZmZmZ d	d
lmZ edZG dd deZdS )    )absolute_import)requests)six)retrycircuit_breaker)
BaseClient)get_config_value_or_defaultvalidate_config)Signer)Sentinel#get_signer_from_authentication_typeAUTHENTICATION_TYPE_FIELD_NAME   )apm_traces_type_mappingZMissingc                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )TraceClienta  
    Use the Application Performance Monitoring Trace Explorer API to query traces and associated spans in Trace Explorer. For more information, see [Application Performance Monitoring](https://docs.oracle.com/iaas/application-performance-monitoring/index.html).
    c              
   K   s  t ||dd d|v r|d }n"t|v rt|}nt|d |d |d |dt|d|dd	}d
|dddi |dd|dtj|dd}d|v rY|d|d< |ddu retj|d< d|v rp|d|d< t	d||t
fi || _|d| _|d| _dS )a  
        Creates a new service client

        :param dict config:
            Configuration keys and values as per `SDK and Tool Configuration <https://docs.cloud.oracle.com/Content/API/Concepts/sdkconfig.htm>`__.
            The :py:meth:`~oci.config.from_file` method can be used to load configuration from a file. Alternatively, a ``dict`` can be passed. You can validate_config
            the dict using :py:meth:`~oci.config.validate_config`

        :param str service_endpoint: (optional)
            The endpoint of the service to call using this client. For example ``https://iaas.us-ashburn-1.oraclecloud.com``. If this keyword argument is
            not provided then it will be derived using the region in the config parameter. You should only provide this keyword argument if you have an explicit
            need to specify a service endpoint.

        :param timeout: (optional)
            The connection and read timeouts for the client. The default values are connection timeout 10 seconds and read timeout 60 seconds. This keyword argument can be provided
            as a single float, in which case the value provided is used for both the read and connection timeouts, or as a tuple of two floats. If
            a tuple is provided then the first value is used as the connection timeout and the second value as the read timeout.
        :type timeout: float or tuple(float, float)

        :param signer: (optional)
            The signer to use when signing requests made by the service client. The default is to use a :py:class:`~oci.signer.Signer` based on the values
            provided in the config parameter.

            One use case for this parameter is for `Instance Principals authentication <https://docs.cloud.oracle.com/Content/Identity/Tasks/callingservicesfrominstances.htm>`__
            by passing an instance of :py:class:`~oci.auth.signers.InstancePrincipalsSecurityTokenSigner` as the value for this keyword argument
        :type signer: :py:class:`~oci.signer.AbstractBaseSigner`

        :param obj retry_strategy: (optional)
            A retry strategy to apply to all calls made by this service client (i.e. at the client level). There is no retry strategy applied by default.
            Retry strategies can also be applied at the operation level by passing a ``retry_strategy`` keyword argument as part of calling the operation.
            Any value provided at the operation level will override whatever is specified at the client level.

            This should be one of the strategies available in the :py:mod:`~oci.retry` module. A convenience :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY`
            is also available. The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.

        :param obj circuit_breaker_strategy: (optional)
            A circuit breaker strategy to apply to all calls made by this service client (i.e. at the client level).
            This client uses :py:data:`~oci.circuit_breaker.DEFAULT_CIRCUIT_BREAKER_STRATEGY` as default if no circuit breaker strategy is provided.
            The specifics of circuit breaker strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/circuit_breakers.html>`__.

        :param function circuit_breaker_callback: (optional)
            Callback function to receive any exceptions triggerred by the circuit breaker.

        :param bool client_level_realm_specific_endpoint_template_enabled: (optional)
            A boolean flag to indicate whether or not this client should be created with realm specific endpoint template enabled or disable. By default, this will be set as None.

        :param allow_control_chars: (optional)
            allow_control_chars is a boolean to indicate whether or not this client should allow control characters in the response object. By default, the client will not
            allow control characters to be in the response object.
        signer)r   tenancyuserfingerprintZkey_filepass_phraseZkey_content)r   r   r   Zprivate_key_file_locationr   Zprivate_key_contentTservice_endpointz	/20200630z2https://apm-trace.{region}.oci.{secondLevelDomain}skip_deserializationFcircuit_breaker_strategy5client_level_realm_specific_endpoint_template_enabled)Zregional_clientr   	base_pathZservice_endpoint_templateZ#service_endpoint_template_per_realmr   r   r   timeoutNallow_control_charstraceretry_strategycircuit_breaker_callback)r	   getr   r   r
   r   r   ZGLOBAL_CIRCUIT_BREAKER_STRATEGYZ DEFAULT_CIRCUIT_BREAKER_STRATEGYr   r   base_clientr   r   )selfconfigkwargsr   Zbase_client_init_kwargs r%   ?usr/lib/python3.10/site-packages/oci/apm_traces/trace_client.py__init__   s>   3





zTraceClient.__init__c                      ddg}d}d}d}d}g d  fdd	t |D }	|	r%td
|	d|i}
dd t |
D }
t |
D ]\}}|du sNt|t jrUt| dkrUtd|q8d|i}dd t |D }dd|	dt
d}dd t |D }| jj|	d| jd}|rt|tjs| j| || j |j| jj|||
||d|	d|||dS | jj|||
||d|	d|||d
S )a  
        Gets the aggregated snapshot identified by trace ID.


        :param str apm_domain_id: (required)
            The APM Domain ID the request is intended for.

        :param str trace_key: (required)
            Unique Application Performance Monitoring trace identifier (traceId).

        :param str opc_request_id: (optional)
            Unique Oracle-assigned identifier for the request.  If you need to contact Oracle about a
            particular request, please provide the request ID.

        :param obj retry_strategy: (optional)
            A retry strategy to apply to this specific operation/call. This will override any retry strategy set at the client-level.

            This should be one of the strategies available in the :py:mod:`~oci.retry` module. This operation will not retry by default, users can also use the convenient :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY` provided by the SDK to enable retries for it.
            The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.

            To have this operation explicitly not perform any retries, pass an instance of :py:class:`~oci.retry.NoneRetryStrategy`.

        :param bool allow_control_chars: (optional)
            allow_control_chars is a boolean to indicate whether or not this request should allow control characters in the response object.
            By default, the response will not allow control characters in strings

        :return: A :class:`~oci.response.Response` object with data of type :class:`~oci.apm_traces.models.AggregatedSnapshot`
        :rtype: :class:`~oci.response.Response`

        :example:
        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/latest/apmtraces/get_aggregated_snapshot.py.html>`__ to see an example of how to use get_aggregated_snapshot API.
        traceKeyapmDomainIdz)/traces/{traceKey}/aggregatedSnapshotDataGETget_aggregated_snapshotzjhttps://docs.oracle.com/iaas/api/#/en/apm-trace-explorer/20200630/AggregatedSnapshot/GetAggregatedSnapshotr   r   opc_request_idc                       g | ]}| vr|qS r%   r%   .0_keyZexpected_kwargsr%   r&   
<listcomp>       z7TraceClient.get_aggregated_snapshot.<locals>.<listcomp>z0get_aggregated_snapshot got unknown kwargs: {!r}c                 S      i | ]\}}|t ur||qS r%   missingr1   kvr%   r%   r&   
<dictcomp>       z7TraceClient.get_aggregated_snapshot.<locals>.<dictcomp>Nr   7Parameter {} cannot be None, whitespace or empty stringc                 S   &   i | ]\}}|t ur|d ur||qS Nr7   r9   r%   r%   r&   r<         & application/jsonr.   acceptzcontent-typezopc-request-idc                 S   r?   r@   r7   r9   r%   r%   r&   r<      rA   r   Zoperation_retry_strategyZclient_retry_strategyZAggregatedSnapshotr   
resource_pathmethodpath_paramsquery_paramsheader_paramsZresponse_typer   operation_nameapi_reference_linkrequired_argumentsr   Ziterkeys
ValueErrorformatZ	iteritems
isinstanceZstring_typeslenstripr    r8   r!   Zget_preferred_retry_strategyr   r   ZNoneRetryStrategyZadd_opc_client_retries_headerZadd_circuit_breaker_callbackr   Zmake_retrying_callZcall_apir"   apm_domain_id	trace_keyr$   rN   rG   rH   rL   rM   extra_kwargsrI   r:   r;   rJ   rK   r   r%   r3   r&   r,   r   v   "$
z#TraceClient.get_aggregated_snapshotc                    s  g d}d}d}d}d}	g d  fddt |D }
|
r%td	|
||d
}dd t |D }t |D ]\}}|du sOt|t jrVt| dkrVtd|q9d|i}dd t |D }dd|	dt
d}dd t |D }| jj|	d| jd}|rt|tjs| j| || j |j| jj|||||d|	d||	|dS | jj|||||d|	d||	|d
S )a  
        Gets the span details identified by spanId.


        :param str apm_domain_id: (required)
            The APM Domain ID the request is intended for.

        :param str span_key: (required)
            Unique Application Performance Monitoring span identifier (spanId).

        :param str trace_key: (required)
            Unique Application Performance Monitoring trace identifier (traceId).

        :param str opc_request_id: (optional)
            Unique Oracle-assigned identifier for the request.  If you need to contact Oracle about a
            particular request, please provide the request ID.

        :param obj retry_strategy: (optional)
            A retry strategy to apply to this specific operation/call. This will override any retry strategy set at the client-level.

            This should be one of the strategies available in the :py:mod:`~oci.retry` module. This operation will not retry by default, users can also use the convenient :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY` provided by the SDK to enable retries for it.
            The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.

            To have this operation explicitly not perform any retries, pass an instance of :py:class:`~oci.retry.NoneRetryStrategy`.

        :param bool allow_control_chars: (optional)
            allow_control_chars is a boolean to indicate whether or not this request should allow control characters in the response object.
            By default, the response will not allow control characters in strings

        :return: A :class:`~oci.response.Response` object with data of type :class:`~oci.apm_traces.models.Span`
        :rtype: :class:`~oci.response.Response`

        :example:
        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/latest/apmtraces/get_span.py.html>`__ to see an example of how to use get_span API.
        )spanKeyr)   r*   z/spans/{traceKey}/{spanKey}r+   get_spanzNhttps://docs.oracle.com/iaas/api/#/en/apm-trace-explorer/20200630/Span/GetSpanr-   c                    r/   r%   r%   r0   r3   r%   r&   r4     r5   z(TraceClient.get_span.<locals>.<listcomp>z!get_span got unknown kwargs: {!r})rZ   r)   c                 S   r6   r%   r7   r9   r%   r%   r&   r<     r=   z(TraceClient.get_span.<locals>.<dictcomp>Nr   r>   r*   c                 S   r?   r@   r7   r9   r%   r%   r&   r<   !  rA   rB   r.   rC   c                 S   r?   r@   r7   r9   r%   r%   r&   r<   (  rA   r   rE   ZSpanr   rF   rO   )r"   rV   Zspan_keyrW   r$   rN   rG   rH   rL   rM   rX   rI   r:   r;   rJ   rK   r   r%   r3   r&   r[      sx   %$
zTraceClient.get_spanc                    r(   )aW  
        Gets the trace details identified by traceId.


        :param str apm_domain_id: (required)
            The APM Domain ID the request is intended for.

        :param str trace_key: (required)
            Unique Application Performance Monitoring trace identifier (traceId).

        :param str opc_request_id: (optional)
            Unique Oracle-assigned identifier for the request.  If you need to contact Oracle about a
            particular request, please provide the request ID.

        :param obj retry_strategy: (optional)
            A retry strategy to apply to this specific operation/call. This will override any retry strategy set at the client-level.

            This should be one of the strategies available in the :py:mod:`~oci.retry` module. This operation will not retry by default, users can also use the convenient :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY` provided by the SDK to enable retries for it.
            The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.

            To have this operation explicitly not perform any retries, pass an instance of :py:class:`~oci.retry.NoneRetryStrategy`.

        :param bool allow_control_chars: (optional)
            allow_control_chars is a boolean to indicate whether or not this request should allow control characters in the response object.
            By default, the response will not allow control characters in strings

        :return: A :class:`~oci.response.Response` object with data of type :class:`~oci.apm_traces.models.Trace`
        :rtype: :class:`~oci.response.Response`

        :example:
        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/latest/apmtraces/get_trace.py.html>`__ to see an example of how to use get_trace API.
        r)   r*   z/traces/{traceKey}r+   	get_tracezPhttps://docs.oracle.com/iaas/api/#/en/apm-trace-explorer/20200630/Trace/GetTracer-   c                    r/   r%   r%   r0   r3   r%   r&   r4   z  r5   z)TraceClient.get_trace.<locals>.<listcomp>z"get_trace got unknown kwargs: {!r}c                 S   r6   r%   r7   r9   r%   r%   r&   r<     r=   z)TraceClient.get_trace.<locals>.<dictcomp>Nr   r>   c                 S   r?   r@   r7   r9   r%   r%   r&   r<     rA   rB   r.   rC   c                 S   r?   r@   r7   r9   r%   r%   r&   r<     rA   r   rE   ZTracer   rF   rO   rU   r%   r3   r&   r\   L  rY   zTraceClient.get_tracec                    s  ddg}d}d}d}d}g d  fdd	t |D }	|	r%td
|	d|i}
dd t |
D }
t |
D ]\}}|du sNt|t jrUt| dkrUtd|q8||	dt
|	dt
|	dt
d}dd t |D }dd|	dt
d}dd t |D }| jj|	d| jd}|rt|tjs| j| || j |j| jj|||
||d|	d|||dS | jj|||
||d|	d|||d
S )a	  
        Gets the trace snapshots data identified by trace ID.


        :param str apm_domain_id: (required)
            The APM Domain ID the request is intended for.

        :param str trace_key: (required)
            Unique Application Performance Monitoring trace identifier (traceId).

        :param str opc_request_id: (optional)
            Unique Oracle-assigned identifier for the request.  If you need to contact Oracle about a
            particular request, please provide the request ID.

        :param bool is_summarized: (optional)
            If enabled, then only span level details will be sent.

        :param str thread_id: (optional)
            Thread id for which snapshots needs to be retrieved. This is an identifier of a thread, and is a positive long number generated when when a thread is created.

        :param str snapshot_time: (optional)
            Epoch time of snapshot.

        :param obj retry_strategy: (optional)
            A retry strategy to apply to this specific operation/call. This will override any retry strategy set at the client-level.

            This should be one of the strategies available in the :py:mod:`~oci.retry` module. This operation will not retry by default, users can also use the convenient :py:data:`~oci.retry.DEFAULT_RETRY_STRATEGY` provided by the SDK to enable retries for it.
            The specifics of the default retry strategy are described `here <https://docs.oracle.com/en-us/iaas/tools/python/latest/sdk_behaviors/retries.html>`__.

            To have this operation explicitly not perform any retries, pass an instance of :py:class:`~oci.retry.NoneRetryStrategy`.

        :param bool allow_control_chars: (optional)
            allow_control_chars is a boolean to indicate whether or not this request should allow control characters in the response object.
            By default, the response will not allow control characters in strings

        :return: A :class:`~oci.response.Response` object with data of type :class:`~oci.apm_traces.models.TraceSnapshot`
        :rtype: :class:`~oci.response.Response`

        :example:
        Click `here <https://docs.cloud.oracle.com/en-us/iaas/tools/python-sdk-examples/latest/apmtraces/get_trace_snapshot.py.html>`__ to see an example of how to use get_trace_snapshot API.
        r)   r*   z/traces/{traceKey}/snapshotDatar+   get_trace_snapshotz`https://docs.oracle.com/iaas/api/#/en/apm-trace-explorer/20200630/TraceSnapshot/GetTraceSnapshot)r   r   r.   is_summarized	thread_idsnapshot_timec                    r/   r%   r%   r0   r3   r%   r&   r4     r5   z2TraceClient.get_trace_snapshot.<locals>.<listcomp>z+get_trace_snapshot got unknown kwargs: {!r}c                 S   r6   r%   r7   r9   r%   r%   r&   r<     r=   z2TraceClient.get_trace_snapshot.<locals>.<dictcomp>Nr   r>   r^   r_   r`   )r*   ZisSummarizedZthreadIdZsnapshotTimec                 S   r?   r@   r7   r9   r%   r%   r&   r<     rA   rB   r.   rC   c                 S   r?   r@   r7   r9   r%   r%   r&   r<     rA   r   rE   ZTraceSnapshotr   rF   rO   rU   r%   r3   r&   r]     s|   +$



zTraceClient.get_trace_snapshotN)	__name__
__module____qualname____doc__r'   r,   r[   r\   r]   r%   r%   r%   r&   r      s    Xkokr   N)
__future__r   Zoci._vendorr   r   ocir   r   Zoci.base_clientr   Z
oci.configr   r	   Z
oci.signerr
   Zoci.utilr   r   r   Zmodelsr   r8   objectr   r%   r%   r%   r&   <module>   s   