o q+WZ@sddlmZddlZddlZddlZddlmZddZddZd d Z e fd d Z e e e fd dZ ej dej ddejeefddZdS))print_functionN) findMachinescCsd|ddS)Nz"{}""z\")formatreplacesr 4/usr/lib/python3/dist-packages/automat/_visualize.py_gvquote sr cCs d|S)Nz<{}>)rrr r r _gvhtmls r cOs6dddt|D}d|}dj|||dS)z? Construct a string from the HTML element description.  css(|]\}}d|tt|VqdS)z{}={}N)rr str).0keyvaluer r r szelementMaker..z#<{name} {attrs}>{children})nameattrschildren)joinsorteditemsr)rrrZformattedAttrsZformattedChildrenr r r elementMakers   rcsi}|r tt||d<dd|ddfd|d|}dd ifd d |D}d |g}|r>|d g|Rd g|RS)z> Construct an HTML table to label a state transition. colspantdfontz menlo-italic)ZfaceZpurple)colorportz point-size9cs(g|]}dd|fiddqS)rrZpink)rr )rZ outputLabel_EZ pointSizer r 0s ztableMaker..trtable)rlenappend) inputLabelZ outputLabelsr r#rZinputLabelCellZoutputLabelCellsZrowsr r"r tableMakers,  r*cstjdddddiddid}|D]}||jurd}d}nd }d}|j|||d |d d qt|D]H\}} | \} } } } d |}|| }d}t|fdd| D|d}|j|t |ddd|j || d||dd| d|||| q6|S)zA Produce a L{graphviz.Digraph} object from an automaton. trueZ100)ZpackZdpiZfontnameZMenlo)Z graph_attrZ node_attrZ edge_attrZboldz Menlo-BoldrZellipseZblue)ZfontameshapeZstylerzt{}Z tableportcsg|]}|qSr r )rZ outputSymboloutputAsStringr r r$\s zmakeDigraph..)r z0.2Znone)ZlabelZmarginr,z{}:{}:w)Z arrowheadz{}:{}:e) graphvizZDigraphZstatesZ initialStateZnode enumerateZallTransitionsrr*r Zedge)Z automatonZ inputAsStringr.Z stateAsStringdigraphstateZ stateShapeZfontNamenZeachTransitionZinStateZ inputSymbolZoutStateZ outputSymbolsZthisTransitionr)r r&r r-r makeDigraph?sN        r4cCsrd}d}|dr|ddtj|||d}|jddd|jd d d d d d|jddddd|jddddd|jdddtjdd|jdddd d d||}|jo_|j p_|j|jk} |j rgdd }||j D]J\} } || d!| } | r| j d" | |jd#|| d$|j|jr|j p| } |j| _ | jd" | |j|j| d%| rd&}nd'}|| ||jqld(S))z/ Entry point for command line utility. zB Visualize automat.MethodicalMachines as graphviz graphs. zz You must have the graphviz tool suite installed. Please visit http://www.graphviz.org for more information. rr)prog descriptionepilogfqpnz@A Fully Qualified Path name representing where to find machines.)helpz--quietz-qzsuppress outputF store_true)r9defaultactionz--dot-directoryz-dzWhere to write out .dot files.z.automat_visualize)r9r;z--image-directoryz-izWhere to write out image files.z --image-typez-tzThe image format.Zpng)r9choicesr;z--viewz-vz.View rendered graphs with default image viewercWsdS)Nr )argsr r r _printsztool.._printz ...discoveredz{}.dot)filename directoryz...wrote dot into)r@rAviewZcleanupz...wrote image intoz...wrote image and dot intoN)insertargparseArgumentParser add_argumentr/ZFORMATS parse_argsZ dot_directoryZimage_directoryquietr8Z asDigraphZsaverZ image_typeZrenderrB)Z _prognameZ_argvZ_syspathZ _findMachinesr?Z DESCRIPTIONZEPILOGZargumentParserr>ZexplicitlySaveDotr8machiner1Z deleteDotmsgr r r toolls        rK)Z __future__rrDsysr/Z _discoverrr r rr*reprr4argvpathprintrKr r r r s&     -