o tf @sJdZddlmZddlmZddlmZGdddeZGdd d Z d S) z"An event schema registry.) annotations)Any) EventSchemac@seZdZdZdS)SchemaRegistryExceptionz:Exception class for Jupyter Events Schema Registry Errors.N)__name__ __module__ __qualname____doc__r r g/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/jupyter_events/schema_registry.pyr src@speZdZdZd$d%ddZd&d d Zd'd dZd(ddZed)ddZ d*ddZ d+ddZ d,ddZ d-d"d#Z dS).SchemaRegistryz>A convenient API for storing and searching a group of schemas.Nschemasdict[str, EventSchema] | NonecCs|pi|_dS)zInitialize the registry.N_schemas)selfrr r r __init__zSchemaRegistry.__init__keystrreturnboolcCs ||jvS)z:Syntax sugar to check if a schema is found in the registryr)rrr r r __contains__s zSchemaRegistry.__contains__cCsddd|jDS)zThe str repr of the registry.z, cSsg|]}t|qSr )r).0sr r r sz+SchemaRegistry.__repr__..)joinrvaluesrr r r __repr__szSchemaRegistry.__repr__ schema_objrNonecCs2|j|jvrd|jd}t|||j|j<dS)Nz The schema, zB, is already registered. Try removing it and registering it again.)idrr)rr!msgr r r _adds  zSchemaRegistry._add list[str]cCst|jSN)listrkeysrr r r schema_ids%rzSchemaRegistry.schema_idsschema$dict[str, Any] | (str | EventSchema)cCs t|ts t|}|||S)zAdd a valid schema to the registry. All schemas are validated against the Jupyter Events meta-schema found here: ) isinstancerr%)rr+r r r register)s  zSchemaRegistry.registerid_cCs2z|j|WStyd|d}t|dw)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?NrKeyErrorrr/r$r r r get4s    zSchemaRegistry.getcCs4z|j|=WdStyd|d}t|dw)z_Remove a given schema. If the schema is not found, this will raise a KeyError. r0r1Nr2r4r r r removeAs   zSchemaRegistry.removedatadict[str, Any]cCs||}||dS)zIValidate an event against a schema within this registry. N)r5validate)rr/r7r+r r r validate_eventNs zSchemaRegistry.validate_eventr')rr)rrrr)rr)r!rrr")rr&)r+r,rr)r/rrr)r/rrr")r/rr7r8rr")rrr r rrr r%propertyr*r.r5r6r:r r r r r s        r N) r __future__rtypingrr+r Exceptionrr r r r r s