OwlCyberSecurity - MANAGER
Edit File: fastapi.cpython-312.pyc
� ��/i� � � � d dl Z d dlmZ d dlmZ d dlZd dlmZ d dlm Z d dl m Z m Z d dl mZ d dlmZ erd d lmZmZmZ d d lmZ d d lmZmZ d dlZdZ G d� de� Z d� Z d� Z y# e$ r ed � �w xY w# e$ r ed � �w xY w)� N)�deepcopy)�wraps)� DidNotEnable)�should_send_default_pii)�SOURCE_FOR_STYLE�TransactionSource)�transaction_from_function)� TYPE_CHECKING)�Any�Callable�Dict)�Event)�StarletteIntegration�StarletteRequestExtractorzStarlette is not installedzFastAPI is not installedzgeneric FastAPI requestc � � e Zd ZdZed� � Zy)�FastApiIntegration�fastapic � � t � y �N)�patch_get_request_handler� � �N/opt/hc_python/lib/python3.12/site-packages/sentry_sdk/integrations/fastapi.py� setup_oncez FastApiIntegration.setup_once% s � � "�#r N)�__name__� __module__� __qualname__� identifier� staticmethodr r r r r r " s � ��J��$� �$r r c �< � d}|dk( r-|j j d� }|rCt |� xs d}n3|dk( r.|j j d� }|rt |dd � }|�|}|st }t j }n t | }| j ||�� y )N� �endpoint�url�route�path)�source) �scope�getr �getattr�_DEFAULT_TRANSACTION_NAMEr �ROUTEr �set_transaction_name)r' �transaction_style�request�namer" r$ r% r& s r � _set_transaction_name_and_sourcer0 + s� � � �D��J�&��=�=�$�$�Z�0�� �,�X�6�<�"�D� �e� #�� � �!�!�'�*�� ��5�&�$�/�D�� ��� �(��"�(�(��!�"3�4�� � � �t�F� �3r c �n �� t j j ��fd�} | t j _ y )Nc �� ���� |j d� }|rQ|j �Et j |j � s&|j �t �� �fd�� }||_ �| i |����fd�}|S )N� dependantc � �� t j � }|j �|j j � t j � }|j �|j j � �| i |��S r )� sentry_sdk�get_current_scope� transaction�update_active_thread�get_isolation_scope�profile�update_active_thread_id)�args�kwargs� current_scope� sentry_scope�old_calls �r � _sentry_callzTpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_callR sm �� � !+� <� <� >� � �,�,�8�!�-�-�B�B�D� )� =� =� ?� � �'�'�3� �(�(�@�@�B� ��0��0�0r c � �� ��K � t j � j t � }|� �| i |��� d { ��� S | d }t t j � |j |� t j � }t |� }|j � � d { ��� ��fd�}t j |_ |j |||� � �| i |��� d { ��� S 7 ��7 �H7 ��w)Nr c � �� �fd�}|S )Nc � �� | j di � }�r"d�v rt � r�d |d< d�v r�d |d< t |� | d< | S )Nr. �cookies�data)r( r r )�event�hint� request_info�infos �r �event_processorz�patch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_app.<locals>._make_request_event_processor.<locals>.event_processort s] �� � $)�9�9�Y��#;�L��$��,�1H�1J�6:�9�o�L�� 3�!�T�>�37��<�L�� 0�'/� �'=�E�)�$� �Lr r )�req� integrationrK rJ s �r � _make_request_event_processorzzpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_app.<locals>._make_request_event_processorr s �� � !� '�&r ) r5 � get_client�get_integrationr r0 r6 r- r9 r �extract_request_infor �_name�add_event_processor) r< r= rM r. r? � extractorrN rJ �old_apps @�r � _sentry_appzSpatch_get_request_handler.<locals>._sentry_get_request_handler.<locals>._sentry_appc s� �� �� �$�/�/�1�A�A�BT�U�K��"� $�d� 5�f� 5�5�5��1�g�G� ,��,�,�.� �0M�0M�w� � &�9�9�;�L�1�'�:�I�"�7�7�9�9�D� '�$ "4�!>�!>�L� � � ,� ,�-�g�{�C� � !�$�1�&�1�1� 1�E 6�� :��2 2�s5 �5C+�C%�A$C+� C'� AC+� C)�!C+�'C+�)C+)r( �call�asyncio�iscoroutinefunctionr )r< r= r3 rA rV rU r@ �old_get_request_handlers @@�r �_sentry_get_request_handlerz>patch_get_request_handler.<locals>._sentry_get_request_handlerH su �� ��J�J�{�+� � ����*��/�/� ���?� �~�~�H� �8�_� 1� � 1� *�I�N�)�4�:�6�:��& 2�P �r )r �routing�get_request_handler)r[ rZ s @r r r D s) �� � %�o�o� A� A��C�J +F�G�O�O�'r ) rX �copyr � functoolsr r5 �sentry_sdk.integrationsr �sentry_sdk.scoper �sentry_sdk.tracingr r �sentry_sdk.utilsr �typingr r r r �sentry_sdk._typesr �!sentry_sdk.integrations.starletter r r � ImportErrorr* r r0 r r r r �<module>rh s� �� � � � � 0� 4� B� 6� ��*�*�'�5��3�� 6� �$�-� $�4�2IF��] � 5� �3� 4�4�5�� � 3� �1� 2�2�3�s �A) �A: �)A7�:B