o
    Á¿i  ã                   @   sV   d dl mZ G dd„ dejƒZG dd„ deejƒZG dd„ deƒZG dd	„ d	eƒZd
S )é    )Údbapic                       s,   e Zd ZdZ‡ fdd„Z‡ fdd„Z‡  ZS )ÚPsycopg3TracedCursorz"TracedCursor for psycopg instancesc                    s   t t| ƒ |||¡ d S )N)Úsuperr   Ú__init__)ÚselfÚcursorÚpinÚcfgÚargsÚkwargs©Ú	__class__© ú[/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/psycopg/cursor.pyr      s   zPsycopg3TracedCursor.__init__c                    sJ   |j jdks|j jdkr| | j¡}tt| ƒj|||||g|¢R i |¤ŽS )NÚSQLÚComposed)r   Ú__name__Ú	as_stringÚ__wrapped__r   r   Ú_trace_method)r   ÚmethodÚnameÚresourceÚ
extra_tagsÚdbm_propagatorr
   r   r   r   r   r   
   s   

ÿÿÿz"Psycopg3TracedCursor._trace_method)r   Ú
__module__Ú__qualname__Ú__doc__r   r   Ú__classcell__r   r   r   r   r      s    r   c                   @   ó   e Zd ZdZdS )ÚPsycopg3FetchTracedCursorz%Psycopg3FetchTracedCursor for psycopgN©r   r   r   r   r   r   r   r   r       ó    r    c                   @   r   )ÚPsycopg2TracedCursorzTracedCursor for psycopg2Nr!   r   r   r   r   r#      r"   r#   c                   @   r   )ÚPsycopg2FetchTracedCursorzFetchTracedCursor for psycopg2Nr!   r   r   r   r   r$      r"   r$   N)Úddtrace.contribr   ÚTracedCursorr   ÚFetchTracedCursorr    r#   r$   r   r   r   r   Ú<module>   s
    