관리-도구
편집 파일: middleware.cpython-38.pyc
U �Mgy � @ s� d Z ddlmZ ddlmZ ddlZddlmZ ddl m Z m Z m Z ddl mZ er�ddl mZ dd l mZ dd l mZ dd l mZ dd lmZ ed edef d�Ze d�ZedkZes�dd� Zn ddlmZ dd� Zdd� ZdS )z1 Create spans from Django middleware invocations � )�wraps)�VERSIONN)�OP)� ContextVar�transaction_from_function�capture_internal_exceptions)� TYPE_CHECKING)�Any)�Callable)�Optional)�TypeVar)�Span�F.)�boundZ$import_string_should_wrap_middleware)� � c C s t S �N)�object)�_� r �W/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/django/middleware.py�<lambda>$ � r r )� _asgi_middleware_mixin_factoryc sD ddl m} | j� � fdd�}|| _| jj��fdd�}|| j_d S )Nr )�basec s � | �}t �d �r t|| �}|S r )�%_import_string_should_wrap_middleware�get�_wrap_middleware)� dotted_path�rv)�old_import_stringr r � sentry_patched_import_string/ s z>patch_django_middlewares.<locals>.sentry_patched_import_stringc s* t �d� z� | |�W �S t �d� X d S )NTF)r �set)�args�kwargs)�old_load_middlewarer r � sentry_patched_load_middleware< s z@patch_django_middlewares.<locals>.sentry_patched_load_middleware)Zdjango.core.handlersr Z import_string� BaseHandlerZload_middleware)r r! r& r )r r% r �patch_django_middlewares) s r( c sl ddl m� � �fdd���fdd��G ���fdd�dt���}d D ] }t�|�rFt||t�|�� qF|S ) Nr )�DjangoIntegrationc sx t �� �� �}|d ks |js d S t| �}�}t| dd �}|rHd�||�}t jtj |� j d�}|� d|� |� d�� |S )N�__name__z{}.{})�op�name�originzdjango.function_namezdjango.middleware_name) � sentry_sdkZ get_clientZget_integrationZmiddleware_spansr �getattr�formatZ start_spanr ZMIDDLEWARE_DJANGOr- Zset_tag)� old_methodZ integrationZ function_name� descriptionZfunction_basename�middleware_span)r) �middleware_namer r �_check_middleware_spanK s � z0_wrap_middleware.<locals>._check_middleware_spanc s^ t � �N �� fdd�}zt� �|�}� j|_W n tk rB Y nX |W 5 Q R � S Q R X � S )Nc s@ � ��}|d kr�| |�S |� �| |�W 5 Q R � S Q R X d S r r )r# r$ r3 )r5 r1 r r �sentry_wrapped_methodf s zL_wrap_middleware.<locals>._get_wrapped_method.<locals>.sentry_wrapped_method)r r �__self__� Exception)r1 r6 �r5 �r1 r �_get_wrapped_methodb s z-_wrap_middleware.<locals>._get_wrapped_methodc sT e Zd Ze�dd�Zeo"e�dd�Zd � �fdd� Z�fdd �Z�fd d �Z � Z S ) z2_wrap_middleware.<locals>.SentryWrappingMiddleware� sync_capableT� async_capableFNc sF |r�|f|�|�| _ n �||�| _ || _d | _| jrBt� �|� d S r )�_inner� get_response� _call_methodr= �super�__init__)�selfr? r# r$ )� __class__� middlewarer r rB � s z;_wrap_middleware.<locals>.SentryWrappingMiddleware.__init__c s0 |dkrt � �t| j|�}� |�}|| j|< |S )N)�process_requestZ process_viewZprocess_template_response�process_responseZprocess_exception)�AttributeErrorr/ r> �__dict__)rC � method_namer1 r )r; r r � __getattr__� s z>_wrap_middleware.<locals>.SentryWrappingMiddleware.__getattr__c s| t | d�r | �� r | j||�S | j}|d kr:| jj | _}� |d�}|d krV|||�S |� |||�W 5 Q R � S Q R X d S )N�async_route_checkr: )�hasattrrL Z __acall__r@ r> �__call__)rC r# r$ �fr3 r9 r r rN � s z;_wrap_middleware.<locals>.SentryWrappingMiddleware.__call__)N) r* � __module__� __qualname__r/ r< � DJANGO_SUPPORTS_ASYNC_MIDDLEWAREr= rB rK rN � __classcell__r )r5 r; rE )rD r �SentryWrappingMiddleware} s � rT )r* rP rQ )Z sentry_sdk.integrations.djangor) r rM �setattrr/ )rE r4 rT �attrr )r) r5 r; rE r4 r r G s �6 r ) �__doc__� functoolsr �djangor ZDJANGO_VERSIONr. Zsentry_sdk.constsr Zsentry_sdk.utilsr r r �typingr r r r r Zsentry_sdk.tracingr r r rR r Zasgir( r r r r r �<module> s, �