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

Do not use argparse

`argparse` is overkill here.
parent cd392c26
No related branches found
No related tags found
No related merge requests found
......@@ -37,19 +37,15 @@ def add_component(G: nx.DiGraph, source: int):
G.nodes[n]["component"] = component
if __name__ == "__main__":
import argparse
# Parse arguments:
parser = argparse.ArgumentParser()
parser.add_argument("edgelist_file")
parser.add_argument(
"node", help="Select component containing node", type=int
)
args = parser.parse_args()
G = report_graph.read_eddy_graph(args.edgelist_file)
add_component(G, args.node)
H = G.subgraph(G.nodes[args.node]["component"])
import sys
if len(sys.argv) != 3:
sys.exit("Required arguments: input-file node-number")
node = int(sys.argv[2])
G = report_graph.read_eddy_graph(sys.argv[1])
add_component(G, node)
H = G.subgraph(G.nodes[node]["component"])
len_H = len(H)
print("Number of nodes:", len_H)
print("Number of edges:", H.number_of_edges())
......
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