o ooi]@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) unflatteningSourcecseZdZUdZeejddddddeddfdej e j e fdej e j e dfdej e d ej e d ej e d ej e d ej e d dfddZejdddddddefddddde dej e j e dfdej e j e dfdej e d ej e d ej e d ej e d ej e dej e j d dffddZe ed<ej e j ed<fddZd eje fddZed e fddZejddd  d"dddej e j e dfdej e j e dfdej ed e ffd d!ZZS)#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 Cst|}|jrtj||nt|}|durt}t d||t ||d } | } Wdn1s8wY|| ||||||||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)rrrrrrrloaded_from_path) rpromote_pathlike_directorypartsospathjoinfspathlocalegetpreferredencodinglogdebugopenread) r rrrrrrrfilepathfdsourcer&D/home/ubuntu/.local/lib/python3.10/site-packages/graphviz/sources.py from_file)s  zSource.from_fileself)rrrr%rc s*tj|||||||d| |_||_dS)N)rrrrrrr)super__init___loaded_from_path_source) r)r%rrrrrrrr __class__r&r'r+Ls  zSource.__init__r-r,c stjd|j|jd|S)z3Return the kwargs to create a copy of the instance.)r%rNr&)r* _copy_kwargsr-r,)r)kwargsr.r&r'r0as zSource._copy_kwargsccsT|jjdd}|ddEdH|ddD]}|ds dnd}||VqdS)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__gs zSource.__iter__cCs|j}|ds |d7}|S)znThe DOT source code as string. Normalizes so that the string always ends in a final newline. r4)r-r7)r)r%r&r&r'r%rs z Source.source) skip_existingr<csD|duo|jotj|j|j}|rtd|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))rrr<) r,rrsamefiler#rr r*save)r)rrr<skipr.r&r'r>}s z Source.save)NN)__name__ __module__ __qualname____doc__ classmethodrdeprecate_positional_argsrtypingUnionrPathLikestrOptionalr(r+__annotations__r0Iteratorr;propertyr%boolr> __classcell__r&r&r.r'r s   !     )rCrloggingrrFrrr5rrrrrr __all__ getLoggerr@rRenderSaveJupyterIntegrationPipe Unflattenr r&r&r&r's$