a 8Xh]9@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZzddlZdZWneydZYn0z0z ddlZWneyddlZYn0dZWneydZYn0Gd d d eZdS) N)settings) BaseCommand CommandError)loader) ModelGraph generate_dot) signalcommandTFcs\eZdZdZdZfddZddZeddZdd d Z dd d Z ddZ ddZ Z S)CommandzCreates a GraphViz dot file for the specified app names. You can pass multiple app names and they will all be combined into a single model. Output is usually directed to a dot file.Tcsdddddddddddddd dddd d dddd d ddddddddddddddddddddddddddddddddddddd d!ddd"d#ddd$d%ddd&d'ddd(d)d*dd+dd)d,dd+d(d-d.dddd/d0dd+d(d1d2dddd3d4dddd5gd6d7d8ddd9d:ddd;dd8d?|_ttd@dA}|r|jD]B}|dB}|dCdDdDdE}||vrN|||j|dF<qNtj|i|dAS)Ga Allow defaults for arguments to be set in settings.GRAPH_MODELS. Each argument in self.arguments is a dict where the key is the space-separated args and the value is our kwarg dict. The default from settings is keyed as the long arg name with '--' removed and any '-' replaced by '_'. For example, the default value for --disable-fields can be set in settings.GRAPH_MODELS['disable_fields']. store_trueF pygraphvizz,Output graph data as image using PyGraphViz.)actiondefaultdesthelppydotz-Output graph data as image using PyDot(Plus).dotzDOutput graph data as raw DOT (graph description language) text data.jsonzOutput graph data as JSONZdisable_fieldsz#Do not show the class member fieldsZdisable_abstract_fieldsz7Do not show the class member fields that were inheritedZ group_modelsz5Group models together respective to their applicationall_applicationsz:Automatically include all applications from INSTALLED_APPSstore outputfileziRender output file. Type of output dependend on file extensions. Use png or jpg to render graph to image.)r rrlayoutzaLayout to be used by GraphViz for visualization. Layouts: circo dot fdp neato nop nop1 nop2 twopi)r rr rthemeZ django2018zTheme to use. Supplied are 'original' and 'django2018'. You can create your own by creating dot templates in 'django_extentions/graph_models/themename/' template directory.Z verbose_namesz%Use verbose_name of models and fieldslanguagez3Specify language used for verbose_name localizationZexclude_columnszPExclude specific column(s) from the graph. Can also load exclude list from file.Zexclude_modelszjExclude specific model(s) from the graph. Can also load exclude list from file. Wildcards (*) are allowed.Zinclude_modelszBRestrict the graph to specified models. Wildcards (*) are allowed.TZ inheritancez$Include inheritance arrows (default) store_falsez!Do not include inheritance arrowsZrelations_as_fieldsz-Do not show relations as fields in the graph.Zrelation_fields_onlyz3Only display fields that are relevant for relationsZ sort_fieldszDo not sort fieldsZhide_edge_labelsz*Do not show relations labels in the graph.Z arrow_shape) ZboxZcrowZcurveZicurveZdiamondrinvnonenormalteeZveezArrow shape to use for relations. Default is dot. Available shapes: box, crow, curve, icurve, diamond, dot, inv, none, normal, tee, vee.)r r rchoicesrZcolor_code_deletionszColor the relations according to their on_delete setting, where it it applicable. The colors are: red (CASCADE), orange (SET_NULL), green (SET_DEFAULT), yellow (SET), blue (PROTECT), grey (DO_NOTHING) and purple (RESTRICT).TBrankdir)rZBTZLRZRLzSet direction of graph layout. Supported directions: "TB", "LR", "BT", "RL", corresponding to directed graphs drawn from top to bottom, from left to right, from bottom to top, and from right to left, respectively. Default is TB.)z --pygraphvizz--pydotz--dotz--jsonz--disable-fields -dz--disable-abstract-fieldsz--group-models -gz--all-applications -az --output -oz --layout -lz --theme -tz--verbose-names -nz --language -Lz--exclude-columns -xz--exclude-models -Xz--include-models -Iz--inheritance -ez--no-inheritance -Ez--hide-relations-from-fields -Rz--relation-fields-onlyz--disable-sort-fields -Sz--hide-edge-labelsz --arrow-shapez--color-code-deletionsz --rankdir GRAPH_MODELSN r-_r ) argumentsgetattrrsplitlstripreplacesuper__init__)selfargskwargsdefaultsargumentZ arg_splitZ setting_opt __class__w/var/www/viveiro_mudafortebrasil/venv/lib/python3.9/site-packages/django_extensions/management/commands/graph_models.pyr+#s      zCommand.__init__cCs:|jddd|jD] }|j|di|j|qdS)z/Unpack self.arguments for parser.add_arguments. app_label*)nargsr"N) add_argumentr%r')r,parserr0r3r3r4 add_argumentss zCommand.add_argumentscs |d}|s8|ds8ttdid}|r0|}ntd|dpDd}tj|\}}|}gdfd d |D}t | }|d krtd d ddD|d krt dd|D} nD|sd} n:|dkrd} n,|dkrd} nt rd} ntrd} ntd|ddkr2| dvr2td| dvrJ|sJtdd tjd d} t|fd!| i|} | | dkr| jd"d#} || |S| jd$d#} |d%} tj d&d'| d(}t|}t| |d)}| dkr|j|fi|S| dkr|j|fi|S|||dS)*Nr5rr!Z app_labelsz&need one or more arguments for appnamer)rr rrcsi|]\}}|vr||qSr3r3).0kvZoutput_opts_namesr3r4 z"Command.handle..zOnly one of %s can be set.z, cSsg|] }d|qS)z--%sr3)r<optr3r3r4 rAz"Command.handle..css|]\}}|r|VqdSNr3)r<keyvalr3r3r4 rAz!Command.handle..rz.dotz.jsonrr rzNeither pygraphviz nor pydotplus could be found to generate the image. To generate text output, use the --json or --dot options.r r)rr rz7--rankdir is not supported for the chosen output format)rr zQAn output file (--output) must be specified when --pydot or --pygraphviz are set.r" cli_optionsT)Zas_jsonFrZdjango_extensions graph_modelsz digraph.dot)template)r&rgetrospathsplitextloweritemssumvaluesjoinnextHAS_PYGRAPHVIZ HAS_PYDOTsysargvrZgenerate_graph_dataZget_graph_datarender_output_jsonrZ get_templaterrender_output_pygraphvizrender_output_pydot print_output)r,r-optionsZdefault_app_labelsrr$Zoutputfile_extZ output_optsZoutput_opts_countoutputrJrK graph_datarZ template_namerLdotdatar3r?r4handles^          zCommand.handleNcCs\t|tr|}|rLt|d}||WdqX1s@0Yn |j|dS)z8Write model data to file or stdout in DOT (text) format.wtN) isinstancebytesdecodeopenwritestdout)r,rb output_fileZ dot_output_fr3r3r4r^s   *zCommand.print_outputcCsR|rr3r3r4rH5rAz3Command.render_output_pygraphviz...)r$rr)progrN)rWrr __version__rstriptupler'tempfileNamedTemporaryFileriseekname ValueErrorZAGraphrZdraw)r,rbr.versionZtmpfilegraphr3r3r4r\.s      z Command.render_output_pygraphvizcKsts tdt|}|s"tdt|ttfrPt|dkrHtj d|d}|d}gd}|| ddd }||vr|nd }|j ||d d S) z'Render model data as image using pydot.z'You need to install pydot python modulezpydot returned an errorrBz9Found more then one graph, rendering only the first one. rr))ZbmpZcanonZcmapZcmapxZcmapx_nprZdiaZemfemZfplusZepsZfiggdZgd2ZgifgvimapZimap_npismapZjpeZjpegZjpgZmetafileZpdfZpicplainz plain-extZpngZpovZpsZps2svgZsvgzZtifZtiffZtkZvmlZvmlzZvrmlZwbmpZxdotroNraw)format) rXrrZgraph_from_dot_datarelistrtlenrYstderrrirfind)r,rbr.r{rkformatsextZformat_r3r3r4r]Bs   zCommand.render_output_pydot)N)N)__name__ __module__ __qualname__rZcan_import_settingsr+r:rrcr^r[r\r] __classcell__r3r3r1r4r s - E r )rYrrNruZ django.confrZdjango.core.management.baserrZdjango.templaterZ%django_extensions.management.modelvizrrZ"django_extensions.management.utilsrr rW ImportErrorZ pydotplusrrXr r3r3r3r4s,