o ni @sddlmZdZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZddlmZddlZddlZddlZddlZddlZddlZe rRddlZdZedZdd d ZGdddZdS)) annotations)SentryN) TracebackType) TYPE_CHECKINGAnyCallableLiteralquotezRhttps://2592b1968ea94cca9b5ef5e348e094a7@o151352.ingest.sentry.io/4504800232407040)okexitedcrashedabnormalfuncrreturncstd fdd }|S) zODecorator to ensure that Sentry methods do nothing if disabled and don't raise.self type[Sentry]argsrkwargsrc sl|jrdSz |g|Ri|WSty5}zjdkr*|djd|WYd}~dSd}~ww)N exceptionz Error in z: ) _disabled Exception__name__r)rrrerT/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/analytics/sentry.pywrapper$s z_safe_noop..wrapperN)rrrrrrrr) functoolswraps)rrrrr _safe_noop!s r c@seZdZUded<d)ddZed*dd Zed)d d Zed+d,ddZ e  d-d.ddZ d/ddZ ed)ddZ ed)d d!Z ed0d1d"d#Ze  d2d3d'd(ZdS)4rboolrrNonecCs@tj |_t|_tjtjj t |_ d|_ t |jdSN)wandbenverror_reporting_enabledrset_sent_messagesosenvironget SENTRY_DSNSENTRY_DEFAULT_DSNdsnscopeatexitregister end_session)rrrr__init__6s zSentry.__init__strcCs&ttjjjd}|rdSdS)z(Return the environment we're running in.z.git development production)pathlibPath__file__parentjoinpathexists)ris_gitrrr environmentAs zSentry.environmentcCs@tj|jd|jtjd}t|_|j |j |dS)zSetup Sentry SDK. We use lower-level APIs (i.e., not sentry_sdk.init) here to avoid the possibility of interfering with the user's own Sentry SDK setup. F)r.default_integrationsr>releaseN) sentry_sdkClientr.r>r$ __version__get_global_scopeforkr/clear set_client)rclientrrrsetupJs z Sentry.setupTmessagerepeat str | NonecCsZ|s ||jvr dS|j|tj|j t|WdS1s&wYdS)zSend a message to Sentry.N)r(addrAr/use_isolation_scopecapture_message)rrJrKrrrrJ\s  $zSentry.messageFNexcjstr | BaseException | tuple[type[BaseException] | None, BaseException | None, TracebackType | None] | NonehandledstatusSessionStatus | Nonec Cst|trtjt|}nt|trtj|}nt}tjj ||j j d|dd\}}d}ztj |j  t|}Wdn1sJwYWn tyYYnw|pa|s`dnd}|j|d|j }|duru||S)zLog an exception to Sentry.generic)typerR)client_options mechanismNr erroredrS) isinstancer4rAutilsexc_info_from_errorr BaseExceptionsysexc_infoevent_from_exceptionr/ get_clientoptionsrN capture_event mark_sessionflush) rrPrRrSr`event_event_idrHrrrres2        zSentry.exceptionrcCs|||td)zRe-raise an exception after logging it to Sentry. Use this for top-level exceptions when you want the user to see the traceback. Must be called from within an exception handler. )rwith_tracebackr_r`)rrPrrrreraises zSentry.reraisecCs0|jdusJ|jj}|dur|jdSdS)zStart a new session.N)r/_session start_session)rsessionrrrrns zSentry.start_sessioncCsN|jdusJ|j}|jj}|dur#|dur%|j|dSdSdS)zEnd the current session.N)r/rbrmr2rf)rrHrorrrr2s   zSentry.end_sessioncCs2|jdusJ|jj}|dur|j|ddSdS)z'Mark the current session with a status.NrZ)r/rmupdate)rrSrorrrres zSentry.mark_sessiontagsdict[str, Any] | Noneprocess_contextcsb|jdusJd}|r|jd|durdS|D]}|d}|dvr-|j||qddr7d}nddr@d}n d drId }nd }|jd |d D]I}|d|d}} | drhqTz,tjd} fdddD\} } |j| | d| d| d|d| WqTtyYqTwd} | rd| i|j_|dS)aWConfigure the Sentry scope for the current thread. This function should be called at the beginning of every thread that will send events to Sentry. It sets the tags that will be applied to all events sent from this thread. It also tries to start a session if one doesn't already exist for this thread. N) entityprojectrun_idrun_url sweep_urlsweep_id deploymentx_disable_servicex_require_legacy_servicelaunch _platformrs)N_colabcolab_jupyterjupyter_ipythonipythonpythonpython_runtime)runsweep_id_urlbase_urlc3s|] }t|VqdSr#r ).0krqrr sz)Sentry.configure_scope..)rtru/zs/email) r/set_tagr+r$utilapp_urlruserrn)rrqrs settings_tagstagvalrobjobj_idobj_urlrrtrurrrrconfigure_scopesJ      "    zSentry.configure_scope)rr")rr4)T)rJr4rKr!rrL)FN)rPrQrRr!rSrTrrL)rPrrr"r#)rSrTrr")NN)rqrrrsrLrr")r __module__ __qualname____annotations__r3propertyr>r rIrJrrlrnr2rerrrrrr3s0      ,   r)rrrr) __future__r__all__r0rr)r7r_typesrtypingrrrr urllib.parser rAsentry_sdk.scopesentry_sdk.utilsr$ wandb.env wandb.util"wandb.sdk.internal.settings_staticr- SessionStatusr rrrrrs.