Skip to content
Snippets Groups Projects
Commit 6edc6da8 authored by Lionel GUEZ's avatar Lionel GUEZ
Browse files

Create a "main part" in the script

Because we are going to add a function and we will need to import
`edgelist_to_Graphviz`.
parent e9b82daf
No related branches found
No related tags found
No related merge requests found
#!/usr/bin/env python3 #!/usr/bin/env python3
import csv if __name__ == "__main__":
import sys import csv
from os import path import sys
from os import path
import pygraphviz as pgv import pygraphviz as pgv
import report_graph import report_graph
if len(sys.argv) != 3: if len(sys.argv) != 3:
sys.exit("Required arguments: input-edgelist output-Graphviz") sys.exit("Required arguments: input-edgelist output-Graphviz")
dir_edgelist = path.dirname(sys.argv[1]) dir_edgelist = path.dirname(sys.argv[1])
fname = path.join(dir_edgelist, "e_overestim.txt") fname = path.join(dir_edgelist, "e_overestim.txt")
with open(fname) as f_obj: e_overestim = int(f_obj.read()) with open(fname) as f_obj: e_overestim = int(f_obj.read())
G = pgv.AGraph(directed = True) G = pgv.AGraph(directed = True)
with open(sys.argv[1], newline = '') as f: with open(sys.argv[1], newline = '') as f:
reader = csv.reader(f, delimiter = " ", skipinitialspace = True) reader = csv.reader(f, delimiter = " ", skipinitialspace = True)
for row in reader: G.add_edge(*row) for row in reader: G.add_edge(*row)
report_graph.partition_graph_date(G, e_overestim) report_graph.partition_graph_date(G, e_overestim)
G.write(sys.argv[2]) G.write(sys.argv[2])
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment