U ꥡcN@sddlZddlZddlZddlZddlZddlZddlZda ddZ ddZ ej ddZ d d Zd d Zdd dZGdddejZGdddeZeZGdddeZddZGdddeZdS)NTcCsdadSNT DEBUG_MODErr9/tmp/pip-unpacked-wheel-y9_o96ar/altair/utils/schemapi.pyenable_debug_modesrcCsdadSrrrrrrdisable_debug_modesrccst}|az dVW5|aXdSNr)argoriginalrrr debug_modes  r ccsFt}|h}|rB||O}tjdd|D}||D] }|Vq4q dS)z=Breadth-first sequence of all classes which inherit from cls.css|]}t|VqdSr )set__subclasses__).0clsrrr 0sz_subclasses..N)r union)rseenZ current_setrrr _subclasses*s rcst|tr|jdSt|tttjfr>fdd|DSt|tr`fdd|DSt |drr|St|tj rt |St|t j tjfrt |S|SdS)z+Convert an object to a dict representation.validatecontextcsg|]}t|qSr)_todictrvrrrr :sz_todict..cs&i|]\}}|tk r|t|qSr) Undefinedrrkrrrr <sz_todict..to_dictN) isinstance SchemaBaser!listtuplenpZndarraydictitemshasattrnumberfloatpdZ TimestampZ datetime64 isoformat)objrrrrrr5s     rc Cs<tj|p |}d|kr8||d }|}W5QRXq|S)zResolve schema references.z$ref) jsonschema RefResolver from_schemaZ resolving)schemarootresolverZresolvedrrr_resolve_referencesKs r5cs4eZdZdZfddZeddZddZZS)SchemaValidationErrorzBA wrapper for jsonschema.ValidationError with friendlier tracebackcs"tt|jf||||_dSr )superr6__init__ _get_contentsr.)selfr.err __class__rrr8WszSchemaValidationError.__init__c stz }Wnbtk rnztj}Wn tk rLtj}YnXfdd|jddD}YnX|S)z7Get a dictionary with the contents of a ValidationErrorcsi|]}|t|qSr)getattr)rkeyr;rrr hsz7SchemaValidationError._get_contents..N)Z _contentsAttributeErrorinspectgetfullargspecr8 getargspecargs)r;contentsspecrr@rr9[s "z#SchemaValidationError._get_contentscCsT|jj}d|j|jg}||jddd|ddD}d||j|j S)Nz{}.{}z->css|]}|dkrt|VqdS)) propertiesZadditionalPropertiesZpatternPropertiesN)strrvalrrrrosz0SchemaValidationError.__str__..zGInvalid specification {}, validating {!r} {} ) r.r=format __module____name__extend schema_pathjoinZ validatormessage)r:rrRrrr__str__ks   zSchemaValidationError.__str__) rPrO __qualname____doc__r8 staticmethodr9rU __classcell__rrr<rr6Ts   r6c@s$eZdZdZdZddZddZdS) UndefinedTypez3A singleton object for marking undefined attributesNcOs&t|j|s tj|f|||_|jSr )r"_UndefinedType__instanceobject__new__)rrFkwargsrrrr]s zUndefinedType.__new__cCsdS)Nrrr:rrr__repr__szUndefinedType.__repr__)rPrOrVrWr[r]r`rrrrrZ~srZcseZdZdZdZdZdZejZ ddZ d*ddZ e fd d Z fd d Zd dZddZddZddZddZd+ddZdgiddfddZeddZed,ddZed-d d!Zed.d"d#Zed/d$d%Zed0d&d'Zd(d)ZZS)1r#zBase class for schema wrappers. Each derived class should set the _schema class attribute (and optionally the _rootschema class attribute) which is used for validation. NTcOsv|jdkrtd|j|r0t|dks@tnt|dks@tt|d|t|d|trr|j rr|j dddS)NzMCannot instantiate object of type {}: _schema class attribute is not defined.r)rrA_args_kwdsTr) _schema ValueErrorrNr=lenAssertionErrorr\ __setattr__r _class_is_valid_at_instantiationr!)r:rFkwdsrrrr8s  zSchemaBase.__init__rc sdd}d fdd z t|}Wntk r:d}YnXd}|rT|sT||dStd|j|j|j}W5QRX|r|D]}|||||<q|S) a Return a copy of the object Parameters ---------- deep : boolean or list, optional If True (default) then return a deep copy of all dict, list, and SchemaBase objects within the object structure. If False, then only copy the top object. If a list or iterable, then only copy the listed attributes. ignore : list, optional A list of keys for which the contents should not be copied, but only stored by reference. cSsFt|tr|jddSt|tr,|ddSt|tr>|S|SdS)NF)deep)r"r#copyr$r')r.rrr _shallow_copys     z&SchemaBase.copy.._shallow_copyrc st|trftfdd|jD}fdd|jD}td|j||W5QRSQRXnDt|trfdd|DSt|t rfdd|DS|SdS) Nc3s|]}|VqdSr r)rr  _deep_copyrrrsz6SchemaBase.copy.._deep_copy..cs*i|]"\}}||kr"|dn|qSignorerrrorrrrr sz7SchemaBase.copy.._deep_copy..Fcsg|]}|dqSrprrrsrrrsz7SchemaBase.copy.._deep_copy..cs*i|]"\}}||kr"|dn|qSrprrrsrrr s) r"r#r%rarbr(r r=r$r')r.rrrFrjrnrqrros       z#SchemaBase.copy.._deep_copyFTrq)r)r$ TypeErrorr r=rarb_get)r:rkrrrmZ deep_is_listrlattrrrnrrls     zSchemaBase.copycCs|j|t}|tkr|}|S)z3Get an attribute, returning default if not present.)rbgetr)r:rvdefaultrrrruszSchemaBase._getcs`|dkrt||jkr"|j|Sztt|j}Wn tk rRtt|j}YnX||SdS)Nrb)rBrbr7r# __getattr____getattribute__)r:rvZ_getattrr<rrrys  zSchemaBase.__getattr__cCs||j|<dSr rbr:itemrLrrrrhszSchemaBase.__setattr__cCs |j|Sr r{)r:r}rrr __getitem__szSchemaBase.__getitem__cCs||j|<dSr r{r|rrr __setitem__ szSchemaBase.__setitem__cCs^|jrDddt|jD}dd|}d|jj|ddSd|jj|jdSdS) Ncss&|]\}}|tk rd||VqdS)z{}: {!r}N)rrNrr?rLrrrrsz&SchemaBase.__repr__.. z, z {0}({{{1} }})z z{}({!r})r) rbsortedr(rSrNr=rPreplacera)r:rFrrrr` s  zSchemaBase.__repr__cCs(t|t|ko&|j|jko&|j|jkSr )typerarb)r:otherrrr__eq__s   zSchemaBase.__eq__c s|dkr i}dkrg|dkr$dnd}|jrJ|jsJt|jd||d}n:|jsttfdd|jD||d}ntd|j|rz||Wn.tj k r}zt ||W5d}~XYnX|S) aReturn a dictionary representation of the object Parameters ---------- validate : boolean or string If True (default), then validate the output dictionary against the schema. If "deep" then recursively validate all objects in the spec. This takes much more time, but it results in friendlier tracebacks for large objects. ignore : list A list of keys to ignore. This will *not* passed to child to_dict function calls. context : dict (optional) A context dictionary that will be passed to all child to_dict function calls Returns ------- dct : dictionary The dictionary representation of this object Raises ------ jsonschema.ValidationError : if validate=True and the dict does not conform to the schema NrkFrrcsi|]\}}|kr||qSrrrrqrrr Gsz&SchemaBase.to_dict..zF{} instance has both a value and properties : cannot serialize to dict) rarbrr(rerNr=rr/ValidationErrorr6)r:rrrrZ sub_validateresultr;rrqrr!"s0 zSchemaBase.to_dictcKs(|j|||d}tj|f||d|S)aEmit the JSON representation for this object as a string. Parameters ---------- validate : boolean or string If True (default), then validate the output dictionary against the schema. If "deep" then recursively validate all objects in the spec. This takes much more time, but it results in friendlier tracebacks for large objects. ignore : list A list of keys to ignore. This will *not* passed to child to_dict function calls. context : dict (optional) A context dictionary that will be passed to all child to_dict function calls indent : integer, default 2 the number of spaces of indentation to use sort_keys : boolean, default True if True, sort keys in the output **kwargs Additional keyword arguments are passed to ``json.dumps()`` Returns ------- spec : string The JSON specification of the chart object. )rrrr)indent sort_keys)r!jsondumps)r:rrrrrrr^dctrrrto_jsonWszSchemaBase.to_jsoncCsttS)z5Return the set of classes used within cls.from_dict())rr#rrrr_default_wrapper_classesxsz#SchemaBase._default_wrapper_classescCs2|r|||dkr|}t|}|||S)aConstruct class from a dictionary representation Parameters ---------- dct : dictionary The dict from which to construct the class validate : boolean If True (default), then validate the input against the schema. _wrapper_classes : list (optional) The set of SchemaBase classes to use when constructing wrappers of the dict inputs. If not specified, the result of cls._default_wrapper_classes will be used. Returns ------- obj : Schema object The wrapped schema Raises ------ jsonschema.ValidationError : if validate=True and dct does not conform to the schema N)rr _FromDict from_dict)rrrZ_wrapper_classes converterrrrr}s  zSchemaBase.from_dictcKstj|f|}|j||dS)aInstantiate the object from a valid JSON string Parameters ---------- json_string : string The string containing a valid JSON chart specification. validate : boolean If True (default), then validate the input against the schema. **kwargs : Additional keyword arguments are passed to json.loads Returns ------- chart : Chart object The altair Chart object built from the specification. rc)rloadsr)rZ json_stringrr^rrrr from_jsonszSchemaBase.from_jsoncCs6|dkr|j}tj|jp|j}tj|||j|dS)zj Validate the instance against the class schema in the context of the rootschema. N)rr4)rdr/r0r1 _rootschemar _validator)rinstancer2r4rrrrszSchemaBase.validatecCst|p |j|jp|jp|dS)zIResolve references in the context of this object's schema or root schema.)r2r3)r5rdr)rr2rrrresolve_referencesszSchemaBase.resolve_referencescCsRt|did}||p|jdi}tj|jp6|j}tj|||i|dS)zf Validate a property against property schema in the context of the rootschema FrrIr4) rrrdrwr/r0r1rr)rnamevaluer2propsr4rrrvalidate_propertyszSchemaBase.validate_propertycCst|jSr )r$rbkeysr_rrr__dir__szSchemaBase.__dir__)Tr)TNN)TN)T)N)N)N)rPrOrVrWrdrrir/ZDraft7Validatorrr8rlrruryrhr~rr`rr!r classmethodrrrrrrrrYrrr<rr#sD =   6 !      r#cOs|r |dS|S)Nrr)rFrjrrr _passthroughsrc@s<eZdZdZdZddZed ddZdddefd d Z dS) ra!Class used to construct SchemaBase class hierarchies from a dict The primary purpose of using this class is to be able to build a hash table that maps schemas to their wrapper classes. The candidate classes are specified in the ``class_list`` argument to the constructor. )Z definitionstitle descriptionz$schemaidcCs<tt|_|D]&}|jdk r|j||j|qdSr ) collections defaultdictr$ class_dictrd hash_schemaappend)r:Z class_listrrrrr8s  z_FromDict.__init__Tcs\jr&t|tr&fdd|D}|r@tj|dd}t|Sfddt|SdS)a Compute a python hash for a nested dictionary which properly handles dicts, lists, sets, and tuples. At the top level, the function excludes from the hashed schema all keys listed in `exclude_keys`. This implements two methods: one based on conversion to JSON, and one based on recursive conversions of unhashable to hashable types; the former seems to be slightly faster in several benchmarks. cs i|]\}}|jkr||qSr)_hash_exclude_keysrrrrr s z)_FromDict.hash_schema..T)rcsft|tr$tfdd|DSt|tr._freeze..)r"r' frozensetr(r mapr$r%)rLrrrrs  z&_FromDict.hash_schema.._freezeN)rr"r'r(rrhash)rr2Zuse_jsonsr)rrrrs   z_FromDict.hash_schemaNc s|dk|dkkrtd|dkr4|p(|j}p2|jp:|t|trJ|S|dkrtj|}|rp|d}n|}t|}d|ksd|kr|dg|dg}|D]V}t j } zt j ||| dWnt j k rYqYqXj|||dSqt|trd|di} i} |D]0\} } | | krNj| | | d } | | | <q(|f| St|tr|d ifd d |D}||S||SdS) z.Construct an object from a dict representationNz0Must provide either cls or schema, but not both.rZanyOfoneOfr)r2 rootschema default_classrIr2rr(csg|]}j|dqS)r)rrKZ item_schemarr:rrrEsz'_FromDict.from_dict..)rerdrr"r#rrr5rwr/r0r1rrrr'r(r$)r:rrr2rrmatchesZschemasZpossible_schemar4rrjr?rLrrrrsV               z_FromDict.from_dict)T) rPrOrVrWrr8rrrrrrrrrs $r)N)r contextlibrCrr/Znumpyr&Zpandasr,rrrcontextmanagerr rrr5rr6r\rZrr#rrrrrrs,   *I