U d/e @s|dZddlZddlZddlZddlZddlmZddlmZdddd d d gZe d ej Z e d Z ddddddhZ ddddddddddh Ze dZej dejd Zeejd!Zejdd"e je je ejefeed#d$dZeed#d%dZejdd"d0ejeed&d'dZejdd"d1ejeed&d(d ZGd)d*d*Zeed+d,d ZGd-d.d.eZ eed+d/d Z!dS)2KQuote strings to be valid DOT identifiers, assemble quoted attribute lists.N)_tools) exceptionsquote quote_edgea_list attr_listescapenohtmlz<.*>$z8([a-zA-Z_][a-zA-Z0-9_]*|-?(\.[0-9]+|[0-9]+(\.[0-9]*)?))$nodeZedgegraphZdigraphZsubgraphstrictnneesesswwnwc_z(?(?:\\{2})*) \\? # treat \" same as " (?P") )flagsz*\g\\\g)Zsupported_number) identifierreturncCsZ||rt|tsnB||r(||krV||rFtjd|tjdd||dS|S)aReturn DOT identifier from string, quote if needed. >>> quote('') # doctest: +NO_EXE '""' >>> quote('spam') 'spam' >>> quote('spam spam') '"spam spam"' >>> quote('-4.2') '-4.2' >>> quote('.42') '.42' >>> quote('<spam>') '<spam>' >>> quote(nohtml('<>')) '"<>"' >>> print(quote('"')) "\"" >>> print(quote('\\"')) "\"" >>> print(quote('\\\\"')) "\\\"" >>> print(quote('\\\\\\"')) "\\\"" z4expect syntax error scanning invalid quoted string: )category") isinstanceNoHtmllowerwarningswarnrZDotSyntaxWarning)rZis_html_stringZ is_valid_idZ dot_keywordsZ"endswith_odd_number_of_backslashesZescape_unescaped_quotesr#4/tmp/pip-unpacked-wheel-n8ok7rre/graphviz/quoting.pyr(s* cCsT|d\}}}t|g}|rJ|d\}}}|t||rJ||d|S)zReturn DOT edge statement node_id from string, quote if needed. >>> quote_edge('spam') # doctest: +NO_EXE 'spam' >>> quote_edge('spam spam:eggs eggs') '"spam spam":"eggs eggs"' >>> quote_edge('spam:eggs:s') 'spam:eggs:s' :) partitionrappendjoin)rr rrestpartsportZcompassr#r#r$r]s   )labelrcCsl|dk rdt|gng}|r8|ddt|D7}|rbt|drPt|}|dd|D7}d|S)zReturn assembled DOT a_list string. >>> a_list('spam', kwargs={'spam': None, 'ham': 'ham ham', 'eggs': ''}) # doctest: +NO_EXE 'label=spam eggs="" ham="ham ham"' Nzlabel=cSs.g|]&\}}|dk rt|dt|qSN=r.0kvr#r#r$ }sza_list..itemscSs.g|]&\}}|dk rt|dt|qSr-r/r0r#r#r$r4s )rrZ mapping_itemshasattrr()r,kwargs attributesresultr#r#r$rss  cCs"t|||d}|sdSd|dS)aReturn assembled DOT attribute list string. Sorts ``kwargs`` and ``attributes`` if they are plain dicts (to avoid unpredictable order from hash randomization in Python < 3.7). >>> attr_list() # doctest: +NO_EXE '' >>> attr_list('spam spam', kwargs={'eggs': 'eggs', 'ham': 'ham ham'}) ' [label="spam spam" eggs=eggs ham="ham ham"]' >>> attr_list(kwargs={'spam': None, 'eggs': ''}) ' [eggs=""]' )r8r9z [])r)r,r8r9contentr#r#r$r sc@s0eZdZdZeeZeeZee Z ee Z dS)QuoterN) __name__ __module__ __qualname____doc__ staticmethodr_quoterZ _quote_edgerZ_a_listr Z _attr_listr#r#r#r$r>s r>)rrcCst|ddS)aReturn string disabling special meaning of backslashes and ``'<...>'``. Args: s: String in which backslashes and ``'<...>'`` should be treated as literal. Returns: Escaped string subclass instance. Raises: TypeError: If ``s`` is not a ``str``. Example: >>> import graphviz # doctest: +NO_EXE >>> print(graphviz.escape(r'\l')) \\l See also: Upstream documentation: https://www.graphviz.org/doc/info/attrs.html#k:escString \z\\)r replacerr#r#r$r sc@seZdZdZdZdS)rzCString subclass that does not treat ``'<...>'`` as DOT HTML string.r#N)r?r@rArB __slots__r#r#r#r$rsrcCst|S)a&Return string not treating ``'<...>'`` as DOT HTML string in quoting. Args: s: String in which leading ``'<'`` and trailing ``'>'`` should be treated as literal. Returns: String subclass instance. Raises: TypeError: If ``s`` is not a ``str``. Example: >>> import graphviz # doctest: +NO_EXE >>> g = graphviz.Graph() >>> g.node(graphviz.nohtml('<>-*-<>')) >>> print(g.source) # doctest: +NORMALIZE_WHITESPACE graph { "<>-*-<>" } )rrGr#r#r$r s)NNN)NNN)"rB functoolsretypingr!r;rr__all__compileDOTALLZ HTML_STRINGZIDZKEYWORDSZCOMPASSZFINAL_ODD_BACKSLASHESVERBOSEZQUOTE_WITH_OPTIONAL_BACKSLASHESpartialsubZESCAPE_UNESCAPED_QUOTESZdeprecate_positional_argsmatchsearchstrrrOptionalrr r>r rr r#r#r#r$sf     4