U ꥡcS @sbdZddgZddlZddlZddlZddlmZddlZddl m Z ddl m Z ddl mZdd lmZzddlZd ZWnek rd ZYnXd ejie jejd dZd ejie jejd dZddZddZeejddddejdddd dejdddddd dZeejdd!d"dejdddd#dejdddddd$dZdS)%z= Magic functions for rendering vega/vega-lite specifications vegavegaliteN)magic_arguments)curried)v3)v4)v5TF5)34)r vega-litecCs`|dkst|tr|St|tjr2t||St|trDd|iSt d t ||SdS)z0Convert input data to data for use within schemaNurlzdata of type {} not recognized) isinstancedictpdZ DataFramerpipegetstrwarningswarnformattype)datadata_transformersr2/tmp/pip-unpacked-wheel-y9_o96ar/altair/_magics.py _prepare_data,s  rcCs:t}|dkrtd||jkr0td||j|S)z+Get a variable from the notebook namespace.Nz[Magic command must be run within an IPython environemnt, in which get_ipython() is defined.z=argument '{}' does not match the name of any defined variable)IPythonZ get_ipython ValueErrorZuser_ns NameErrorr)nameiprrr _get_variable9s r"r*zClocal variable name of a pandas DataFrame to be used as the dataset)nargshelpz-vz --versionversion)destdefaultz-jz--jsonjson store_true)r'actionc Cstt|}|j}|tdks"ttd|}td|}dd}ztt||j }Wn"t k rxt d |YnX|j rt |}nDtszt |}Wqt jk rt dYqXntj|tjd}|rg|d<|D]0\} } t| } t| |} | | d<|d| q||S) a_Cell magic for displaying Vega visualizations in CoLab. %%vega [name1:variable1 name2:variable2 ...] [--json] [--version='5'] Visualize the contents of the cell using Vega, optionally specifying one or more pandas DataFrame objects to be used as the datasets. If --json is passed, then input is parsed as json rather than yaml. rcSsT|d}t|dkr&|d|dfSt|dkrB|d|dfStd|dS)N:rzinvalid identifier: '{}')splitlenrr)srrrnamevarbs    zvega..namevarzCould not parse arguments: '{}'zD%%vega: spec is not valid JSON. Install pyyaml to parse spec as yamlLoaderrr )rparse_argstringrr& RENDERERSAssertionError TRANSFORMERSlistmaprrrr)loadsYAML_AVAILABLEJSONDecodeErroryamlload FullLoaderr"rappend) linecellargsr&Vegarr2rspecr valZpreppedrrrrIs8        ?zBlocal variablename of a pandas DataFrame to be used as the datasetr cCstt|}|j}|tdks"ttd|}td|}|jrLt|}nDt szt|}Wqtj k r|t dYqXnt j |t jd}|jdk rt|j}t|||d<||S)aECell magic for displaying vega-lite visualizations in CoLab. %%vegalite [dataframe] [--json] [--version=3] Visualize the contents of the cell using Vega-Lite, optionally specifying a pandas DataFrame object to be used as the dataset. if --json is passed, then input is parsed as json rather than yaml. r zH%%vegalite: spec is not valid JSON. Install pyyaml to parse spec as yamlr3Nr)rr5rr&r6r7r8r)r;r<r=rr>r?r@rr"r)rBrCrDr&VegaLiterrFrrrrrs&       ) __doc____all__r)rrZ IPython.corerZpandasrZtoolzrZaltair.vegaliterZ vegalite_v3rZ vegalite_v4Z altair.vegarZvega_v5r>r< ImportErrorrErIr6rr8rr"argumentrrrrrrsZ          7