3 g]9@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZyddlZdZWnek rdZYnXy2y ddlZWnek rddlZYnXdZWnek rdZYnXGd 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.Tc$sdddddddddddddd 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(d1d2dddd3d4dddd5d6d7d8d9d:dd;dd?g d@dAdddBdCdddDdEdDdFdGdHgdIdAdJ|_ttdKdL}|rxL|jD]B}|jdM}|dNjdOjdOdP}||krf|||j|dQ<qfWtj||dLS)Ra 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_shapeZboxZcrowZcurveZicurveZdiamondinvnonenormalteeZveezArrow 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).TBrankdirZBTZLRZRLzSet 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__p/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/management/commands/graph_models.pyr+#s      zCommand.__init__cCs:|jdddx&|jD]}|j|jd|j|qWdS)z/Unpack self.arguments for parser.add_arguments. app_label*)nargsr"N) add_argumentr%r')r,parserr0r2r2r3 add_argumentss zCommand.add_argumentscs$|d}| r<|d rsz"Command.handle..zOnly one of %s can be set.z, cSsg|] }d|qS)z--%sr2)r;optr2r2r3 sz"Command.handle..css|]\}}|r|VqdS)Nr2)r;keyvalr2r2r3 sz!Command.handle..z.dotz.jsonzNeither pygraphviz nor pydotplus could be found to generate the image. To generate text output, use the --json or --dot options.r rz7--rankdir is not supported for the chosen output formatzQAn output file (--output) must be specified when --pydot or --pygraphviz are set.r" cli_optionsT)Zas_jsonFrZdjango_extensions graph_modelsz digraph.dot)template)rr r)rr )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_countoutputrGrH graph_datarZ template_namerIdotdatar2)r>r3handles^          zCommand.handleNc CsHt|tr|j}|r8t|d}|j|WdQRXn |jj|dS)z8Write model data to file or stdout in DOT (text) format.wtN) isinstancebytesdecodeopenwritestdout)r,r_ output_fileZ dot_output_fr2r2r3r[s   zCommand.print_outputc Cs>|r(t|d}tj||WdQRXn|jjtj|dS)z2Write model data to file or stdout in JSON format.raN)rerdumprgrfdumps)r,r^rhZ json_output_fr2r2r3rX&s zCommand.render_output_jsonc Ksts tdtjjd}yBtdd|jdDd krXtj}|j ||j d|j }Wnt k rnYnXtj |}|j|dd |j|d d S) z,Render model data as image using pygraphviz.z,You need to install pygraphviz python modulez-svncss|]}t|VqdS)N)int)r;r=r2r2r3rE5sz3Command.render_output_pygraphviz...r$r)progrN)rrm)rTrr __version__rstriptupler'tempfileNamedTemporaryFilerfseekname ValueErrorZAGraphrZdraw)r,r_r.versionZtmpfilegraphr2r2r3rY.s     z Command.render_output_pygraphvizc)Ksts tdtj|}|s"tdt|ttfrPt|dkrHtj j d|d}|d}ddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/g)}||j d0dd1}||kr|nd2}|j ||d3d1S)4z'Render model data as image using pydot.z'You need to install pydot python modulezpydot returned an errorr@z9Found more then one graph, rendering only the first one. rrZbmpZcanonZcmapZcmapxZcmapx_nprZdiaZemfemZfplusZepsZfiggdZgd2ZgifgvimapZimap_npismapZjpeZjpegZjpgZmetafileZpdfZpicplainz plain-extZpngZpovZpsZps2svgZsvgzZtifZtiffZtkZvmlZvmlzZvrmlZwbmpZxdotrlNraw)format) rUrrZgraph_from_dot_datarblistrqlenrVstderrrfrfind)r,r_r.rxrhformatsextZformat_r2r2r3rZBs$   zCommand.render_output_pydot)N)N)__name__ __module__ __qualname__rZcan_import_settingsr+r9rr`r[rXrYrZ __classcell__r2r2)r1r3r s - F r )rVrrKrrZ django.confrZdjango.core.management.baserrZdjango.templaterZ%django_extensions.management.modelvizrrZ"django_extensions.management.utilsrr rT ImportErrorZ pydotplusrrUr r2r2r2r3s,