o ooia4@sddZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z gd Z d ejd ejedefddZdejefddZdejefddZd ejdefddZddddejejefd edejedejedejf ddZd ejejefdejfddZej    d.dddd d!ed edejejefdejedejed"ejeedfd#ed ejejedfd$ed%edefd&d'Zej      (d/dddd d!ed ejedejejedfdejedejed"ejeedfd#ed ejejedfd$ed%edefd)d'Zej      (d/dddd d!ed ejedejejedfdejedejed"ejeedfd#ed ejejedfd$ed%edefd*d'Zejd+d,      (d0dd(d(d d!ed ejedejejedfdejedejed"ejeedfd#ed ejejedfd$ed%edefd-d'ZdS)1z.Render DOT source files with Graphviz ``dot``.N)DEFAULT_SOURCE_EXTENSION)_tools) exceptions) parameters) dot_command)execute) get_format get_filepathrenderoutfileformatreturnc Cszt|}Wn7ty=|dur'd|jdt|dtd}t|tj d|jdd|dtj d |YSw|dusDJ|dur^| |kr^tj d |d |tj d |S|S) a Return format inferred from outfile suffix and/or given ``format``. Args: outfile: Path for the rendered output file. format: Output format for rendering (``'pdf'``, ``'png'``, ...). Returns: The given ``format`` falling back to the inferred format. Warns: graphviz.UnknownSuffixWarning: If the suffix of ``outfile`` is empty/unknown. graphviz.FormatSuffixMismatchWarning: If the suffix of ``outfile`` does not match the given ``format``. N*cannot infer rendering format from suffix of outfile: z/ (provide format or outfile with a suffix from )zunknown outfile suffix z (expected: .)categoryzexpected format z) from outfile differs from given format: ) infer_format ValueErrorsuffixosfspathget_supported_suffixesrRequiredArgumentErrorwarningswarnUnknownSuffixWarninglowerFormatSuffixMismatchWarning)r rinferred_formatmsgr#N/home/ubuntu/.local/lib/python3.10/site-packages/graphviz/backend/rendering.pyr s8     r cCsddtDS)zReturn a sorted list of supported outfile suffixes for exception/warning messages. >>> get_supported_suffixes() # doctest: +ELLIPSIS ['.bmp', ...] cSsg|]}d|qS)rr#).0rr#r#r$ Csz*get_supported_suffixes..)get_supported_formatsr#r#r#r$r=srcCs ttjS)zReturn a sorted list of supported formats for exception/warning messages. >>> get_supported_formats() # doctest: +ELLIPSIS ['bmp', ...] )sortedrFORMATSr#r#r#r$r'Fs r'c Cs|jstdt|d|jd\}}}|r|r#J|jd|}zt|W|StyLtd|jdt|d|dtd w) aReturn format inferred from outfile suffix. Args: outfile: Path for the rendered output file. Returns: The inferred format. Raises: ValueError: If the suffix of ``outfile`` is empty/unknown. >>> infer_format(pathlib.Path('spam.pdf')) # doctest: +NO_EXE 'pdf' >>> infer_format(pathlib.Path('spam.gv.svg')) 'svg' >>> infer_format(pathlib.Path('spam.PNG')) 'png' >>> infer_format(pathlib.Path('spam')) Traceback (most recent call last): ... ValueError: cannot infer rendering format from outfile: 'spam' (missing suffix) >>> infer_format(pathlib.Path('spam.wav')) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE Traceback (most recent call last): ... ValueError: cannot infer rendering format from suffix '.wav' of outfile: 'spam.wav' (unknown format: 'wav', provide outfile with a suffix from ['.bmp', ...]) z,cannot infer rendering format from outfile: z (missing suffix)rz.startswith('.')rrz (unknown format: z%, provide outfile with a suffix from r) rrrr partitionrr verify_formatr)r startsepformat_r#r#r$rOs*     r)renderer formatterfilepathr/r0cCsht|}tj|ddtj|ddtj|dd|||f}ddd|D}||jd|S)zReturn ``filepath`` + ``[[.formatter].renderer].format``. See also: https://www.graphviz.org/doc/info/command.html#-O T)requiredFrcss|] }|dur|VqdS)Nr#)r%ar#r#r$ szget_outfile..) rpromote_pathlikerr+verify_rendererverify_formatterjoin with_suffixr)r1rr/r0 suffix_argsrr#r#r$ get_outfiles  r;cCst|}|dtS)z&Return ``outfile.with_suffix('.gv')``.r)rr5r9r)r r#r#r$r s r .)r raise_if_result_existsoverwrite_filepathengine neato_no_opquietr<r=c CdS)zBRequire ``format`` and ``filepath`` with default ``outfile=None``.Nr# r>rr1r/r0r?r@r r<r=r#r#r$r r Fc CrA)z>> doctest_mark_exe() >>> import pathlib >>> import graphviz >>> assert pathlib.Path('doctest-output/spam.gv').write_text('graph { spam }') == 14 >>> graphviz.render('dot', 'png', 'doctest-output/spam.gv').replace('\\', '/') 'doctest-output/spam.gv.png' >>> graphviz.render('dot', filepath='doctest-output/spam.gv', ... outfile='doctest-output/spam.png').replace('\\', '/') 'doctest-output/spam.png' >>> graphviz.render('dot', outfile='doctest-output/spam.pdf').replace('\\', '/') 'doctest-output/spam.pdf' Note: The layout command is started from the directory of ``filepath``, so that references to external files (e.g. ``[image=images/camelot.png]``) can be given as paths relative to the DOT source file. See also: Upstream docs: https://www.graphviz.org/doc/info/command.html zAoverwrite_filepath cannot be combined with raise_if_result_existsN)rzoutfile z# must be different from input file z+ (pass overwrite_filepath=True to override)z-oz1filepath: (required if outfile is not given, got rz/format: (required if outfile is not given, got )rr/r0z-O)r/r0r?zoutput file exists: zwork around pytype false alarmT)cwdr@capture_output)rmaprr5r r nameresolveparentrrr;rcommandrpathexistsFileExistsErrorrr run_checkparts) r>rr1r/r0r?r@r r<r= outfile_argargscmdr#r#r$r sZN       )....).....F)NNNNNF)__doc__rpathlibtypingr _defaultsrrrrrr __all__PathOptionalstrr Listrr'rUnionPathLiker;r overloadboolintr deprecate_positional_argsr#r#r#r$s4      *  6