2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

buildgraph: add options whether to draw the graph

Browse source
This commit is contained in:
Laura Klünder 2016-12-04 14:03:11 +01:00
parent 01e6674cd0
commit 59143db553
3 changed files with 30 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/c3nav/routing/management/commands/buildgraph.py
View file

@ -6,6 +6,18 @@ from c3nav.routing.graph import Graph
class Command(BaseCommand):
help = 'build the routing graph'
def add_arguments(self, parser):
parser.add_argument('--draw-graph', action='store_const', const=True, default=False,
help='render a graph image')
parser.add_argument('--dont-draw-graph-points', action='store_const', const=True, default=False,
help='dont draw points on the graph image')
parser.add_argument('--dont-draw-graph-lines', action='store_const', const=True, default=False,
help='dont draw lines on the graph image')
def handle(self, *args, **options):
graphbuilder = Graph()
graphbuilder.build()
graph = Graph()
graph.build()
if options['draw_graph']:
graph.draw_pngs(points=not options['dont_draw_graph_points'], lines=not options['dont_draw_graph_lines'])