from django.core.management.base import BaseCommand, CommandError from optparse import make_option from django_extensions.management.modelviz import generate_dot class Command(BaseCommand): option_list = BaseCommand.option_list + ( make_option('--disable-fields', '-d', action='store_true', dest='disable_fields', help='Do not show the class member fields'), make_option('--group-models', '-g', action='store_true', dest='group_models', help='Group models together respective to there application'), make_option('--all-applications', '-a', action='store_true', dest='all_applications', help='Automaticly include all applications from INSTALLED_APPS'), make_option('--output', '-o', action='store', dest='outputfile', help='Render output file. Type of output dependend on file extensions. Use png or jpg to render graph to image.'), make_option('--layout', '-l', action='store', dest='layout', default='dot', help='Layout to be used by GraphViz for visualization. Layouts: circo dot fdp neato nop nop1 nop2 twopi'), ) help = ("Creates 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.") args = "[appname]" label = 'application name' requires_model_validation = True can_import_settings = True def handle(self, *args, **options): if len(args) < 1 and not options['all_applications']: raise CommandError("need one or more arguments for appname") dotdata = generate_dot(args, **options) if options['outputfile']: self.render_output(dotdata, **options) else: self.print_output(dotdata) def print_output(self, dotdata): print dotdata def render_output(self, dotdata, **kwargs): try: import pygraphviz except ImportError, e: raise CommandError("need pygraphviz python module ( apt-get install python-pygraphviz )") vizdata = ' '.join(dotdata.split("\n")).strip() version = pygraphviz.__version__.rstrip("-svn") try: if [int(v) for v in version.split('.')]<(0,36): # HACK around old/broken AGraph before version 0.36 (ubuntu ships with this old version) import tempfile tmpfile = tempfile.NamedTemporaryFile() tmpfile.write(vizdata) tmpfile.seek(0) vizdata = tmpfile.name except ValueError: pass graph = pygraphviz.AGraph(vizdata) graph.layout(prog=kwargs['layout']) graph.draw(kwargs['outputfile'])