B ��Y�@srdZddlmZddlZddlZddlZddlmZddlm Z ddl mZe�d�Z dd �ZGd d�de �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.clientcCsddddddd�S)NZAWS_LAMBDA_FUNCTION_NAMEZAWS_LAMBDA_FUNCTION_VERSIONZAWS_LAMBDA_FUNCTION_MEMORY_SIZEZAWS_LAMBDA_LOG_GROUP_NAMEZAWS_LAMBDA_LOG_STREAM_NAMEZ AWS_REGION)�lambda�versionZmemory_sizeZ log_groupZ log_streamZregion�rrr�Q/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/awslambda/__init__.py�get_default_tagssr csfeZdZdZ�fdd�Z�fdd�Z�fdd�Zdd d �Zedd��Z ed d��Z edd��Z�ZS)�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!') cs*|�dt�}tt|�j|d|i|��dS)N� transport)�getr�superr�__init__)�self�args�kwargsr)� __class__rr r8szLambdaClient.__init__cs�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_datarr�capture)rrrrrrZ user_infoZ http_info)rrr r<s zLambdaClient.capturecsntt|�j||�}x.t��D] \}}|d�|tj�|��qW|�dtj�d��|�dtj�d��|S)NZtags�releaseZSENTRY_RELEASE�environmentZSENTRY_ENVIRONMENT) rr� build_msgr �items� setdefault�os�environr )rrrr�option�default)rrr rMszLambdaClient.build_msgNcsBt�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``. )� exceptionscsNy�||f|�|�S�k rH�jf||d�|��j��YnXdS)N)rr)ZcaptureExceptionr�clear)rrrr)r&�frrr �wrappedts z0LambdaClient.capture_exceptions..wrapped)� isinstancer� functools�partial�capture_exceptions� Exception�wraps)rr(r&r)r)r&r(rr r-Vs zLambdaClient.capture_exceptionscCsr|�d�rn|dd}|rn|�dd�p0|�dd�|�dd�|�dd�|�dd�|�dd�|�d�d �}d|iSdS) NZrequestContext�identityZcognitoIdentityId�userZsourceIpZcognitoIdentityPoolIdZcognitoAuthenticationTypeZ userAgent)�id�usernameZ ip_addressZcognito_identity_pool_idZcognito_authentication_typeZ user_agent)r )rr0r1rrr r~s z LambdaClient._get_user_interfacecCsN|�d�rJ|�d�rJ|�d�|�d�|�dd�|�dd�p s