:vj VdZddlmZddlmZddlmZGddeZGdd Z y ) z"An event schema registry.) annotations)Any) EventSchemaceZdZdZy)SchemaRegistryExceptionz:Exception class for Jupyter Events Schema Registry Errors.N)__name__ __module__ __qualname____doc__o/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_events/schema_registry.pyrr sDrrcdeZdZdZd d dZddZddZddZeddZ ddZ dd Z dd Z dd Z y)SchemaRegistryz>A convenient API for storing and searching a group of schemas.Nc|xsi|_y)zInitialize the registry.N_schemas)selfschemass r__init__zSchemaRegistry.__init__s 07 2 rc||jvS)z:Syntax sugar to check if a schema is found in the registryr)rkeys r __contains__zSchemaRegistry.__contains__sdmm##rcdj|jjDcgc] }t|c}Scc}w)zThe str repr of the registry.z, )joinrvaluesstr)rss r__repr__zSchemaRegistry.__repr__s6zz4==+?+?+AB+Aa3q6+ABCCBsAc|j|jvrd|jd}t|||j|j<y)Nz The schema, zB, is already registered. Try removing it and registering it again.)idrr)r schema_objmsgs r_addzSchemaRegistry._addsM ==DMM )z}}o.HH *#. .'1 jmm$rcHt|jjSN)listrkeys)rs r schema_idszSchemaRegistry.schema_ids%sDMM&&())rc^t|ts t|}|j||S)zAdd a valid schema to the registry. All schemas are validated against the Jupyter Events meta-schema found here: ) isinstancerr%)rschemas rregisterzSchemaRegistry.register)s) &+. (F & rc` |j|S#t$rd|d}t|dwxYw)z^Fetch a given schema. If the schema is not found, this will raise a KeyError. The requested schema, R, was not found in the schema registry. Are you sure it was previously registered?NrKeyErrorrid_r$s rgetzSchemaRegistry.get4sK *==% % *(.NN 3-T )  *s-c^ |j|=y#t$rd|d}t|dwxYw)z_Remove a given schema. If the schema is not found, this will raise a KeyError. r0r1Nr2r4s rremovezSchemaRegistry.removeAsH * c" *(.NN 3-T )  *s ,cH|j|}|j|y)zIValidate an event against a schema within this registry. N)r6validate)rr5datar-s rvalidate_eventzSchemaRegistry.validate_eventNs#rr')rzdict[str, EventSchema] | None)rrreturnbool)r=r)r#rr=None)r=z list[str])r-z$dict[str, Any] | (str | EventSchema)r=r)r5rr=r)r5rr=r?)r5rr;zdict[str, Any]r=r?)r r r r rrr r%propertyr*r.r6r8r<r rrrr sBH>$D2**  * *rrN) r __future__rtypingrr-r Exceptionrrr rrrDs+ "EiEFFr