U ku f@srdZddlZddlZddlZddlmZddlmZddl m Z ddZ d d Z d d Z d dZGdddZdS)z JSON Schema URI resolution scopes and dereferencing https://tools.ietf.org/id/draft-zyp-json-schema-04.html#rfc.section.7 Code adapted from https://github.com/Julian/jsonschema N)parse)unquote)JsonSchemaDefinitionExceptioncCs|d|ddS)z9 Originally ID was `id` and since v7 it's `$id`. $idid)get)schemar ?/tmp/pip-unpacked-wheel-8kk9cw5n/fastjsonschema/ref_resolver.pyget_idsr cCsz|d}|rt|dng}|D]P}|dddd}t|trT|t|}q$||krf||}q$td|q$|S)zk Return definition from path. Path is unescaped according https://tools.ietf.org/html/rfc6901 /~1~0~zUnresolvable ref: {}) lstriprsplitreplace isinstancelistintrformat)r fragmentpartspartr r r resolve_paths   rcCst|SN)urlparseurlsplitgeturl)urir r r normalize,sr"c Cst|j}||kr"|||}npddlm}||}|pDd}zt| |}Wn2t k r}zt d ||W5d}~XYnX|S)z Resolve a remote ``uri``. .. note:: urllib library is used to fetch requests from the remote ``uri`` if handlers does notdefine otherwise. r)urlopenzutf-8z{} failed to decode: {}N)rrschemeurllib.requestr#infoget_content_charsetjsonloadsreaddecode ValueErrorrr)r!handlersr$resultr#reqencodingexcr r r resolve_remote0s  "r2c@sveZdZdZidifddZeifddZeje ddd Z eje d d d Z d dZ ddZ edddZdS) RefResolverz" Resolve JSON References. TcCs2||_||_||_||_||_||_||dS)a `base_uri` is URI of the referring document from the `schema`. `store` is an dictionary that will be used to cache the fetched schemas (if `cache=True`). Please notice that you can have caching problems when compiling schemas with colliding `$ref`. To force overwriting use `cache=False` or explicitly pass the `store` argument (with a brand new dictionary) N)base_uriresolution_scoper storecacher-walk)selfr4r r6r7r-r r r __init__Ns zRefResolver.__init__cKs(|t|trt|nd|fd|i|S)zA Construct a resolver from a JSON schema object. rr-)rdictr )clsr r-kwargsr r r from_schema`szRefResolver.from_schema)scopeccs,|j}t|||_z dVW5||_XdS)z: Context manager to handle current scope. N)r5rurljoin)r9r?Z old_scoper r r in_scopels  zRefResolver.in_scope)refc cst|j|}t|\}}|r>t||jkr>|jt|}n6|rL||jkrT|j}n t||j }|j rt||jt|<|j|j}}|||_|_z&| |t ||VW5QRXW5|||_|_XdS)zt Context manager which resolves a JSON ``ref`` and enters the resolution scope of this ref. N) rr@r5 urldefragr"r6r4r r2r-r7rAr)r9rBZnew_urir!rr Z old_base_uriZ old_schemar r r resolvingxs  zRefResolver.resolvingcCs t|jSr)r"r5)r9r r r get_uriszRefResolver.get_uricCsFdt|jdddddd}tdd|}|d}|S)zK Get current scope and return it as a valid function name. Z validate_r_r"rz($[^a-zA-Z]|[^a-zA-Z0-9]))rr5rresublowerrstrip)r9namer r r get_scope_names&zRefResolver.get_scope_name)nodec Cst|tr nd|kr>t|dtr>|d}t|j||d<nd|ksNd|krtt|tr|t|<||jt |j<| D]\}}t|t r| |qW5QRXn&| D]\}}t|t r| |qdS)zR Walk thru schema and dereferencing ``id`` and ``$ref`` instances z$refrrN) rboolstrrr@r5r rAr6r"itemsr;r8)r9rNrBrFitemr r r r8s   zRefResolver.walkN)__name__ __module__ __qualname____doc__r: classmethodr> contextlibcontextmanagerrPrArDrErMr;r8r r r r r3Hs   r3)rVrXr(rHurllibrr urllib.parser exceptionsrr rr"r2r3r r r r s