ci] (dZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z d gZejeZGd d e j$e j&e j(e j*e j,Zy) zDSave DOT code objects, render with Graphviz dot, and open in viewer.N)DEFAULT_ENCODING)_tools)saving)jupyter_integration)piping) rendering) unflatteningSourcec`eZdZUdZeej dddddeddfdeje je fdeje je dfdeje d eje d eje d eje d eje d dfdZej ddddddefddddde deje je dfdeje je dfdeje d eje d eje d eje d eje deje jd dffdZe ed<eje jed<fdZd ej$e fdZed e fdZej dd ddddeje je dfdeje je dfdejed e ffdZxZS)r a{Verbatim DOT source code string to be rendered by Graphviz. Args: source: The verbatim DOT source code string. filename: Filename for saving the source (defaults to ``'Source.gv'``). directory: (Sub)directory for source saving and rendering. format: Rendering output format (``'pdf'``, ``'png'``, ...). engine: Layout engine used (``'dot'``, ``'neato'``, ...). encoding: Encoding for saving the source. Note: All parameters except ``source`` are optional. All of them can be changed under their corresponding attribute name after instance creation. rcls)supported_number ignore_argNfilename directoryformatengineencodingrenderer formatterreturnc tj|}|jr tjj ||ntj |}|tj}tjd||t||5} | j} ddd| |||||||| S#1swYxYw)aReturn an instance with the source string read from the given file. Args: filename: Filename for loading/saving the source. directory: (Sub)directory for source loading/saving and rendering. format: Rendering output format (``'pdf'``, ``'png'``, ...). engine: Layout command used (``'dot'``, ``'neato'``, ...). encoding: Encoding for loading/saving the source. Nzread %r with encoding %r)r)rrrrrrrloaded_from_path) rpromote_pathlike_directorypartsospathjoinfspathlocalegetpreferredencodinglogdebugopenread) r rrrrrrrfilepathfdsources 7/tmp/pip-target-z3e9_cxr/lib/python/graphviz/sources.py from_filezSource.from_file)s$55i@ 9BBGGLLH58,   224H ,hA (X . "WWYF 6$  ($ $, . .  s B;;Cself)rrrr(rc Nt ||||||||| |_||_y)N)rrrrrrr)super__init___loaded_from_path_source) r+r(rrrrrrrr __class__s r)r.zSource.__init__Ls9 (i &v"*i"*  ,"2 r0r/c Pt|d|j|jd|S)z3Return the kwargs to create a copy of the instance.)r(r)r- _copy_kwargsr0r/)r+kwargsr1s r)r5zSource._copy_kwargsas1w#.4<<595K5K.&,. .r2c#K|jjd}|ddEd{|ddD]}|jdsdnd}||z y7+w)zwYield the DOT source code read from file line by line. Yields: Line ending with a newline (``'\n'``). T)keependsN )r0 splitlinesendswith)r+lineslinesuffixs r)__iter__zSource.__iter__gsd  '''6":"#J D!%t!4T"F-   s%AA,AcJ|j}|jds|dz }|S)znThe DOT source code as string. Normalizes so that the string always ends in a final newline. r:)r0r=)r+r(s r)r(z Source.sourcers' t$ dNF r2) skip_existingrCc|duxrB|jxr4tjj|j|j}|r t j d|jt|!|||S)aSave the DOT source to file. Ensure the file ends with a newline. Args: filename: Filename for saving the source (defaults to ``name`` + ``'.gv'``) directory: (Sub)directory for source saving and rendering. skip_existing: Skip write if file exists (default: ``None``). By default skips if instance was loaded from the target path: ``.from_file(self.filepath)``. Returns: The (possibly relative) path of the saved source file. Nz;.save(skip_existing=None) skip writing Source.from_file(%r))rrrC) r/rrsamefiler&r"r#r-save)r+rrrCskipr1s r)rFz Source.save}sz %L$*@*@LGG$$T%;%;T]]K  IISmm %w|X*.0 0r2)NN)__name__ __module__ __qualname____doc__ classmethodrdeprecate_positional_argsrtypingUnionrPathLikestrOptionalr*r.__annotations__r5IteratorrApropertyr(boolrF __classcell__)r1s@r)r r s %V%%qUKDH15153C3748 .bkk3.>!?.#\\"++sD*@A. //#..!//#..#OOC0 . #OOC0 . $__S1 . >F .L.B&V%%qVLBFCG04042B  3737BFs!<< S$(>?"LLc4)?@ - -  "??3/  "??3/ #OOC0$*??2;;#?LPM"Lr{{33.  &//#.  &V%%qVLDH?C0480V\\"++sD*@A0 R[[#t%;<0"OOD10=@0M0r2)rKr loggingrrNrrr;rrrrr r __all__ getLoggerrHr"RenderSaveJupyterIntegrationPipe Unflattenr r4r2r)r`svJ  &! *g!}0Y  v{{ 3 3V[[  # #}0r2