o wi@sddlmZddlmZddlmZmZmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZmZdd lmZed Zd#ddZee ee ee efZ GdddeeZd#ddZd#ddZd#ddZd#ddZd#ddZd$ddZ d$d d!Z!d"S)%) annotations)Enum) AnnotatedAnyTypeVar)BeforeValidatorJsonPlainSerializerPydanticUseDefault)to_json)AndMongoLikeFilterOr) simplify_exprTvrreturncCst|ttfr |St|S)zIn case the incoming value isn't serialized JSON, reserialize it. This lets us use `Json[...]` fields with values that are already deserialized. ) isinstancestrbytesr rrQ/home/ubuntu/.local/lib/python3.10/site-packages/wandb/automations/_validators.py ensure_jsonsrc@s(eZdZdZd ddZed d d Zd S)LenientStrEnumzA string enum allowing for case-insensitive lookups by value. May include other internal customizations if needed. Note: This is a bespoke, internal implementation and NOT intended as a backport of `enum.StrEnum` from Python 3.11+. rrcCs|jSN)name)selfrrr__repr__-szLenientStrEnum.__repr__valueobjectrcs.t|tr|tfdd|DdSdS)Nc3s"|] }|jkr|VqdSr)r!lower).0errr 5s z+LenientStrEnum._missing_..)rrr#next)clsr!rrr _missing_0s zLenientStrEnum._missing_N)rr)r!r"rr)__name__ __module__ __qualname____doc__r classmethodr)rrrrr$s  rcCs|durt|S)zA "before"-mode field validator that coerces `None` to the field default. See: https://docs.pydantic.dev/2.11/api/pydantic_core/#pydantic_core.PydanticUseDefault Nr rrrrdefault_if_none9sr/cCst|tr |S|Sr)rrstripupperrrrr upper_if_strCsr2cCs`ddlm}m}ddlm}m}m}t||r||St||r.| r'|n|}||S|S)zHConvert eligible objects (including wandb types) to an automation scope.r)ArtifactCollectionProjectr ) ProjectScope_ArtifactPortfolioScope_ArtifactSequenceScope) wandb.apis.publicr3r4scopesr5r6r7rmodel_validate is_sequence)rr3r4r5r6r7typrrr parse_scopeHs    r=cCsddlm}m}m}m}m}m}t||r#|ddd|jii| St||r6|ddd|jii| St||rD|di| S|S)zBIf necessary (and possible), convert the object to a saved action.r  DoNothingSavedNoOpActionSavedNotificationActionSavedWebhookActionSendNotification SendWebhook integrationidNr) actionsr?r@rArBrCrDrintegration_id model_dumprr?r@rArBrCrDrrrparse_saved_actionVs    rKcCsddlm}m}m}m}m}m}t||r"|dd|jj i| St||r4|dd|jj i| St||rB|di| S|S)zCIf necessary (and possible), convert the object to an input action.r r>rHNr) rGr?r@rArBrCrDrrErFrIrJrrrparse_input_actionns    rLfrcCstt|S)zWrap a run filter in an `And` operator if it's not already. This is a necessary constraint imposed elsewhere by backend/frontend code. )rwraprrMrrrwrap_run_event_run_filtersrPcCsttt|S)zWrap filters as `{"$or": [{"$and": []}]}`. This awkward format is necessary because the frontend expects it. )rrNrrrOrrrwrap_mutation_event_filtersrQN)rrrr)rMrrr)" __future__renumrtypingrrrpydanticrrr pydantic_corer wandb._pydanticr _filtersrrr_filters.filterutilsrrr JsonEncodedrrr/r2r=rKrLrPrQrrrrs(