Make this into more of a CLI thing. Also write some code

2026-06-08 16:08:00 -06:00 · 2024-07-10 20:39:15 -06:00
parent 3377dd7e9a
commit 3662b18e9b
14 changed files with 394 additions and 73692 deletions
@@ -0,0 +1,43 @@
+import sys
+import traceback
+
+import graphs
+import ip_parser
+order = 16
+
+def printHelp():
+  print("""
+    Usage:
+    
+    ./main.py <command> [options]
+    
+    Commands:
+    parse <input file>
+    graph_density <input file>
+    graph_ports <input file>
+    graph_nonstandard_ports <input file>
+    generate_random_ips <input file> <count>
+  """)
+
+if __name__ == '__main__':
+    try:
+        argv = sys.argv
+        match argv[1]:
+            case 'parse':
+                for ip in ip_parser.process_ips(sys.argv[2], verbose=False):
+                    print(ip)
+            case 'graph_density':
+                graphs.graph_density(ip_parser.process_ips(sys.argv[2]))
+            case 'graph_ports':
+                graphs.graph_ports(ip_parser.process_ips(sys.argv[2]))
+            case 'graph_nonstandard_ports':
+                graphs.graph_nonstandard_ports(ip_parser.process_ips(sys.argv[2]))
+            case 'generate_random_ips':
+                graphs.generate_random_ips(ip_parser.process_ips(sys.argv[2]), int(sys.argv[3]))
+            case _:
+                printHelp()
+    except Exception as e:
+        ex_type, ex, tb = sys.exc_info()
+        traceback.print_tb(tb)
+        print(str(e))
+        printHelp()