U d/e@sdZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z d gZeeZGd d d e je je je je jZdS) zDSave DOT code objects, render with Graphviz dot, and open in viewer.N)DEFAULT_ENCODING)_tools)saving)jupyter_integration)piping) rendering) unflatteningSourcecseZdZdZeejdddddeddfej e j e fej e j e dfej e ej e ej e ej e ej e ddddZejddddddefdddde ej e j e dfej e j e dfej e ej e ej e ej e ej e ej e j dd fd d Zfd d Zeje dddZee dddZejdddddej e j e dfej e j e dfej ee dfddZZS)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. )Zsupported_numberN)filename directoryformatengineencodingrenderer formatterreturnc Cst|}|jrtj||nt|}|dkr8t}t d||t ||d} | } W5QRX|| ||||||||d S)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)r r rrrrrloaded_from_path) rZpromote_pathlike_directorypartsospathjoinfspathlocalegetpreferredencodinglogdebugopenread) clsr r rrrrrfilepathfdsourcer$4/tmp/pip-unpacked-wheel-n8ok7rre/graphviz/sources.py from_file)s& zSource.from_file)rrr) r#r r rrrrrrrc s*tj|||||||d| |_||_dS)N)r r rrrrr)super__init___loaded_from_path_source) selfr#r r rrrrrr __class__r$r%r(Ls zSource.__init__c stjf|j|jd|S)z3Return the kwargs to create a copy of the instance.)r#r)r' _copy_kwargsr*r))r+kwargsr,r$r%r.]s  zSource._copy_kwargs)rccsR|jjdd}|ddEdH|ddD] }|ds>dnd}||Vq,dS)zwYield the DOT source code read from file line by line. Yields: Line ending with a newline (``'\n'``). T)keependsN )r* splitlinesendswith)r+lineslinesuffixr$r$r%__iter__cs zSource.__iter__cCs|j}|ds|d7}|S)znThe DOT source code as string. Normalizes so that the string always ends in a final newline. r2)r*r5)r+r#r$r$r%r#ns z Source.source) skip_existing)r r r:rcsD|dko|jotj|j|j}|r2td|jtj|||dS)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))r r r:) r)rrsamefiler!rrr'save)r+r r r:skipr,r$r%r<ys z Source.save)NN)__name__ __module__ __qualname____doc__ classmethodrZdeprecate_positional_argsrtypingUnionrPathLikestrOptionalr&r(r.Iteratorr9propertyr#boolr< __classcell__r$r$r,r%r sj !     )rArloggingrrCrrr3rrrrrr __all__ getLoggerr>rZRenderZSaveZJupyterIntegrationZPipeZ Unflattenr r$r$r$r%s$