d/ dZdgZddlZddlZddlZddlmZddlZddl m Z ddl m Z  ddlZdZn #e$rdZYnwxYwd d e jiiZd d e jiiZd Zd Zejejd ddejddddejdddddZdS)z8 Magic functions for rendering vega-lite specifications vegaliteN)magic_arguments)curried)v5TF vega-lite5cX|t|tr|St|tjr't j||St|trd|iStj d t|d|S)z0Convert input data to data for use within schemaNurlzdata of type {} not recognized) stacklevel) isinstancedictpd DataFramerpipegetstrwarningswarnformattype)datadata_transformerss .lib/python3.11/site-packages/altair/_magics.py _prepare_datar&s |z$--| D", ' '|D"3"7"7"9"9::: D#  t} 6==d4jjIIVWXXXX ctj}|td||jvr"t d||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)IPython get_ipython ValueErroruser_ns NameErrorr)nameips r _get_variabler%3sl    B z >    2: ++16$<<    :d rr?zBlocal variablename of a pandas DataFrame to be used as the dataset)nargshelpz-vz --versionversionr)destdefaultz-jz--jsonjson store_true)r*actionc8tjt|}ddi}||j}|tdvsJtd|}t d|}|jrt j|}ndts= t j|}nG#t j $r}td|d}~wwxYwtj |tj }|j't|j} t!| ||d<||S)aHCell magic for displaying vega-lite visualizations in CoLab. %%vegalite [dataframe] [--json] [--version='v5'] 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. rrrzH%%vegalite: spec is not valid JSON. Install pyyaml to parse spec as yamlN)Loaderr)rparse_argstringrr) RENDERERS TRANSFORMERSr,loadsYAML_AVAILABLEJSONDecodeErrorr yamlload SafeLoaderrr%r) linecellargsexisting_versionsr)VegaLiterspecerrrs rrrCs)$  *8T : :Ds  -G i , , , , ,%g.H$[1': y 7z$  7 :d##DD#   7   ydo666 yTY''$T+<==V 8D>>sBC+B;;C)__doc____all__r,rr IPython.corerpandasrtoolzraltair.vegaliter vegalite_v5r7r5 ImportErrorr>r2rr3rr%argumentrrrrKs , ((((((------KKKNNNNN  [ !   [ *        ! "" M $ )TJJJ$vlKKK""LKKJ #""""s 0::