mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-08-05 16:58:59 +00:00
60 lines
2.7 KiB
Python
60 lines
2.7 KiB
Python
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'])
|