o
    i                     @   sd   d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 eeZde jddfd	d
ZdS )    N)_get_test_id_from_item)PLUGIN_METRICS_V2)
get_logger)BenchmarkDurationData)InternalTestitemreturnc                 C   s   zHt | do| jd}|r|jsW d S | jdjj}i }t D ]\}}t ||r2t||||< q"t| }tdi |}t	j
||dd W d S  tyZ   tjd| dd Y d S w )Nfuncargs	benchmarkT)is_benchmarkz(Unable to set benchmark data for item %s)exc_info )hasattrr	   getstatsr   itemsgetattrr   r   r   set_benchmark_data	Exceptionlogdebug)r   fixturestat_objectdata_kwargs	data_attr
stats_attrtest_idbenchmark_datar   r   d/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/pytest/_benchmark_utils.py_set_benchmark_data_from_item   s"   

r   )pytest&ddtrace.contrib.internal.pytest._utilsr   3ddtrace.contrib.internal.pytest_benchmark.constantsr   ddtrace.internal.loggerr   1ddtrace.internal.test_visibility._benchmark_mixinr   $ddtrace.internal.test_visibility.apir   __name__r   Itemr   r   r   r   r   <module>   s    