o
    Á¿i¥  ã                   @   sp   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efdd	„Zdeeef fd
d„Zdd„ Zdd„ ZdS )é    N)Úwrap_function_wrapper)Úunwrap)Úconfigé   )Ú_wrap_create_engineÚreturnc                   C   s   t tddƒS )NÚ__version__Ú )ÚgetattrÚ
sqlalchemy© r   r   ú]/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/sqlalchemy/patch.pyÚget_version
   s   r   c                   C   s   ddiS )Nr   z>=1.3r   r   r   r   r   Ú_supported_versions   s   r   c                  C   s`   t tjddƒr	d S dtj_tddtƒ tddtƒ tjr.ddlm	}  dd	l
m} | |ƒ d S d S )
NÚ__datadog_patchFTr   Úcreate_enginezsqlalchemy.enginer   )Ú"_set_metric_iast_instrumented_sink)ÚVULN_SQL_INJECTION)r
   r   Úenginer   Ú_wr   Ú
asm_configÚ_iast_enabledÚddtrace.appsec._iast._metricsr   Úddtrace.appsec._iast.constantsr   )r   r   r   r   r   Úpatch   s   ür   c                   C   s4   t tjddƒrdtj_ttdƒ ttjdƒ d S d S )Nr   Fr   )r
   r   r   r   r   r   r   r   r   Úunpatch"   s
   
ýr   )r   Úwraptr   r   Ú$ddtrace.contrib.internal.trace_utilsr   Úddtrace.internal.settings.asmr   r   r   r   Ústrr   Údictr   r   r   r   r   r   r   Ú<module>   s    