o w¡·ió#ã@sàUddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z ddlmZm Z mZmZddlmZddlmZmZmZedƒZedƒZd Zed Zddd„ZGdd„dƒZdaded<e ¡Zddd„Z dS)é)ÚannotationsN)Ú TracebackType)ÚAnyÚCallableÚLiteralÚTypeVar©Úquote)ÚConcatenateÚNeverÚ ParamSpecÚ_PÚ_TzRhttps://2592b1968ea94cca9b5ef5e348e094a7@o151352.ingest.sentry.io/4504800232407040)ÚokÚexitedÚcrashedÚabnormalÚmethodú%Callable[Concatenate[Sentry, _P], _T]Úreturnú,Callable[Concatenate[Sentry, _P], _T | None]cst ˆ¡d‡fd d „ƒ}|S) z¶Make a Sentry method safe, lazy, and non-raising. The wrapped method becomes a no-op if Sentry is disabled, this instance belongs to a different PID, or lazy boot fails ÚselfÚSentryÚargsú_P.argsÚkwargsú _P.kwargsrú _T | Nonecsº|jsdS|jt ¡krdS|js| ¡sdSzˆ|g|¢Ri|¤ŽWSty\}z-ˆjdkrQt t¡| dˆj›d|›¡Wdƒn1sLwYWYd}~dSd}~ww)NÚ exceptionz Error in z: )Ú_enabledÚ_pidÚosÚgetpidÚ_bootedÚ_bootÚ ExceptionÚ__name__Ú contextlibÚsuppressr)rrrÚe©r©úJ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/analytics/sentry.pyÚwrapper#s ÿ€ûz_guard..wrapperN)rrrrrrrr)Ú functoolsÚwraps)rr-r+r*r,Ú_guards r0c@s eZdZd-dd„Zed.dd „ƒZd/dd„Ze d0d1dd„ƒZe d2d3dd„ƒZ d4d d!„Z ed5d"d#„ƒZed5d$d%„ƒZed6d7d&d'„ƒZ e d8d9d+d,„ƒZdS):rÚpidÚintrÚNonecCs`ddlm}||_t| ¡ƒ|_d|_t ¡|_ d|_ tƒ|_d|_ d|_tj |jt¡|_dS)Nr)ÚenvF)Úwandbr4r ÚboolÚerror_reporting_enabledrr#Ú threadingÚLockÚ _boot_lockÚ_atexit_registeredÚsetÚ_sent_messagesÚ_sdkÚscoper!ÚenvironÚgetÚ SENTRY_DSNÚSENTRY_DEFAULT_DSNÚdsn)rr1Ú_envr+r+r,Ú__init__As zSentry.__init__ÚstrcCs&t t¡jjj d¡ ¡}|rdSdS)Nz.gitÚdevelopmentÚ production)ÚpathlibÚPathÚ__file__ÚparentÚjoinpathÚexists)rÚis_gitr+r+r,ÚenvironmentPszSentry.environmentr6c Cs"ddlm}|j~|js WdƒdS|jr" WdƒdSz?ddl}ddl}ddl}||_|jj |j d|j|d}|j ¡ ¡}| ¡| |¡||_d|_|js`t |j¡d|_Wnty}d|_d|_d|_d|_YWdƒdSw WdƒdS1sŠwYdS)z*Import sentry_sdk and set up client/scope.r)Ú__version__NFT)rDÚdefault_integrationsrQÚrelease)r5rRr:rr#Ú sentry_sdkÚsentry_sdk.scopeÚsentry_sdk.utilsr>ÚClientrDrQÚget_global_scopeÚforkÚclearÚ set_clientr?r;ÚatexitÚregisterÚend_sessionr%)rrRrUÚclientr?r+r+r,r$UsLþûü €Ú $ØzSentry._bootTÚinfoÚmessageÚrepeatÚlevelú str | NonecCsb|s ||jvr dS|j |¡|jj |j¡|jj||dWdƒS1s*wYdS)N)rd)r=Úaddr>r?Úuse_isolation_scopeÚcapture_message)rrbrcrdr+r+r,rbƒs$ÿzSentry.messageFNÚexcújstr | BaseException | tuple[type[BaseException] | None, BaseException | None, TracebackType | None] | NoneÚhandledÚstatusúSessionStatus | Nonec Cs t|tƒr|jj t|ƒ¡}nt|tƒr|jj |¡}nt ¡}|jjj ||j ¡jd|dœd\}}d}t t¡&|jj |j ¡|j |¡}Wdƒn1sTwYWdƒn1scwY|po|sndnd}|j|d|j ¡}|durƒ| ¡|S)NÚgeneric)Útyperk)Úclient_optionsÚ mechanismrÚerrored©rl)Ú isinstancerGr>ÚutilsÚexc_info_from_errorr%Ú BaseExceptionÚsysÚexc_infoÚevent_from_exceptionr?Ú get_clientÚoptionsr'r(rgÚ capture_eventÚmark_sessionÚflush) rrirkrlryÚeventÚ_Úevent_idr`r+r+r,rs. ýÿ€ÿ zSentry.exceptionrrc Csjz| |¡Wt ¡\}}}|durt|dƒr| |¡‚|‚t ¡\}}}|dur3t|dƒr3| |¡‚|‚)z?Re-raise after logging, preserving traceback. Safe if disabled.NÚwith_traceback)rrxryÚhasattrrƒ)rrirÚtbr+r+r,Úreraise·s ý zSentry.reraisecCs,|jdurdS|jjdur|j ¡dSdS©N)r?Ú_sessionÚ start_session)rr+r+r,r‰Ás ÿzSentry.start_sessioncCsN|jdurdS|j ¡}|jj}|dur#|dur%|j ¡| ¡dSdSdSr‡)r?r{rˆr_r)rr`Úsessionr+r+r,r_Ès þzSentry.end_sessioncCs2|jdurdS|jj}|dur|j|ddSdS)Nrs)r?rˆÚupdate)rrlrŠr+r+r,r~Òs ÿzSentry.mark_sessionÚtagsúdict[str, Any] | NoneÚprocess_contextcsjddl}|jdurdSd}|r|j d|¡ˆdurdS|D]}ˆ |d¡}|dvr1|j ||¡qˆ dd¡r;d}nˆ dd¡rDd}nˆ d d¡rMd }nd}|j d|¡d D]I}|›d|›d} } ˆ | d¡rlqXz,|j ˆd¡}‡fdd„dDƒ\}} |j | |›d|›d| ›d|›dˆ| › ¡WqXty¡YqXwˆ d¡}|r¯d|i|j_| ¡dS)Nr) ÚentityÚprojectÚrun_idÚrun_urlÚ sweep_urlÚsweep_idÚ deploymentÚlaunchÚ _platformrŽ)NÚÚ_colabÚcolabÚ_jupyterÚjupyterÚ_ipythonÚipythonÚpythonÚpython_runtime)ÚrunÚsweepÚ_idÚ_urlÚbase_urlc3s|] }tˆ|ƒVqdSr‡r)Ú.0Úk©rŒr+r,Ú s€z)Sentry.configure_scope..)rrú/zs/Úemail) Ú wandb.utilr?Úset_tagrAÚutilÚapp_urlr%Úuserr‰)rrŒrŽr5Ú settings_tagsÚtagÚvalr ÚobjÚobj_idÚobj_urlr¯rrr«r+r¨r,Úconfigure_scopeÚsN €"þÿ zSentry.configure_scope)r1r2rr3)rrG)rr6)Tra)rbrGrcr6rdrGrre)FN)rirjrkr6rlrmrre)rirrr)rr3r‡)rlrmrr3)NN)rŒrrŽrerr3)r&Ú __module__Ú__qualname__rFÚpropertyrQr$r0rbrr†r‰r_r~r·r+r+r+r,r@s0 .üõ & ýrz Sentry | NoneÚ _singletoncCstt ¡}t*tdurtj|krtWdƒStdus"tj|kr't|datWdƒS1s3wYdS)z¦Return the Sentry singleton for the current process (fork-aware). Creates a new instance in child processes after fork. Thread-safe within each process. N©r1)r!r"Ú_singleton_lockr»r rr¼r+r+r,Ú get_sentry sþ $ùr¾)rrrr)rr)!Ú __future__rr]r'r.r!rJrxr8ÚtypesrÚtypingrrrrÚurllib.parser Útyping_extensionsr rrr rrCÚ SessionStatusr0rr»Ú__annotations__r9r½r¾r+r+r+r,Ús.ÿ &]