관리-도구
편집 파일: caching.cpython-38.pyc
U �Mg0 � @ s� d dl Z d dlmZ d dlmZmZ d dlmZ d dl m Z d dl m Z d dlZd dlmZmZ d dlmZmZ er�d d lmZ d d lmZ d d lmZ d d ddgZdd� Zdd� Zddd�Zdd� Z dd� Z dS )� N)� TYPE_CHECKING)� _get_safe_key�_key_as_string)� parse_url)�VERSION)� CacheHandler)�OP�SPANDATA)�capture_internal_exceptions�ensure_integration_enabled)�Any)�Callable)�Optional�setZset_many�getZget_manyc C s t t| ||��S �N)r r )� method_name�args�kwargs� r �T/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/django/caching.py�_get_span_description s r c s\ ddl m� t����t� ��� fdd���t���������fdd��}t��|� d S )Nr ��DjangoIntegrationc sR |� d�}| }|rtjntj} t|||�} tj| | � jd���} |||�} t� �� |d k rj| � t j |� |d k r�| � t j |� t |||�} | d k r�| � t j| � d }|r�| r�tt| ��}| � t jd� n| � t jd� n<ztt|d ��}W n& tk �r tt|d ��}Y nX |d k �r.| � t j|� W 5 Q R X | W 5 Q R � S Q R X d S )Nr )�op�name�originTF� r )� startswithr Z CACHE_PUTZ CACHE_GETr � sentry_sdkZ start_spanr r �set_datar ZNETWORK_PEER_ADDRESSZNETWORK_PEER_PORTr Z CACHE_KEY�len�strZ CACHE_HIT� IndexErrorZCACHE_ITEM_SIZE)�cacher �original_methodr r �address�portZis_set_operationZis_get_operationr � description�span�value�keyZ item_sizer r r �_instrument_call* s@ � z-_patch_cache_method.<locals>._instrument_callc s � ���| |���S r r )r r )r, r& r$ r r% r' r r � sentry_method[ s �z*_patch_cache_method.<locals>.sentry_method)� sentry_sdk.integrations.djangor �getattrr � functools�wraps�setattr)r$ r r& r' r- r )r r, r& r$ r r% r' r �_patch_cache_method$ s 0r3 c C s, t | d�s(tD ]}t| |||� qd| _d S )N�_sentry_patchedT)�hasattr�METHODS_TO_INSTRUMENTr3 r4 )r$ r& r' r r r r � _patch_cachee s r7 c C sp | � d�}t|t�sdS d|krPt|�}d�|jp4d|jp<d|jpDd�}|j}n|}d }||d k rjt |�nd fS )NZLOCATION)NNz://z {}://{}{}� ) r � isinstancer" �urlparse�format�scheme�hostname�pathr' �int)�settings�location� parsed_urlr& r' r r r �_get_address_portm s �rC c sx ddl m� ttd�sttdk rJtj�t���� �fdd��} | t_dt_n*tj �t���� �fdd ��}|t_ dt_d S ) Nr r r4 )� � c sX �| |�}t �� �� �}|d k rT|jrTddlm} t|j|p>d �\}}t|||� |S )Nr )r@ �default) r � get_client�get_integration� cache_spansZ django.confr@ rC ZCACHESr7 )�self�aliasr$ � integrationr@ r& r' )r �original_get_itemr r �sentry_get_item� s � z&patch_caching.<locals>.sentry_get_itemTc sL �| |�}t �� �� �}|d k rH|jrHt| j|p2d �\}}t|||� |S )NrF )r rG rH rI rC r@ r7 )rJ rK r$ rL r& r' )r �original_create_connectionr r �sentry_create_connection� s z/patch_caching.<locals>.sentry_create_connection) r. r r5 r �DJANGO_VERSION� __getitem__r0 r1 r4 �create_connection)rN rP r )r rO rM r � patch_caching� s rT )NN) r0 �typingr Z#sentry_sdk.integrations.redis.utilsr r Z urllib3.utilr r: �djangor rQ Zdjango.core.cacher r Zsentry_sdk.constsr r Zsentry_sdk.utilsr r r r r r6 r r3 r7 rC rT r r r r �<module> s, �A