U d/e4@sdZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z d d d gZ ejejeeddd ZejedddZejedddZejedddZdddejejefeejeejeejdddZejejefejddd Zejd'ddddeeejejefejeejeejeedfeejejedfeeed d d Zejd(ddddeejeejejedfejeejeejeedfeejejedfeeed d"d Zejd)ddddeejeejejedfejeejeejeedfeejejedfeeed d#d Zejd$d%d*dd!d!deejeejejedfejeejeejeedfeejejedfeeed d&d ZdS)+z.Render DOT source files with Graphviz ``dot``.N)DEFAULT_SOURCE_EXTENSION)_tools) exceptions) parameters) dot_command)execute get_format get_filepathrender)outfileformatreturnc Csz t|}Wnptk r||dkrPd|jdt|dtd}t|tj d|jdd|dtj d |YSX|dk st |dk r| |krtj d |d |tj d |S|SdS) 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_suffixesrRequiredArgumentErrorwarningswarnZUnknownSuffixWarningAssertionErrorlowerZFormatSuffixMismatchWarning)r rZinferred_formatmsgr!>/tmp/pip-unpacked-wheel-n8ok7rre/graphviz/backend/rendering.pyr s" "   )rcCsddtDS)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', ...] )sortedrZFORMATSr!r!r!r"r%Fsr%)r rc Cs|jstdt|d|jd\}}}|r6|rFt|jd|}zt|Wn@tk rtd|jdt|d|dt d YnX|S) 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 partitionrrr verify_formatr)r startsepZformat_r!r!r"rOs 2r)renderer formatter)filepathrr+r,rcCsht|}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|]}|dk r|VqdS)Nr!)r#ar!r!r" szget_outfile..) rpromote_pathlikerr(Zverify_rendererZverify_formatterjoin with_suffixr)r-rr+r,Z suffix_argsrr!r!r" get_outfiles  r4cCst|}|dtS)z&Return ``outfile.with_suffix('.gv')``.r)rr1r3r)r r!r!r"r s .)r raise_if_result_existsoverwrite_filepath) enginerr-r+r, neato_no_opquietr r5r6rc CsdS)zBRequire ``format`` and ``filepath`` with default ``outfile=None``.Nr! r7rr-r+r,r8r9r r5r6r!r!r"r s Fc CsdS)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+r,z-O)r+r,r8zoutput file exists: T)cwdr9capture_output)rmaprr1r r nameresolveparentrrr4rcommandrpathexistsFileExistsErrorrr Z run_checkparts) r7rr-r+r,r8r9r r5r6Z outfile_argargscmdr!r!r"r sNN  )....).....F).....F)NNNNNF)__doc__rpathlibtypingr _defaultsrrrrrr __all__PathOptionalstrr Listrr%rUnionPathLiker4r overloadboolintr Zdeprecate_positional_argsr!r!r!r"s       *  6