yjfVddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZddlmZddlmZmZdd lmZdd lmZ dd lmZn#e$rdCdZYnwxYwdZ GddeZ!GddeZ"GddeZ#GddeZ$GddeZ%GddeZ&GddeZ'Gd d!eZ(ed"e e!e"ze#ze$ze%ze&ze'ze(zed#fZ)e e*ed$%ed&d'ifZ+ Gd(d)eZ,Gd*d+eZ-Gd,d-eZ.Gd.d/eZ/Gd0d1eZ0Gd2d3eZ1Gd4d5eZ2dDd;Z3Gd<d=eZ4dEd@Z5dFdBZ6dS)G) annotationsN)MappingSequence)cached_property) AnnotatedAnyLiteral) BaseModelFieldValidationErrorWithJsonSchemafield_validatormodel_validator) TypeAliasType)LocalVariablesOptionsVariablesOptions)ResolvedVariable)Variable) DiscriminatorargsrkwargsreturncdSN)rrs U/home/lighthouse/health/venv/lib/python3.11/site-packages/logfire/variables/config.pyrrs )KeyIsNotPresent KeyIsPresent LabeledValueLabelRef LatestVersionrRolloutRolloutOverrideValueDoesNotEqualValueDoesNotMatchRegex ValueEquals ValueIsIn ValueIsNotInValueMatchesRegexVariableConfigVariablesConfigVariableTypeConfigcDeZdZUdZded< ded< dZded< dd ZdS)r'zACondition that matches when an attribute equals a specific value.str attributervaluez value-equalszLiteral['value-equals']kind attributesMapping[str, Any]rboolcb||jt|jkS)z1Check if the attribute equals the expected value.getr0objectr1selfr3s rmatcheszValueEquals.matches:s#~~dnfhh774:EErNr3r4rr5__name__ __module__ __qualname____doc____annotations__r2r<rrrr'r'0sbKKNNN-JJJ-$2D22221FFFFFFrr'cDeZdZUdZded< ded< dZded< dd ZdS)r%zICondition that matches when an attribute does not equal a specific value.r/r0rr1zvalue-does-not-equalzLiteral['value-does-not-equal']r2r3r4rr5cb||jt|jkS)z:Check if the attribute does not equal the specified value.r7r:s rr<zValueDoesNotEqual.matchesIs%~~dnfhh774:EErNr=r>rrrr%r%?sbSSNNN-JJJ1,BDBBBB1FFFFFFrr%cDeZdZUdZded< ded< dZded< dd ZdS)r(zECondition that matches when an attribute value is in a set of values.r/r0 Sequence[Any]valuesz value-is-inzLiteral['value-is-in']r2r3r4rr5cb||jt}||jvS)z3Check if the attribute value is in the allowed set.r8r0r9rHr;r3r1s rr<zValueIsIn.matchesYs(t~vxx88 ##rNr=r>rrrr(r(Os_OONNN-5#0D00001$$$$$$rr(cDeZdZUdZded< ded< dZded< dd ZdS)r)zICondition that matches when an attribute value is not in a set of values.r/r0rGrHzvalue-is-not-inzLiteral['value-is-not-in']r2r3r4rr5cb||jt}||jvS)z8Check if the attribute value is not in the excluded set.rJrKs rr<zValueIsNotIn.matchesis*t~vxx88DK''rNr=r>rrrr)r)_s_SSNNN-9'8D88881((((((rr)cDeZdZUdZded< ded< dZded< dd ZdS)r*zGCondition that matches when an attribute value matches a regex pattern.r/r0str | re.Pattern[str]patternzvalue-matches-regexzLiteral['value-matches-regex']r2r3r4rr5c||j}t|tsdSt t j|j|S)z7Check if the attribute value matches the regex pattern.F)r8r0 isinstancer/r5researchrPrKs rr<zValueMatchesRegex.matcheszsHt~..%%% 5BIdlE22333rNr=r>rrrr*r*ps_QQNNN-"""";+@D@@@@1444444rr*cDeZdZUdZded< ded< dZded< dd ZdS)r&zNCondition that matches when an attribute value does not match a regex pattern.r/r0rOrPzvalue-does-not-match-regexz%Literal['value-does-not-match-regex']r2r3r4rr5c||j}t|tsdSt j|j| S)z>Check if the attribute value does not match the regex pattern.T)r8r0rRr/rSrTrPrKs rr<zValueDoesNotMatchRegex.matchessEt~..%%% 49T\51111rNr=r>rrrr&r&s_XXNNN-""""?2NDNNNN1222222rr&c8eZdZUdZded< dZded< d d Zd S)rz8Condition that matches when an attribute key is present.r/r0zkey-is-presentzLiteral['key-is-present']r2r3r4rr5c|j|vS)z4Check if the attribute key exists in the attributes.r0r:s rr<zKeyIsPresent.matchess~++rNr=r>rrrrrsPBBNNN=&6D66661,,,,,,rrc8eZdZUdZded< dZded< d d Zd S)rzrrrrrsPFFNNNA*>D>>>>1000000rr Conditionr2z^[a-zA-Z_][a-zA-Z0-9_]*$)rPtypestringc*eZdZUdZded< ded<dS)r z>A label pointing to a version with an inline serialized value.intversionr/serialized_valueNr?r@rArBrCrrrr r s0HHLLL255rr c.eZdZUdZdZded< ded<dS)r!z\A label pointing to a version via a reference to another label, 'latest', or 'code_default'.Nz int | Nonerar/ref)r?r@rArBrarCrrrr!r!s5ffG` HHHBBrr!c*eZdZUdZded< ded<dS)r"z-The latest (highest) version of the variable.r`rar/rbNrcrrrr"r"s077LLL3::rr"c~eZdZUdZiZded< ed dZede ddZ dd Z d S)r#z>Configuration for label selection with weighted probabilities.dict[str, float]labelsr$tuple[list[str | None], list[float]]c(g}g}|jD]/\}}||||0dt|z }|dkr*|d||||fS)Nr)riitemsappendsum)r; populationweightskvp_nones r_population_and_weightszRollout._population_and_weightss(* !K%%''  DAq   a NN1    S\\! A::   d # # # NN6 " " "7""rrsctd|Drtdt|dkrtd|S)Nc3"K|] }|dkV dS)rNr).0weights r z6Rollout._validate_label_proportions..s&33fvz333333rz'Label proportions must not be negative.g0D?z.Label proportions must not sum to more than 1.)anyrH ValueErrorro)clsrss r_validate_label_proportionsz#Rollout._validate_label_proportionssh 33 333 3 3 HFGG G qxxzz??Z ' 'MNN Nrseed str | Nonec|jsdStj|}|j\}}|||dS)aSelect a label based on configured weights using optional seeded randomness. Args: seed: Optional seed for deterministic label selection. If provided, the same seed will always select the same label. Returns: The name of the selected label, or None if no label is selected (when labels is empty or weights sum to less than 1.0). Nr)rirandomRandomruchoices)r;rrandrprqs r select_labelzRollout.select_labelsH{ 4}T""": G||J0033rN)rrj)rsrh)rrrr) r?r@rArBrirCrrur classmethodr~rrrrr#r#sHH!F!!!!/ # # #_ #_X[444444rr#c*eZdZUdZded< ded<dS)r$zDAn override of the default rollout when specific conditions are met.list[Condition] conditionsr#rolloutNrcrrrr$r$s3NNLEErr$ceZdZUdZded< dZded< iZded< d ed < d ed < dZd ed< dZded< dZ ded< dZ ded< dZ ded< e ddZ d'd(dZ d'ddd)d Z d*d+d&ZdS),r+zZConfiguration for a single managed variable including labels, versions, and rollout rules. VariableNamenameNr descriptionz"dict[str, LabeledValue | LabelRef]rir#rzlist[RolloutOverride] overrideszLatestVersion | Nonelatest_versionzdict[str, Any] | None json_schema type_namezlist[VariableName] | Nonealiasesexampleaftermodec |jjD]}||jvrtd|dt|jD]3\}}|jjD]!}||jvrtd|d|d"4|jD]L\}}t |tr2|jdvr)|j|jvrtd|d|jdM|S)NzLabel z8 present in `rollout.labels` is not present in `labels`.z present in `overrides[z&].rollout` is not present in `labels`.)latest code_defaultz has ref z" which is not present in `labels`.) rrir| enumeraterrmrRr!re)r;rrioverride labeled_values r_validate_labelszVariableConfig._validate_labels:s:$ i iA ## !g!!g!g!ghhh$%T^44 u uKAx%, u uDK''$%sa%s%s!%s%s%sttt( u !% 1 1 3 3 u u A}-22 u}7HPj7j7j $DK77$%sa%s%sM7:: rrLabeledValue | LabelRef_visitedset[str] | Nonetuple[str | None, int | None]ct|tr|jdkrdS|jdkr(|j|jj|jjfSd|jfS|t }|j|vr d|jfS||j|j |j}| d|jfS| ||S|j|jfS)a Follow ref chains to get the actual serialized value. Args: labeled_value: The LabeledValue or LabelRef to resolve. _visited: Set of already-visited ref names to detect cycles. Returns: A tuple of (serialized_value, version). rNNrN) rRr!rerrbrasetaddrir8r)r;rrtargets rrzVariableConfig.follow_refs mX . . 5 N22!z H,,&2.?ATA\\\]22255 H,,]222 LL* + + +[__]%677F~]222??6844 4-}/DDDrr)rrr3rrr)rrr3rrrrrr)rrrrrr)r?r@rArBrCrrirrrrrrrrrrrrrr+r+sydd 0"K""""&13F33339<$$$$Q+/N////N)-K----M I    E)-G----VGi_'"""#"*X\!3!3!3!3!3J%)/3/ ! / / / / / / dSW#E#E#E#E#E#E#Err+ceZdZUdZded< eddZedd Zd d Z d!d d d"dZ d#dZ d$dZ e d%dZd&dZd S)'r,z2Container for all managed variable configurations.z"dict[VariableName, VariableConfig] variablesrrc|jD]+\}}|j|krtd|d|jd,|S)Nz#`variables` has invalid lookup key z for value with name .)rrmrr|)r;rrrss r_validate_variablesz#VariablesConfig._validate_variablessbN((** n nDAqv{{ !lq!l!labag!l!l!lmmm rrdict[VariableName, str]cxi}|jD]}|jr|jD] }|j||< |Sr)rrHrr)r; alias_map var_configaliass r _alias_mapzVariablesConfig._alias_mapsY79 .//11 7 7J! 7'/77E'1Ie$$rNonec<|jdddS)zBInvalidate the cached alias map so it gets rebuilt on next access.rN)__dict__pop)r;s r_invalidate_alias_mapz%VariablesConfig._invalidate_alias_maps  ,-----rNrrrrrr3rrResolvedVariable[str | None]c||}|t|ddS||||\}}}t|j|||dS)aEvaluate a managed variable configuration and resolve the selected value. Args: name: The name of the variable to resolve. targeting_key: Optional key for deterministic label selection. attributes: Optional attributes for condition matching. label: Optional explicit label to select, bypassing rollout. Returns: A ResolvedVariable containing the serialized value (or None if not found). Nunrecognized_variable)rr1reasonrresolved)rr1rrar)_get_variable_configrrr) r;rrr3rvariable_configrbrras rresolve_serialized_valuez(VariablesConfig.resolve_serialized_values&33D99  "#TBYZZZ Z4C4Q4Q :U5R5 5 1.'  %"      rVariableConfig | Nonec|j|}||S||jvr%|j|j|SdSr)rr8r)r;rconfigs rrz$VariablesConfig._get_variable_configsS##D))  M 4? " ">%%dod&;<< <trlist[Variable[Any]]&dict[str, dict[str | None, Exception]]ci}|D]Y} ||j}|td|j|jD]p\}}t |t rV |j|j ;#t$r(}|| |ji|<Yd}~hd}~wwxYwq|j [ |j|j j n5#t$r(}|| |jid<Yd}~nd}~wwxYw%#t$r)}|| |jid<Yd}~Sd}~wwxYw|S)aFValidate that all variable label values can be deserialized to their expected types. Args: variables: List of Variable instances to validate against this configuration. Returns: A dict mapping variable names to dicts of label names (or None for general errors) to exceptions. Nz!No config for variable with name r)rrr|rirmrRr type_adapter validate_jsonrbr setdefaultr Exception)r;rerrorsvariablerrrres rget_validation_errorsz%VariablesConfig.get_validation_errorss:<! ? ?H ?228=AA>$%Z%Z%Z[[[(. (;(;(=(=HH$A}!->>HH$1?? @^____.HHHFGF--hmR@@CCCCCCHH (4K -;;F!!(-44T:::::: ? slA&D*0BD* CB=8D*=C D*$C32D*3 D%=D D* D%%D** E4EEc fddlm}i}|D]}|j}d}||js2|j|jd}t|j|j itig||}|||j<t|S)aCreate a VariablesConfig from a list of Variable instances. This creates a minimal config with just the name, schema, and example for each variable. No labels or versions are created - use this to generate a template config that can be edited. Args: variables: List of Variable instances to create configs from. Returns: A VariablesConfig with minimal configs for each variable. r)is_resolve_functionNzutf-8)ri)rrrirrrrr) logfire.variables.variablerrrdefault dump_jsondecoder+rrr#r,)rrvariable_configsrrrrs rfrom_variableszVariablesConfig.from_variables3s CBBBBB?A! 5 5H"/;;==K#'G&&x'788 \"/99(:JKKRRSZ[[#]$0r***'F/5 X] + +)9::::rotherc~t|j}||jt|S)a(Merge another VariablesConfig into this one. Variables in `other` will override variables with the same name in this config. Args: other: Another VariablesConfig to merge. Returns: A new VariablesConfig with variables from both configs. r)dictrupdater,)r;rmerged_variabless rmergezVariablesConfig.mergeYs< //000)9::::r)rr)rrr) rrrrr3rrrrr)rrrr)rrrr)rrrr,)rr,rr,)r?r@rArBrCrrrrrrrr staticmethodrrrrrr,r,s<<1111<_'"""#"_....%)/3  !       D    @#;#;#;\#;J ; ; ; ; ; ;rr,rrr3r4r5c@|D]}||sdSdS)zCheck if all conditions match the provided attributes. Args: conditions: List of conditions to evaluate. attributes: Attributes to match against. Returns: True if all conditions match, False otherwise. FT)r<)rr3 conditions rrris:    ,, 55  4rcJeZdZUdZded< ded< dZded< dZded <dS) r-z?Configuration for a variable type (reusable schema definition).r/rzdict[str, Any]rNrr source_hint)r?r@rArBrCrrrrrr-r-ysXII IIIO4"K""""2"K""""ffrr-tr/cXt|tr|jSt|S)a?Get the default name for a Python type. For regular classes, returns __name__ (e.g., 'FeatureConfig'). For unions, generics, or other complex types, returns str(type) (e.g., 'FeatureConfig | None'). Args: t: The Python type to get a name for. Returns: A string name for the type. )rRr]r?r/)rs rget_default_type_namers)!Tz q66Mrrct|tr-t|dd}t|dd}|r |r|d|SdS)zGet a source hint for a type based on its module and qualname. Args: t: The Python type to get a source hint for. Returns: A source hint string like 'myapp.config.FeatureConfig', or None if not available. r@NrAr)rRr]getattr)rmodulequalnames rget_source_hintrs`!T*L$//1nd33  *h *))x)) ) 4r)rrrrrr)rrr3r4rr5)rrrr/)rrrr)7 __future__r _annotationsrrScollections.abcrr functoolsrtypingrrr pydanticr r r r rrtyping_extensionsrlogfire._internal.configrrlogfire.variables.abstractrrrr ImportError__all__r'r%r(r)r*r&rrr\r/rr r!r"r#r$r+r,rr-rrrrrrs2222222 --------%%%%%%**********hhhhhhhhhhhhhhhh++++++877777////// &&&&&&&           * F F F F F) F F F F F F F F F F F $ $ $ $ $ $ $ $ (((((9((("44444 444$22222Y222( , , , , ,9 , , , 0 0 0 0 0i 0 0 0 M           !  !       f     ee,GHHH..Z`bjYkJlJllm 8 666669666CCCCCyCCC;;;;;I;;;0404040404i040404fFFFFFiFFFiEiEiEiEiEYiEiEiEX];];];];];i];];];@     g g g g g g g g"sA A%$A%