Օƃ] d Z ddlmZ ddlZddlZddlZddlmZ ddlm Z ddl
m Z ej d Z
d Z G 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 dddddddS )NAWS_LAMBDA_FUNCTION_NAMEAWS_LAMBDA_FUNCTION_VERSION AWS_LAMBDA_FUNCTION_MEMORY_SIZEAWS_LAMBDA_LOG_GROUP_NAMEAWS_LAMBDA_LOG_STREAM_NAME
AWS_REGION)lambdaversion memory_size log_group
log_streamregion r r/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/raven/contrib/awslambda/__init__.pyget_default_tagsr s# ,0802
r c e Zd ZdZ fdZ fdZ fdZd
dZed Z ed Z
ed Z xZ 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 | dt } t t | j |d|i| d S )N transport)popr superr __init__)selfargskwargsr __class__s r r zLambdaClient.__init__8 sC JJ{M:: *lD!!*DPIPPPPPPr c d|vri x|d< }n|d }| dd }| dd }|rX| | }| | }|r| | |r| | |r|r| || |d< t t | j |i |S )Ndataeventcontextextra)get_get_http_interface_get_user_interfaceupdate_get_extra_datar r capture) r r r! r$ r% r& http_info user_infor" s r r- zLambdaClient.capture< s $& &F6NTT&>D
7D))**Y-- ' 0077I 0077I
' I&&&
' I&&& AW A 0 0 @ @DM0u\4((0$A&AAAr c t t | j |i |}t D ]>\ }}|d |t j | ?| dt j d | dt j d |S )NtagsreleaseSENTRY_RELEASE environmentSENTRY_ENVIRONMENT) r r build_msgr items
setdefaultosenvironr( )r r r! r$ optiondefaultr" s r r6 zLambdaClient.build_msgQ s 2u\4((2DCFCC / 1 1 7 7 9 9 E EOFG L # #FBJNN7,C,C D D D D 2:>>2B#C#CDDD
rz~~6J'K'KLLL r Nc t t st j j S pt
ft j fd }|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 | |g|R i |S # $ r+ j d| |d| j w xY w)N)r% r& r )captureExceptionr& clear)r% r& r r! r> fr s r wrappedz0LambdaClient.capture_exceptions..wrappedx s|
q9$999&999
%%ME7MMfMMM ""$$$
s 0A )
isinstancer functoolspartialcapture_exceptions Exceptionwraps)r rB r> rC s ``` r rG z LambdaClient.capture_exceptionsZ s . !\** L $T%<KKK K /I<
r c | d r| d d }|r| dd p| dd | dd | dd | dd | dd | d d }d|iS d S d S )
NrequestContextidentitycognitoIdentityIdusersourceIpcognitoIdentityPoolIdcognitoAuthenticationType userAgent)idusername
ip_addresscognito_identity_pool_idcognito_authentication_type
user_agentr( )r% rL rN s r r* z LambdaClient._get_user_interface s 99% & & & -.z:H
&",,':DAA_X\\RXZ^E_E_ ( VT : :"*,,z4"@"@08 =TVZ0[0[3;<<@[]a3b3b"*,,{";";
~% & &
&
&r c | d rp| d r]| d | d | dd | dd pg d}d|iS d S d S )Npath
httpMethodqueryStringParametersheaders)urlmethod query_stringr^ requestrY )r% rb s r r) z LambdaClient._get_http_interface s 99V (r} s ' & & & & & . . . . . . 1 2 2 J J J J J 6 J J J J J r