B '=[ ! @sddlmZddlZddlZddlZddlZddlZyddlmZddlm Z Wn6e k rZ zdZ e ee e WddZ [ XYnXddlmZiZddZd d Zdd d ZddZdddZddZdZdZddZGdddeZddZd ddZdS)!)print_functionN)ValidationError)Draft4Validatorz Jupyter notebook format depends on the jsonschema package: https://pypi.python.org/pypi/jsonschema Please install it first. ) import_itemcCspt|tr@x`|D]&\}}|dkr*d}nt|}|||<qWn,t|trlx t|D]\}}t|||<qTW|S)z relax any `additionalProperties`ZadditionalPropertiesT) isinstancedictitems_relax_additional_propertieslist enumerate)objkeyvalueir1lib/python3.7/site-packages/nbformat/validator.pyr s  r cCs8|dddddi|dddddi|S)NZ definitionsZcellZoneOfz$refz#/definitions/unrecognized_celloutputz!#/definitions/unrecognized_output)append)Zschemarrr_allow_undefined,s   rFcCs|dkrddlm}|}td|}t|dd}|dkr<|}||f}|tkry t|}Wntk rldSX||krt|}t|}t |t|<|ry t|}Wntk rdSXt|}t |t|<t|S)z%Load the JSON schema into a ValidatorN)current_nbformatz nbformat.v%sZnbformat_minorr) rrgetattr validators_get_schema_jsonAttributeErrorr r Validator)version version_minorrelax_add_propsrvZ current_minorZ version_tuple schema_jsonrrr get_validator5s2       r"c Cs<tjtj|j|j}t|}t|}WdQRX|S)zP Gets the json schema from a given imported library a nbformat version. N) ospathjoindirname__file__Znbformat_schemaopenjsonload)r schema_pathfr!rrrr^s rcCs0yt||||Wntk r&dSXdSdS)aChecks whether the given notebook JSON conforms to the current notebook format schema. Returns True if the JSON is valid, and False otherwise. To see the individual errors that were encountered, please use the `validate` function instead. FTN)validater)nbjsonrefrrrrrisvalidgs r0cCs |sdSdddd|DS)z (from jsonschema._utils.format_as_index, copied to avoid relying on private API) Construct a single string containing indexing operations for the indices. For example, [1, 2, "foo"] -> [1][2]["foo"] rz[%s]z][css|]}t|VqdS)N)repr).0indexrrr sz#_format_as_index..)r%)indicesrrr_format_as_indexvs r6@cCst|trddt|dtD}t|dtrNdt|dg|d<t|dtrtdt|dg|d<t|tkrdt|t|d <|St|trd d |dtD}t|tkr|d t|t|St|tr|dt }t|t kr |d 7}|S|SdS) zTruncate objects for use in validation tracebacks Cell and output lists are squashed, as are long strings, lists, and dicts. cSsi|]\}}t||qSr) _truncate_obj)r2kr rrr sz!_truncate_obj..NZcellsz...%i cells...Zoutputsz...%i outputs...z%i keys truncatedz...cSsg|] }t|qSr)r9)r2itemrrr sz!_truncate_obj..z...%i items truncated...) rrr r _ITEM_LIMITgetlenrstr _STR_LIMIT)r Z truncatedrrrr9s(      r9c@s8eZdZdZd ddZddZddZejd kr4eZ dS) NotebookValidationErrorzcSchema ValidationError with truncated representation to avoid massive verbose tracebacks. NcCs$||_t|jd||_|jj|_dS)Nr/)originalrr/message)selfrDr/rrr__init__sz NotebookValidationError.__init__cCs t|j|S)N)rrD)rFr rrr __getattr__sz#NotebookValidationError.__getattr__c Cs`|j}t|j}d|jdd|j|jp(dtt|j ddfddt|j t j |dd gS) zbCustom str for validation errors avoids dumping full schema and notebook to logs  rzFailed validating %r in %s%s:ZnotebookNzOn instance%s:N)width) rDr9instancer%rE validatorr/r6r Zrelative_schema_path relative_pathpprintZpformat)rFerrorrMrrr __unicode__s  z#NotebookValidationError.__unicode__))N) __name__ __module__ __qualname____doc__rGrHrRsys version_info__str__rrrrrCs   rCc Cs|jd}d}|drt|jtrRd|jkr>|jdd}nd|jkrR|jd}|ryt|j|||dWnftk r}z6|j|j|j|_t |||}|j dkr||_ |Sd}~XYnt k rYnXt ||S)zGet better ValidationError on oneOf failures oneOf errors aren't informative. if it's a cell type or output_type error, try validating directly based on the type for a better error message rJNZOfZ cell_typeZ_cellZ output_type)rr) r+endswithrrMrr-rrOextendbetter_validation_errorr/ ExceptionrC)rQrrr r/Z sub_errorZbetterrrrr]s0         r]c Cs|dkr ddlm}||\}}t|||d}|dkrJtd|tdSy&|rd||dd|iS||SWn.tk r}zt|||Wdd}~XYnXdS)zChecks whether the given notebook JSON conforms to the current notebook format schema. Raises ValidationError if not valid. N) get_version)rz&No schema for validating v%s notebooksz$refz#/definitions/%s) readerr`r"warningswarn UserWarningr-rr])r.r/rrrr`rNerrrr-s  r-)NNF)NNN)NNNF)Z __future__rr)r#rPrXrbZ jsonschemarrr ImportErrorreZ verbose_msgrAZipython_genutils.importstringrrr rr"rr0r6r>rBr9rCr]r-rrrrs2  "  )  $'