Mh ddZddlmZddlmZddlmZGddeZGdd Z d S) z"An event schema registry.) annotations)Any) EventSchemaceZdZdZdS)SchemaRegistryExceptionz:Exception class for Jupyter Events Schema Registry Errors.N)__name__ __module__ __qualname____doc__^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_events/schema_registry.pyrr sDDDDrrcleZdZdZdddZdd Zdd ZddZed dZ d!dZ d"dZ d#dZ d$dZ dS)%SchemaRegistryz>A convenient API for storing and searching a group of schemas.Nschemasdict[str, EventSchema] | Nonec|pi|_dS)zInitialize the registry.N_schemas)selfrs r__init__zSchemaRegistry.__init__s07 2 rkeystrreturnboolc||jvS)z:Syntax sugar to check if a schema is found in the registryr)rrs r __contains__zSchemaRegistry.__contains__sdm##rcndd|jDS)zThe str repr of the registry.z, c,g|]}t|Sr )r).0ss r z+SchemaRegistry.__repr__..sBBBa3q66BBBr)joinrvaluesrs r__repr__zSchemaRegistry.__repr__s1zzBB4=+?+?+A+ABBBCCCr schema_objrNonect|j|jvrd|jd}t|||j|j<dS)Nz The schema, zB, is already registered. Try removing it and registering it again.)idrr)rr(msgs r_addzSchemaRegistry._addsV =DM ) )Hz}HHH *#.. .'1 jm$$$r list[str]cNt|jSN)listrkeysr&s r schema_idszSchemaRegistry.schema_ids%sDM&&(()))rschema$dict[str, Any] | (str | EventSchema)cxt|tst|}|||S)zAdd a valid schema to the registry. All schemas are validated against the Jupyter Events meta-schema found here: ) isinstancerr-)rr4s rregisterzSchemaRegistry.register)s: &+.. ) ((F & rid_cf |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?NrKeyErrorrr9r,s rgetzSchemaRegistry.get4s^ *=% % * * *NNNN 3--T )  *s !0c` |j|=dS#t$rd|d}t|dwxYw)z_Remove a given schema. If the schema is not found, this will raise a KeyError. r;r<Nr=r?s rremovezSchemaRegistry.removeAsa * c""" * * *NNNN 3--T )  *s !-datadict[str, Any]cZ||}||dS)zIValidate an event against a schema within this registry. N)r@validate)rr9rCr4s rvalidate_eventzSchemaRegistry.validate_eventNs+#rr0)rr)rrrr)rr)r(rrr))rr.)r4r5rr)r9rrr)r9rrr))r9rrCrDrr))r r r r rrr'r-propertyr3r8r@rBrGr rrrr sHH>>>>>$$$$DDDD2222***X*     * * * * * * * *rrN) r __future__rtypingrr4r Exceptionrrr rrrLs """"""EEEEEiEEEFFFFFFFFFFr