def write_graph(graph, output_file, filename, output_type='gr'): output = cStringIO.StringIO() symtab = labels = False if output_type == 'gr': from utils.graph_output import write_gr as write_graph write_graph(graph, symtab=symtab, labels=labels, output=output, gtfs_filename=filename) elif output_type == 'gml': from utils.graph_output import write_gml as write_graph write_graph(graph, symtab=symtab, labels=labels, output=output, gtfs_filename=filename) if output_file != '-': #logging.warning('Writing output to file') with open(output_file, 'w') as f: f.write(output.getvalue()) f.flush() #logging.warning('Output written to: %s' % output_file) else: print output.getvalue()
labels=labels, output=output, gtfs_filename=filename) if output_file != '-': #logging.warning('Writing output to file') with open(output_file, 'w') as f: f.write(output.getvalue()) f.flush() #logging.warning('Output written to: %s' % output_file) else: print output.getvalue() for filename in os.listdir(path): # if not filename.endswith('.csvdata'): # continue # if filename.startswith('vertices'): # continue if not filename in ('out.opsahl-powergrid'): continue filepath = os.path.join(path, filename) print 'filepath=', filepath basename = os.path.basename(filepath) g = create_graph(filepath) #write_graph(graph=g,filename='test.gr') write_graph(output_file='%s.gr' % filename, graph=g, filename=basename, output_type='gr') #write_graph(output_file='%s.gml'%filename, graph=g,filename=basename, output_type='gml')