U /e!@sNddlmZddlZeeZddlmZddlm Z dZ ddZ dd Z dS) ) annotationsN)StaticLayoutProvider) GraphRenderer from_networkxc  st}ddt|jddD}tt|}|D].fdd|jddD}t|}||<q0d|kr|ddlm}|d t||d<t}d d|jddD}tt|}|D].fd d|jddD}t|}||<qd |ksd |krddlm}|ddd|D|d <dd|D|d <t} || j j _ || j j _ t |rl||f|} n<|} | j j j d} t| t|krddlm}|dt| d| _| S)a* Generate a ``GraphRenderer`` from a ``networkx.Graph`` object and networkx layout function. Any keyword arguments will be passed to the layout function. Only two dimensional layouts are supported. Args: graph (networkx.Graph) : a networkx graph to render layout_function (function or dict) : a networkx layout function or mapping of node keys to positions. The position is a two element sequence containing the x and y coordinate. Returns: instance (GraphRenderer) .. note:: Node and edge attributes may be lists or tuples. However, a given attribute must either have *all* lists or tuple values, or *all* scalar values, for nodes or edges it is defined on. .. warning:: Node attributes labeled 'index' and edge attributes labeled 'start' or 'end' are ignored. If you want to convert these attributes, please re-label them to other names. Raises: ValueError cSs"g|]}|dD]}|qqSkeys).0nodeattr_keyr8/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/plotting/graph.py Esz!from_networkx..T)datacs(g|] \}}|kr |ndqSNr )r _Z node_attrr rrrJsindexr)warnzConverting node attributes labeled 'index' are skipped. If you want to convert these attributes, please re-label with other names.cSs"g|]}|dD]}|qqS)rr )r Zedger rrrrZscs*g|]"\}}}|kr"|ndqSrr )r rZ edge_attrrrrr_sstartendzConverting edge attributes labeled 'start' or 'end' are skipped. If you want to convert these attributes, please re-label them with other names.cSsg|] }|dqS)rrr xrrrrkscSsg|] }|dqSrrrrrrrlsznNode keys in 'layout_function' don't match node keys in the graph. These nodes may not be displayed correctly.) graph_layout)dictlistZnodesset_handle_sublistswarningsredgesrZ node_rendererZ data_sourcerZ edge_renderercallabler rZlayout_provider) graphZlayout_functionkwargsZ node_dictZnode_attr_keysvaluesrZ edge_dictZedge_attr_keysZgraph_rendererrZ node_keysrrrr$sN                cCs>tdd|Dr:tdd|Ds,tddd|DS|S)Ncss|]}t|ttfVqdSr isinstancertuplerrrr sz#_handle_sublists..css$|]}|dk rt|ttfVqdSrr&rrrrr)sz;Can't mix scalar and non-scalar values for graph attributescSs g|]}|dkrgnt|qSr)rrrrrrsz$_handle_sublists..)anyall ValueError)r%rrrrs r) __future__rlogging getLogger__name__logZ models.graphsrZmodels.renderersr__all__rrrrrr s    e