관리-도구
편집 파일: asgi.cpython-38.pyc
U �Mg� � @ s* d Z ddlZddlZddlZddlmZ ddlZddlmZ ddl m Z ddl m Z ddl mZmZ ddlmZ er�dd lmZmZmZmZ dd lmZ dd lmZ dd lmZm Z ed edef d�Z e ed�r�ej Z ej Z nej Z d d d�dd�Z dd� Z!dd� Z"dd� Z#dd� Z$dd � Z%d d � Z&dS ) z� Instrumentation for Django 3.0 Since this file contains `async def` it is conditionally imported in `sentry_sdk.integrations.django` (depending on the existence of `django.core.handlers.asgi`. � N)� WSGIRequest)�OP)�SentryAsgiMiddleware)�should_send_default_pii)�capture_internal_exceptions�ensure_integration_enabled)� TYPE_CHECKING)�Any�Callable�Union�TypeVar)� ASGIRequest)� HttpResponse)�Event�EventProcessor�_F.)�bound�markcoroutinefunction)�func�returnc C s t jj| _| S �N)�asyncioZ coroutinesZ _is_coroutine)r � r �Q/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/django/asgi.pyr 1 s c s � fdd�}|S )Nc sr ddl m}m} � d kr | S t� �tkr,| S t� � |� ��| � W 5 Q R X t� rnt� � |� | � W 5 Q R X | S )Nr )�DjangoRequestExtractor�_set_user_info)� sentry_sdk.integrations.djangor r �typer r Zextract_into_eventr )�event�hintr r ��requestr r � asgi_request_event_processor8 s zH_make_asgi_request_event_processor.<locals>.asgi_request_event_processorr )r! r" r r r �"_make_asgi_request_event_processor6 s r# c s\ ddl m� �j�� ��fdd�}|�_t�d�}|rX�j�t� ���fdd��}|�_d S )Nr ��DjangoIntegrationc � sZ t �� �� �}|d kr*�| |||�I d H S t��| ��d� j|jd�j}||||�I d H S )NT�Zunsafe_context_dataZ span_origin�http_methods_to_capture)� sentry_sdk� get_client�get_integrationr �__get__�originr' Z _run_asgi3)�self�scope�receive�send� integration� middleware�r% �cls�old_appr r �sentry_patched_asgi_handler[ s �zCpatch_django_asgi_handler_impl.<locals>.sentry_patched_asgi_handler�create_requestc s2 � | f|�|�\}}t �� }|�t|�� ||fS r )r( �get_isolation_scopeZadd_event_processorr# )r- �args�kwargsr! Zerror_responser. )�old_create_requestr r � sentry_patched_create_requestp s zEpatch_django_asgi_handler_impl.<locals>.sentry_patched_create_request)r r% �__call__�hasattrr7 r )r4 r6 Zmodern_django_asgi_supportr< r )r% r4 r5 r; r � patch_django_asgi_handler_implT s r? c s | j �� �fdd�}|| _ d S )Nc � s � |� �| |�I d H S r r )r- r! ��_before_get_responseZold_get_response_asyncr r �!sentry_patched_get_response_async� s zCpatch_get_response_async.<locals>.sentry_patched_get_response_async)Zget_response_async)r4 rA rB r r@ r �patch_get_response_async| s rC c sH dd l }ddlm� |jdk r<�j�� ��fdd�}|�_nt�� d S )Nr r$ z3.0.0c � s^ t �� ���}|d kr(�� ||�I d H S t��� fdd�d�j|jd�}|� j�||�I d H S )Nc s �� �� �S r )r+ )Z_scope)r4 r5 r- r r �<lambda>� � zWpatch_channels_asgi_handler_impl.<locals>.sentry_patched_asgi_handler.<locals>.<lambda>Tr& )r( r) r* r r, r' r. )r- r/ r0 r1 r2 r3 �r- r r6 � s �zEpatch_channels_asgi_handler_impl.<locals>.sentry_patched_asgi_handler)�channelsr r% � __version__r= r? )r4 rG r6 r r3 r � patch_channels_asgi_handler_impl� s rI c s( ddl m� t���� �fdd��}|S )Nr r$ c � sb t �� }|jd k r |j�� t jtj| jj� j d��$ �| f|�|�I d H W 5 Q R � S Q R X d S )N)�op�namer, ) r( r8 ZprofileZupdate_active_thread_idZ start_spanr Z VIEW_RENDERZresolver_matchZ view_namer, )r! r9 r: Z sentry_scope�r% �callbackr r �sentry_wrapped_callback� s �z0wrap_async_view.<locals>.sentry_wrapped_callback)r r% � functools�wraps)rM rN r rL r �wrap_async_view� s rQ c s G � fdd�d�}|S )zh Mixin class factory that generates a middleware mixin for handling requests in async mode. c s8 e Zd ZerdZdd� Zdd� Zdd� Z� fdd �ZdS ) z7_asgi_middleware_mixin_factory.<locals>.SentryASGIMixinNc S s || _ d | _| �� d S r )� get_response� _acall_method� _async_check)r- rR r r r �__init__� s z@_asgi_middleware_mixin_factory.<locals>.SentryASGIMixin.__init__c S s t | j�rt| � dS )z� If get_response is a coroutine function, turns us into async mode so a thread is not consumed during a whole request. Taken from django.utils.deprecation::MiddlewareMixin._async_check N)�iscoroutinefunctionrR r rF r r r rT � s zD_asgi_middleware_mixin_factory.<locals>.SentryASGIMixin._async_checkc S s t | j�S )z� Function that checks if we are in async mode, and if we are forwards the handling of requests to __acall__ )rV rR rF r r r �async_route_check� s zI_asgi_middleware_mixin_factory.<locals>.SentryASGIMixin.async_route_checkc � s� | j }|d kr6t| jd�r*| jj | _ }n | j | _ }� |d�}|d krX|||�I d H S |� |||�I d H W 5 Q R � S Q R X d S )N� __acall__)Z old_method)rS r> �_innerrX )r- r9 r: �fZmiddleware_span��_check_middleware_spanr r rX � s zA_asgi_middleware_mixin_factory.<locals>.SentryASGIMixin.__acall__) �__name__� __module__� __qualname__r rY rU rT rW rX r r[ r r �SentryASGIMixin� s r` r )r\ r` r r[ r � _asgi_middleware_mixin_factory� s -ra )'�__doc__r rO �inspectZdjango.core.handlers.wsgir r( Zsentry_sdk.constsr Z sentry_sdk.integrations.asgir Zsentry_sdk.scoper Zsentry_sdk.utilsr r �typingr r r r r Zdjango.core.handlers.asgir Zdjango.http.responser Zsentry_sdk._typesr r r r> rV r r# r? rC rI rQ ra r r r r �<module> s6 (