σ ¦–Υ\c@`sMddlmZmZmZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZmZedd ƒZd „Zd „Zd „Zejd ƒZejdƒZdd„Zd„Zdddiddd„ZeddgƒZeddgƒZd„Z ddd„Z!d„Z"dS(i(tabsolute_importtdivisiontprint_functionN(tpartiali(tapply(tistasktget_dependenciest ishashable(tfuncnametimport_requiredt key_splittgraphvizsmDrawing dask graphs requires the `graphviz` python library and the `graphviz` system library to be installed.cC`s²|d}|tkr#|d}nt|dƒrwt|jƒdkradjt|jdƒƒSt|jdƒ}n t|ƒ}td„|dDƒƒrͺdj|ƒS|SdS(sΔLabel for a task on a dot graph. Examples -------- >>> from operator import add >>> task_label((add, 1, 2)) 'add' >>> task_label((add, (add, 1, 2), 3)) 'add(...)' iitfuncss{0}(...)cs`s|]}t|ƒVqdS(N(t has_sub_tasks(t.0ti((s'lib/python2.7/site-packages/dask/dot.pys 'sN(RthasattrtlenR tformatRtany(ttasktfuncthead((s'lib/python2.7/site-packages/dask/dot.pyt task_labels     cC`s;t|ƒrtSt|tƒr3td„|DƒƒStSdS(s&Returns True if the task has sub taskscs`s|]}t|ƒVqdS(N(R (RR((s'lib/python2.7/site-packages/dask/dot.pys 2sN(RtTruet isinstancetlistRtFalse(R((s'lib/python2.7/site-packages/dask/dot.pyR -s  cC`s?ytt|ƒƒSWn$tk r:ttt|ƒƒƒSXdS(N(tstrthasht TypeError(tx((s'lib/python2.7/site-packages/dask/dot.pytname7s s([0-9a-z]{32})s>([0-9a-z]{8}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{12})cC`s·t|ƒ}x€ttfD]–}tj||ƒ}|dk rxo|jƒD]^}|dk r|j|t|ƒƒ}dj |ƒ}|||>> label('x') 'x' >>> label(('x', 1)) "('x', 1)" >>> from hashlib import md5 >>> x = 'x-%s-hello' % md5(b'1234').hexdigest() >>> x 'x-81dc9bdb52d04dc20036dbd8313ed055-hello' >>> label(x) 'x-#-hello' >>> from uuid import uuid1 >>> x = 'x-%s-hello' % uuid1() >>> x # doctest: +SKIP 'x-4c1a3d7e-0b45-11e6-8334-54ee75105593-hello' >>> label(x) 'x-#-hello' s#{0}t#N( Rt_HASHPATt_UUIDPATtretsearchtNonetgroupstgetRRtreplace(Rtcachetstpatterntmthtntlabel((s'lib/python2.7/site-packages/dask/dot.pyR0Bs    cC`sIt|tƒrA|d}t|ƒdkr7|\}nt|ƒSdSdS(s’ Label boxes in graph by chunk index >>> box_label(('x', 1, 2, 3)) '(1, 2, 3)' >>> box_label(('x', 123)) '123' >>> box_label('x') '' itN(RttupleRR(tkey((s'lib/python2.7/site-packages/dask/dot.pyt box_labelks    tBTcK`se|dkri}n|dkr*i}n|p3i}||d<|j|ƒtjd|d|d|ƒ}tƒ} xκ|jƒD]ά\} } t| ƒ} | | krύ| j| ƒ|j| iƒ} | j dt | ƒƒ| j ddƒ|j | | nt | ƒr,t| dfƒ}|| kr| j|ƒ|j| iƒ} | j dt | ƒƒ| j dd ƒ|j || n|j|| ƒxΛt|| ƒD]†}t|ƒ}|| kr| j|ƒ|j|iƒ} | j dt |ƒƒ| j ddƒ|j || n|j||ƒqŸWqt| ƒr| |kr|jt| ƒ| ƒqqW|S( Ntrankdirt graph_attrt node_attrt edge_attrR0tshapetboxtfunctiontcircle(R&tupdateR tDigraphtsettitemsR taddR(t setdefaultR4tnodeRR tedgeRR(tdsktdata_attributestfunction_attributesR6R7R8R9tkwargstgtseentktvtk_nametattrst func_nametdeptdep_name((s'lib/python2.7/site-packages/dask/dot.pyt to_graphviz~sP                  tjpegtpngtdottpdfcC`s‡d„}yddlj}Wntk r0|SX|tkrA|S|tkr`t|jd|ƒS|dkrs|jStd|ƒ‚dS(s Get the appropriate IPython display class for `format`. Returns `IPython.display.SVG` if format=='svg', otherwise `IPython.display.Image`. If IPython is not importable, return dummy function that swallows its arguments and returns None. c_`sdS(N(R&(targsRI((s'lib/python2.7/site-packages/dask/dot.pyt½R1iNRtsvgs)Unknown format '%s' passed to `dot_graph`( tIPython.displaytdisplayt ImportErrortIPYTHON_NO_DISPLAY_FORMATStIPYTHON_IMAGE_FORMATSRtImagetSVGt ValueError(RtdummyR\((s'lib/python2.7/site-packages/dask/dot.pyt_get_display_cls³s     tmydaskcK`st||}t|||ƒS(sΧ Render a task graph using dot. If `filename` is not None, write a file to disk with that name in the format specified by `format`. `filename` should not include an extension. Parameters ---------- dsk : dict The graph to display. filename : str or None, optional The name (without an extension) of the file to write to disk. If `filename` is None, no file will be written, and we communicate with dot using only pipes. Default is 'mydask'. format : {'png', 'pdf', 'dot', 'svg', 'jpeg', 'jpg'}, optional Format in which to write output file. Default is 'png'. **kwargs Additional keyword arguments to forward to `to_graphviz`. Returns ------- result : None or IPython.display.Image or IPython.display.SVG (See below.) Notes ----- If IPython is installed, we return an IPython.display object in the requested format. If IPython is not installed, we just return None. We always return None if format is 'pdf' or 'dot', because IPython can't display these formats natively. Passing these formats with filename=None will not produce any useful output. See Also -------- dask.dot.to_graphviz (RStgraphviz_to_file(RFtfilenameRRIRJ((s'lib/python2.7/site-packages/dask/dot.pyt dot_graphΡs%c`s ddddddg}|dkrkt‡fd†|Dƒƒrktjjˆƒ\‰}|djƒ}n|dkr€d }n|jd |ƒ}|s§td ƒ‚nt|ƒ}ˆsΖ|d |ƒSd j ˆ|gƒ}t |dƒ}|j |ƒWdQX|d|ƒS(Ns.pngs.pdfs.dots.svgs.jpegs.jpgc3`s$|]}ˆjƒj|ƒVqdS(N(tlowertendswith(Rtfmt(Rg(s'lib/python2.7/site-packages/dask/dot.pys όsiRURsΜGraphviz failed to properly produce an image. This probably means your installation of graphviz is missing png support. See: https://github.com/ContinuumIO/anaconda-issues/issues/485 for more information.tdatat.twbRg( R&RtostpathtsplitextRitpipet RuntimeErrorRdtjointopentwrite(RJRgRtfmtsRlt display_clst full_filenametf((Rgs'lib/python2.7/site-packages/dask/dot.pyRfϊs (    (#t __future__RRRR$Rot functoolsRt compatibilityRtcoreRRRtutilsRR R R RR R tcompileR"R#R&R0R4RSt frozensetR_R^RdRhRf(((s'lib/python2.7/site-packages/dask/dot.pyts*     ) 0 )