관리-도구
편집 파일: __init__.cpython-37.pyc
B ��Y � @ sr d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl m Z e� d�Z dd � ZG d d � d e �ZdS ) z� raven.contrib.awslambda ~~~~~~~~~~~~~~~~~~~~ Raven wrapper for AWS Lambda handlers. :copyright: (c) 2010-2012 by the Sentry Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. � )�absolute_importN)� FunctionType)�Client)� HTTPTransportzsentry.errors.clientc C s ddddddd�S )NZAWS_LAMBDA_FUNCTION_NAMEZAWS_LAMBDA_FUNCTION_VERSIONZ AWS_LAMBDA_FUNCTION_MEMORY_SIZEZAWS_LAMBDA_LOG_GROUP_NAMEZAWS_LAMBDA_LOG_STREAM_NAMEZ AWS_REGION)�lambda�versionZ memory_sizeZ log_groupZ log_streamZregion� r r r �Q/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/awslambda/__init__.py�get_default_tags s r c sf e Zd ZdZ� fdd�Z� fdd�Z� fdd�Zdd d �Zed d � �Z ed d� �Z edd� �Z � Z S )� LambdaClienta� Raven decorator for AWS Lambda. By default, the lambda integration will capture unhandled exceptions and instrument logging. Usage: >>> from raven.contrib.awslambda import LambdaClient >>> >>> >>> client = LambdaClient() >>> >>> @client.capture_exceptions >>> def handler(event, context): >>> ... >>> raise Exception('I will be sent to sentry!') c s* |� dt�}tt| �j|d|i|�� d S )N� transport)�getr �superr �__init__)�self�args�kwargsr )� __class__r r r 8 s zLambdaClient.__init__c s� d|kri |d< }n|d }|� dd �}|� dd �}| �|�}|rN|�|� |rz| �|�}|rj|�|� | �||�|d< tt| �j||�S )N�data�event�context�extra)r �_get_user_interface�update�_get_http_interface�_get_extra_datar r �capture)r r r r r r Z user_infoZ http_info)r r r r < s zLambdaClient.capturec sn t t| �j||�}x.t� �� D ] \}}|d �|tj�|�� q W |�dtj�d�� |�dtj�d�� |S )NZtags�releaseZSENTRY_RELEASE� environmentZSENTRY_ENVIRONMENT) r r � build_msgr �items� setdefault�os�environr )r r r r �option�default)r r r r M s zLambdaClient.build_msgNc sB t �t�stj�j�d�S � p"tf� t���� ��fdd��}|S )a� Wrap a function or code block in try/except and automatically call ``.captureException`` if it raises an exception, then the exception is reraised. By default, it will capture ``Exception`` >>> @client.capture_exceptions >>> def foo(): >>> raise Exception() >>> with client.capture_exceptions(): >>> raise Exception() You can also specify exceptions to be caught specifically >>> @client.capture_exceptions((IOError, LookupError)) >>> def bar(): >>> ... ``kwargs`` are passed through to ``.captureException``. )� exceptionsc sN y�| |f|�|�S � k rH �j f | |d�|�� �j�� � Y nX d S )N)r r )ZcaptureExceptionr �clear)r r r r )r& �fr r r �wrappedt s z0LambdaClient.capture_exceptions.<locals>.wrapped)� isinstancer � functools�partial�capture_exceptions� Exception�wraps)r r( r& r) r )r& r( r r r- V s z LambdaClient.capture_exceptionsc C sr | � d�rn| d d }|rn|� dd �p0|� dd �|� dd �|� dd �|� dd �|� dd �|� d�d �}d|iS d S ) NZrequestContext�identityZcognitoIdentityId�userZsourceIpZcognitoIdentityPoolIdZcognitoAuthenticationTypeZ userAgent)�id�usernameZ ip_addressZcognito_identity_pool_idZcognito_authentication_typeZ user_agent)r )r r0 r1 r r r r ~ s z LambdaClient._get_user_interfacec C sN | � d�rJ| � d�rJ| � d�| � d�| � dd �| � dd �p<g d�}d|iS d S )N�pathZ httpMethodZqueryStringParameters�headers)�url�methodZ query_stringr5 �request)r )r r8 r r r r � s z LambdaClient._get_http_interfacec C sZ | |j t|�d�}|jrV|jjj|jjj|jjj|jjj|jjj|jj |jj d�|d< |S )N)r �aws_request_idr )zclient.installation_idzclient.app_titlezclient.app_version_namezclient.app_version_codezclient.app_package_name�custom�env�client_context) r9 �varsr< ZclientZinstallation_idZ app_titleZapp_version_nameZapp_version_codeZapp_package_namer: r; )r r Z extra_contextr r r r � s z LambdaClient._get_extra_data)NN) �__name__� __module__� __qualname__�__doc__r r r r- � staticmethodr r r � __classcell__r r )r r r $ s ( r )rA � __future__r r" �loggingr+ �typesr Z raven.baser Zraven.transport.httpr � getLogger�loggerr r r r r r �<module> s