o /h F@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZzddlZdZWn eyEdZYnwzzddlZWn eyZddlZYnwdZWn eyidZYnwifd d ZGd d d eZdS) N)settings) BaseCommand CommandError)loader) ModelGraph generate_dot) signalcommandTFc Cst|trAtj|r?zt|d }t|}Wdn1s!wYWnty>}z t d|t|d}~ww|S|dD] }|D]}d|vrd|dD]}|d}||vrc|||d<qSqIqE|S)NrtzInvalid app style file graphsnamemodelsapp_namestyle) isinstancestrospathexistsopenjsonload Exceptionprint) graph_data app_stylefegcgmr r m/var/www/html/env_mimamsha/lib/python3.10/site-packages/django_extensions/management/commands/graph_models.pyrethemes.       r"cs`eZdZdZ dZfddZddZeddZdd d Z dd d Z ddZ ddZ Z S)Commandz8Creates a GraphViz dot file for the specified app names.Tcsiddddddddd d d d d dd ddd ddd ddd ddd ddd ddd ddd ddd ddd ddd ddd ddd d d!d d"dd d#d$d d%dd&d'd(d)dd*dd+d,d-dd.d/d0d,d1dd d2d3d d4dd5d6d(d7dd8d9d(d:dd;dd(dd?d@dAd dBd d@dCd dBd?dDdEd dd dFdGd dBd?dHdId dd dJdKd dddLgdMdNdOdd dPdQd ddRdSgdTdUdOddVdWdXdYgdZdOd[ |_ttd\dV}|r|jD]}|d]}|d^d_d_d`}||vr|||j|da<qtj|i|dVS)ba 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']. z --app-stylestorez1Path to style json to configure the style per app app-stylez.app-style.json)actionhelpdestdefaultz --pygraphviz store_trueF pygraphvizz,Output graph data as image using PyGraphViz.)r&r)r(r'z--pydotpydotz-Output graph data as image using PyDot(Plus).z--dotdotzDOutput graph data as raw DOT (graph description language) text data.z--jsonrzOutput graph data as JSONz--disable-fields -ddisable_fieldsz#Do not show the class member fieldsz--disable-abstract-fieldsdisable_abstract_fieldsz7Do not show the class member fields that were inheritedz--display-field-choicesdisplay_field_choicesz%Display choices instead of field typez--group-models -g group_modelsz5Group models together respective to their applicationz--all-applications -aall_applicationsz:Automatically include all applications from INSTALLED_APPSz --output -o outputfileziRender output file. Type of output dependend on file extensions. Use png or jpg to render graph to image.)r&r(r'z --layout -llayoutzaLayout to be used by GraphViz for visualization. Layouts: circo dot fdp neato nop nop1 nop2 twopi)r&r(r)r'z --theme -ttheme 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-names -n verbose_namesz%Use verbose_name of models and fieldsz --language -Llanguagez3Specify language used for verbose_name localizationz--exclude-columns -xexclude_columnszPExclude specific column(s) from the graph. Can also load exclude list from file.z--exclude-models -Xexclude_modelszjExclude specific model(s) from the graph. Can also load exclude list from file. Wildcards (*) are allowed.include_modelszBRestrict the graph to specified models. Wildcards (*) are allowed.T inheritancez$Include inheritance arrows (default) store_falsez!Do not include inheritance arrowsrelations_as_fieldsz-Do not show relations as fields in the graph.relation_fields_onlyz3Only display fields that are relevant for relations sort_fieldszDo not sort fieldshide_edge_labelsz*Do not show relations labels in the graph. arrow_shape) boxcrowcurveicurvediamondr-invnonenormalteeveezArrow shape to use for relations. Default is dot. Available shapes: box, crow, curve, icurve, diamond, dot, inv, none, normal, tee, vee.)r&r)r(choicesr'color_code_deletionszColor the relations according to their on_delete setting, where it is applicable. The colors are: red (CASCADE), orange (SET_NULL), green (SET_DEFAULT), yellow (SET), blue (PROTECT), grey (DO_NOTHING), and purple (RESTRICT).TBrankdir)rOBTLRRLzSet direction of graph layout. Supported directions: TB, LR, BT and 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.NorderinginoutzControls how the edges are arranged. Supported orderings: "in" (incoming relations first), "out" (outgoing relations first). Default is None.) z--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 --rankdirz --ordering GRAPH_MODELS r-_r)) argumentsgetattrrsplitlstripreplacesuper__init__)selfargskwargsdefaultsargument arg_split setting_opt __class__r r!ra<sr  "(.4:@HOW]bho     L  zCommand.__init__cCs:|jddd|jD]}|j|di|j|q dS)z/Unpack self.arguments for parser.add_arguments. app_label*)nargsrXN) add_argumentr[r])rbparserrfr r r! add_argumentss zCommand.add_argumentscs>|d}|s|dsttdid}|r|}ntd|dp"d}tj|\}}|}gdfd d |D}t | }|d krVtd d ddD|d krft dd|D} n!|skd} n|dkrrd} n|dkryd} nt r~d} n trd} ntd|ddkr| dvrtd|dr| dvrtd| dvr|std d! tjd"d} t|fd#| i|} | | dkr| jd$d%} || |S| jd&d%} |d'} tj d(d)| d*}t|}t| |d+d,} t| |d-}| dkr |j|fi|S| dkr|j|fi|S|||dS).Nrkr2rW app_labelsz&need one or more arguments for appnamer3)r,r+rr-csi|] \}}|vr||qSr r ).0kvoutput_opts_namesr r! 3sz"Command.handle..zOnly one of %s can be set.z, cSsg|]}d|qS)z--%sr )rsoptr r r! 8sz"Command.handle..css|] \}}|r|VqdSNr )rskeyvalr r r! <sz!Command.handle..r-z.dotz.jsonrr+r,zNeither pygraphviz nor pydotplus could be found to generate the image. To generate text output, use the --json or --dot options.rPrO)r,r+r-z7--rankdir is not supported for the chosen output formatrTz8--ordering is not supported for the chosen output format)r,r+zQAn output file (--output) must be specified when --pydot or --pygraphviz are set.rX cli_optionsT)as_jsonFr5django_extensions graph_modelsz digraph.dotr%)r)template)r\rgetrrrsplitextloweritemssumvaluesjoinnextHAS_PYGRAPHVIZ HAS_PYDOTsysargvrgenerate_graph_dataget_graph_datarender_output_jsonr get_templater"rrender_output_pygraphvizrender_output_pydot print_output)rbrcoptionsdefault_app_labelsr3rZoutputfile_ext output_optsoutput_opts_countoutputrrrr5 template_namerdotdatar rvr!handle#s          zCommand.handleNcCs`t|tr |}|r(t|d}||WddS1s!wYdS|j|dS)z8Write model data to file or stdout in DOT (text) format.wtN)rbytesdecoderwritestdout)rbr output_file dot_output_fr r r!rs   "zCommand.print_outputcCsV|r t|d}t||WddS1swYdS|jt|dS)z2Write model data to file or stdout in JSON format.rN)rrdumprrdumps)rbrr json_output_fr r r!rs  "zCommand.render_output_jsoncKststdtjd}z!tdd|dDdkr,t}| || d|j }Wn t y6Ynwt |}|j|dd ||d d S) z,Render model data as image using pygraphviz.z,You need to install pygraphviz python modulez-svncss|]}t|VqdSr|)int)rsrur r r!rsz3Command.render_output_pygraphviz...)r$rr4)progr3N)rrr+ __version__rstriptupler]tempfileNamedTemporaryFilerseekr ValueErrorAGraphr4draw)rbrrdversiontmpfilegraphr r r!rs      z Command.render_output_pygraphvizcKststdt|}|stdt|ttfr(t|dkr$tj d|d}|d}gd}|| ddd }||vrA|nd }|j ||d d S) z'Render model data as image using pydot.z'You need to install pydot python modulezpydot returned an errorryz9Found more then one graph, rendering only the first one. rr3)*bmpcanoncmapcmapxcmapx_npr-diaemfemfplusepsfiggdgd2gifgvimapimap_npismapjpejpegjpgmetafilepdfpicplainz plain-extpngpovpsps2svgsvgztiftifftkvmlvmlzvrmlwbmpwebpxdotrNraw)format) rrr,graph_from_dot_datarlistrlenrstderrrrfind)rbrrdrrformatsextformat_r r r!rs   ,zCommand.render_output_pydotr|)__name__ __module__ __qualname__r'can_import_settingsrarprrrrrr __classcell__r r rir!r#5s b  \ r#)rrrr django.confrdjango.core.management.baserrdjango.templater%django_extensions.management.modelvizrr"django_extensions.management.utilsrr+r ImportError pydotplusr,rr"r#r r r r!s6