yjhfddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z m Z ddlmZmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZm Z dd l!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5er ddl6Z6ddl)m7Z7dZ8Gdde'Z9dS)) annotationsN)Mapping)datetime timedeltatimezone) TYPE_CHECKINGAny)urljoin)Once)ValidationError)RequestExceptionSession) UA_HEADER)VariablesOptions)ServerResponseCallbackinstall_logfire_response_hook)UnexpectedResponse)ResolvedVariableVariableAlreadyExistsErrorVariableNotFoundErrorVariableProviderVariableWriteError) KeyIsNotPresent KeyIsPresent LabeledValueLabelRefValueDoesNotEqualValueDoesNotMatchRegex ValueEquals ValueIsIn ValueIsNotInValueMatchesRegexVariableConfigVariablesConfigVariableTypeConfig)LogfireRemoteVariableProviderceZdZdZ d@dAd Zd ZdBdZdCdZdZdZ dDdZ dZ dEdFdZ dGdHd%Z dIfd' ZdJdKd*ZdLd-ZdMd/ZdNd2ZdOd3ZdPd4ZdQd6ZedRd8ZdSd;ZdTd=ZdUd?ZxZS)Vr'zVariable provider that fetches configuration from a remote Logfire API. The threading implementation draws heavily from opentelemetry.sdk._shared_internal.BatchProcessor. Nbase_urlstrtokenoptionsrserver_response_hookServerResponseCallback | Nonec|j}|j}||_||_||_t |_|jjd|tdt|j||j |_ ||_ t|tt frt#|n||_t'j|_t-|_d|_d|_d|_d|_d|_t'j|_t'j|_t'j|_ d|_!d|_"d|_#d|_$d|_%tMj'|_(dS)a|Create a new remote variable provider. Args: base_url: The base URL of the Logfire API. token: Authentication token for the Logfire API. options: Options for retrieving remote variables. server_response_hook: Optional override for the API response hook (see `AdvancedOptions.server_response_hook`). bearer ) Authorization User-Agent)secondsFN))block_before_first_resolvepolling_interval _base_url_token_server_response_hookr_sessionheadersupdaterrtimeout_timeout_block_before_first_fetch isinstancefloatintr_polling_interval threadingLock _session_lockr _reset_once_has_attempted_fetch_last_fetched_at_config _shutdown_shutdown_timeout_exceeded _refresh_lockEvent_worker_awaken_force_refresh_event_sse_connected _sse_thread_logfire_started_worker_threadosgetpid_pid)selfr)r+r,r-r4r5s U/home/lighthouse/health/venv/lib/python3.11/site-packages/logfire/variables/remote.py__init__z&LogfireRemoteVariableProvider.__init__:sk &-%G""3! %9"   $$7H7H7HXa%b%bccc%dm5IJJJ )C&3=>NQVX[P\3]3] sI. / / / /cs '^--66*/!15/3 */'&^--'o//$-O$5$5!$4815  8<IKK ctj|_tj|_tj|_tj|_|jrUtjd|j d|_ |j d|_ | tj|_dS)NLogfireRemoteProviderTnametargetdaemonF)rCrDrLrErMrNrOrSThread_workerrTstartrP_start_sse_listenerrUrVrWrXs rY_at_fork_reinitz-LogfireRemoteVariableProvider._at_fork_reinitss&^--&^--'o//$-O$5$5! = '"+"2,|###D    % % ' ' '"'D   $ $ & & &IKK r[logfire_instancelogfire.Logfire | NonereturnNonec||d|_|jrdSd|_tjd|jd|_|j|ttdr6tj |j d fd }tj| dSdS) aStart background polling with the given logfire instance for error logging. Args: logfire_instance: The Logfire instance to use for error logging, or None if variable instrumentation is disabled (errors will be reported via warnings). Nzvariables.provider)custom_scope_suffixTr]r^register_at_forkrjrkc8}| |dSdSN)method weak_reinits rY_after_in_childz._after_in_childs+$%FHHHHH&%r[)after_in_child)rjrk) with_settingsrRrSrCrbrcrTrdrehasattrrUweakref WeakMethodrgrn)rXrhrtrss @rYrdz#LogfireRemoteVariableProvider.starts  ',::Oc:ddDM =  F (.(<    !!###   """ 2) * * @!,T-ABBK         ? ? ? ? ? ? @ @r[messageexc Exceptionc|j-|jd|t||dStj|d|t dS)zLog an error using logfire if available, otherwise warnings. Args: message: The error message. exc: The exception that occurred. Nz{message}: {error})rzerror _exc_infoz: )category)rRr~r*warningswarnRuntimeWarning)rXrzr{s rY _log_errorz(LogfireRemoteVariableProvider._log_errorsc = $ M   4gSQTXXad  e e e e e MW---- G G G G G Gr[c|j|jrdStjd|jd|_|jdS)z4Start the SSE listener thread for real-time updates.NLogfireRemoteProviderSSETr^)rQis_aliverCrb _sse_listenerrdrfs rYrez1LogfireRemoteVariableProvider._start_sse_listenersg   'D,<,E,E,G,G ' F$++%         r[c^t|jd}d}d}|js t5}|jd|jtdddt||j | |dd }|j d kr"!E<$#zz||#%$ ??733 %'+ABBx~~'7'7H %-1Z-A-A -7^^G-D-D #-1R#R#R$($=$A$A$C$C$C$($7$;$;$=$=$=$($8)#D%%% $% %K1%1%1%1%1%1%1%1%1%1%1%1%1%1%1%f T T T&+#~T,,_===&)/A*=?R&S&SO  Tm.; T; T; T; T; TshGBG GA5GAF('G(G>GGG GGGGGAH H delayr@cd}||krT|jsQtd||z }tj|||z }||kr |jKdSdSdSdS)z|jtjt j|jz kr ddddS |j5|j t|j d|j }tj||}dddn #1swxYwYn9#t"$r,}|d|Yd}~ddddSd}~wwxYw t'j|}||_tjt j|_n-#t,$r }|d|Yd}~nd}~wwxYwd|_n #d|_wxYw ddddS#1swxYwYdS) zFetch the latest variable configuration from the remote API. Args: force: If True, fetch configuration even if the polling interval hasn't elapsed. FN)tz/v1/variables/r<zError retrieving variablesz8Failed to parse variables configuration from Logfire APIT)rLlockedrHrnowrutcrBrEr9rr r6r=rraise_for_statusrr|rr$model_validaterIr rG)rXrvariables_responsevariables_config_datae new_configs rYrz%LogfireRemoteVariableProvider.refresh$s   $ $ & & E   1 1 )5)HLHL,I,I,IDLb,bbb  1 1 1 1 1 1 1 1 'FF)-):):0@AA4=*;**&'78JKKK,>,C,C,E,E) FFFFFFFFFFFFFFF     rrIrresolve_serialized_value)rXrrrs rYget_serialized_valuez2LogfireRemoteVariableProvider.get_serialized_valueOs 9 # #   $ $T%9 : : :( T-K  LLNNN < #dK[\\\ \|44]MS]^^^r[labelc|jtjkr|j|j|js|jr|t ||S)aNResolve a variable's serialized value for a specific label from the remote configuration. Args: variable_name: The name of the variable to resolve. label: The name of the label to select. Returns: A ResolvedVariable containing the serialized value (or None if not found). ) rWrUrVrFrrgrGr>rsuperget_serialized_value_for_label)rXrr __class__s rYrz  F !!! ^c1155<.3.22T9   *   $ $^ $ < < <   '   ! !+ ! 6 6 6   " " M   ! ! ! " " " " " " " " " " " " " " " " " "s!CC C r_VariableConfig | NonecH|jdS|j|S)aRetrieve the full configuration for a variable from the cached config. This method supports alias-based lookup, so you can pass either the variable's current name or any of its configured aliases. Args: name: The name (or alias) of the variable. Returns: The VariableConfig if found, or None if the variable doesn't exist. N)rI_get_variable_config)rXr_s rYget_variable_configz1LogfireRemoteVariableProvider.get_variable_configs' < 4|00666r[r$c>|jtiS|jS)zRetrieve all variable configurations from the cached config. Returns: A VariablesConfig containing all variable configurations. Returns an empty VariablesConfig if no config has been fetched yet. N) variables)rIr$rfs rYget_all_variables_configz6LogfireRemoteVariableProvider.get_all_variables_configs% < "R000 0|r[configr#c||} |j5|jt |jd||j}|jdkrtd|j dtj |dddn #1swxYwYn,#ttf$r}td||d}~wwxYw|d |S) aeCreate a new variable configuration via the remote API. Args: config: The configuration for the new variable. Returns: The created VariableConfig. Raises: VariableAlreadyExistsError: If a variable with this name already exists. VariableWriteError: If the API request fails. rrr<i Variable 'z' already existsNzFailed to create variable: Tr)_config_to_api_bodyrEr9postr r6r=rrr_rrr rrrXrbodyrrs rYcreate_variablez-LogfireRemoteVariableProvider.create_variablesL''// O# > >=--DN,<==DRVR_.'3..45_&+5_5_5_```"3H===  > > > > > > > > > > > > > > >#$45 O O O$%F1%F%FGGQ N O 4     s;B#A-B B#BB#BB##C 4CC c ||} |j5|jt |jd|d||j}|jdkrtd|dtj |dddn #1swxYwYn,#ttf$r}td||d}~wwxYw| d |S) aUpdate an existing variable configuration via the remote API. Args: name: The name of the variable to update. config: The new configuration for the variable. Returns: The updated VariableConfig. Raises: VariableNotFoundError: If the variable does not exist. VariableWriteError: If the API request fails. r/rr ' not foundNzFailed to update variable: Tr)rrEr9putr r6r=rrrrr rr)rXr_rrrrs rYupdate_variablez-LogfireRemoteVariableProvider.update_variablesU''// O# > >=,,DN,DT,D,D,DEEDZ^Zg-'3../0NT0N0N0NOOO"3H===  > > > > > > > > > > > > > > >#$45 O O O$%F1%F%FGGQ N O 4     s;B"A,B B"BB"BB""C 3CC c  |j5|jt|jd|d|j}|jdkrtd|dtj |dddn #1swxYwYn,#ttf$r}td||d}~wwxYw| d dS) aDelete a variable configuration via the remote API. Args: name: The name of the variable to delete. Raises: VariableNotFoundError: If the variable does not exist. VariableWriteError: If the API request fails. rrrrrrNzFailed to delete variable: Tr) rEr9deleter r6r=rrrrr rr)rXr_rrs rYdelete_variablez-LogfireRemoteVariableProvider.delete_variables= O# > >=//DN,DT,D,D,DEEt}0'3../0NT0N0N0NOOO"3H===  > > > > > > > > > > > > > > >#$45 O O O$%F1%F%FGGQ N O 4      s;B A+B4 B BB BB B5B00B5dict[str, Any]c@d|ji}|j|d<|j |j|d<d|jji|d<fd|jD|d<|j |j|d <|j |j|d <|jr(fd |jD|d<|S) zConvert a VariableConfig to the API request body format. Args: config: The VariableConfig to convert. Returns: A dictionary suitable for the API request body. r_ descriptionN json_schemalabelsrolloutcTg|]$}fd|jDd|jjid%S)cXg|]&}|j|jd|'S))kind attribute)rr_condition_extra_fields).0condrXs rY zPLogfireRemoteVariableProvider._config_to_api_body...sI"YT^jjtGcGcdhGiGijr[r) conditionsr)rrr)roverriderXs rYrzELogfireRemoteVariableProvider._config_to_api_body..si      ( 3%h&6&=>       r[ overridesaliasesexamplecBi|]\}}||Srq)_label_to_api_data)r label_name label_valuerXs rY zELogfireRemoteVariableProvider._config_to_api_body../s;E\ZQ\ D33K@@r[) r_rrrrr r r items)rXrrs` rYrz1LogfireRemoteVariableProvider._config_to_api_bodys!' 4%0]   )"("4D #V^%:;Y     #,    [ > %$nDO > %$nDO = `f`m`s`s`u`uDN r[LabeledValue | LabelRefct|trd|j|jdS|jdkrddiS|jdkrddiSd|jdS)zConvert a label value to the API label data format. Args: label: The label to convert. Returns: A dictionary suitable for the API labels field. version) target_typerserialized_valuelatestr code_defaultr)r target_label)r?rrrref)rs rYr z0LogfireRemoteVariableProvider._label_to_api_data5su e\ * * ( =$)$:  9 !8, , Y. ( (!>2 2#*EIFF Fr[ conditionr ct|ttfr d|jiSt|tt frdt |jiSt|ttfr,|j }dt|tj r|j n|iSt|ttfriSiS)zExtract extra fields from a condition based on its type. Args: condition: The condition object. Returns: A dictionary of extra fields for the condition. rvaluespattern)r?rrrr r!listrr"rrrePatternrr)rXrrs rYrz5LogfireRemoteVariableProvider._condition_extra_fieldsMs i+/@!A B B Y_- -  I|#< = = d9#3445 5  $57M#N O O 'G*Wbj2Q2Q^wW^_ _  L/#B C C I r[dict[str, VariableTypeConfig]c ddlm} |j5|jt |jd|j}tj || }dddn #1swxYwYn%#t$r}td||d}~wwxYwi}|D][}||d|di|d |d  }|||j <\|S) zList all variable types from the remote API. Returns: A dictionary mapping type names to their configurations. rr%/v1/variable-types/rNzFailed to list variable types: r_rr source_hint)r_rrr%) logfire.variables.configr&rEr9rr r6r=rrrrr_)rXr&r types_datarresult type_datars rYlist_variable_typesz1LogfireRemoteVariableProvider.list_variable_typescs| @????? S# - -=,,WT^EZ-[-[eier,ss"3H===%]]__  - - - - - - - - - - - - - - -" S S S$%Jq%J%JKKQR R S02# ) )I''v&%MM-<<%MM-88%MM-88 F #)F6;   s;BAA8, B8A<<B?A<B B&B!!B&r&cp|j|j|jd}|j |j|d< |j5|jt|jd||j }tj |dddn #1swxYwYn%#t$r}td||d}~wwxYw|S)aCreate or update a variable type via the remote API. If a type with the given name exists, it will be updated. Otherwise, a new type will be created. Args: config: The type configuration to upsert. Returns: The created or updated VariableTypeConfig. Raises: VariableWriteError: If the API request fails. )r_rrNr%r$rz Failed to upsert variable type: ) r_rrr%rEr9rr r6r=rrrrs rYupsert_variable_typez2LogfireRemoteVariableProvider.upsert_variable_type}s+ K!-!-     )"("4D  T# > >=--DN,ABBW[Wd.#3H===  > > > > > > > > > > > > > > > " T T T$%K%K%KLLRS S T s;BA B9 BB  B B  B B3B..B3rp)r)r*r+r*r,rr-r.)rhrirjrk)rzr*r{r|rjrk)rr@)F)rr)NN)rr*rrrrrjr)rr*rr*rjr)r)rr@)r_r*rjr)rjr$)rr#rjr#)r_r*rr#rjr#)r_r*rjrk)rr#rjr)rrrjr)rr rjr)rjr")rr&rjr&)__name__ __module__ __qualname____doc__rZrgrdrrerrrcrrrrrrrrrr staticmethodr rr*r, __classcell__)rs@rYr'r'4s'?C 7 7 7 7 7 r   &!@!@!@!@F H H H H ! ! !ATATATF!!!!.)1)1)1)1)1\%)/3 _____8LLLLLL,"""""07777    :<!!!!0,,,,\GGG\G.,4!!!!!!!!r[r'): __future__r _annotationsrrUr rCrrxcollections.abcrrrrtypingrr urllib.parser opentelemetry.util._oncer pydanticr requestsr rlogfire._internal.clientrlogfire._internal.configr!logfire._internal.server_responserrlogfire._internal.utilsrlogfire.variables.abstractrrrrrr&rrrrrrrr r!r"r#r$logfirer&__all__r'rqr[rYrBsq222222 ######2222222222%%%%%%%% ))))))$$$$$$..............555555cccccccc666666                            <NNN;;;;;; -j j j j j $4j j j j j r[